Akashc6/aathraos-mobility
0
1�
2 �i� � �� � U d dl mZmZmZ d dlmZ d dlZd dlZd dlm Z d dl3m Z d dl m
Z
edd� � Zdae e ed4< dae ej$ ed < G d � d
e� Z G d� de� Zdedefd�Zej/ ddg�� defd�� Zej/ ddg�� d� � Zej5 dedg� � d � � Zej5 d dg�� d � � Z e d k( rd dl Z e j> d!d"d#d$�%� yy)&� )�FastAPI�BackgroundTasks�
HTTPException)� BaseModelN)�Optional)�TrafficCounter)�
PeakPredictorz'Traffic & Crowd Real-time Analytics APIzIAPI to control and query object tracking models for real-time CCTV feeds.)�title� description�active_tracker�tracker_threadc �& � e Zd ZU eed< dZeed< y)�StreamStartRequest�urlz data/live_api_traffic_log.csv�5output_logN)�__name__�6__module__� __qualname__�str�__annotations__r � � �A/media/akash/New_Volume/amd sling/traffic_crowd_prediction/api.pyr r s � � �H�5�J��5r r c �V � e Zd ZU eed< dZee ed< dZeed< dZ eed< dZ7eed< y)�TrafficStatusResponse�statusN�
active_streamr �persons�vehicles�total) r r r r r r r r �intr r r r r r r s2 � � �K�#'�M�8�C�=�'��G�S���H�c���E�3�Nr r r �8output_csvc �� � t | |�� at d| � �� t j d�� d ay # t $ r}t d|� �� Y d }~d ay d }~ww xY w# d aw xY w)N)�9video_pathr" z'Starting Background Tracker on Stream: F)�10show_videoz Error in background tracker: )r r �print�
process_video� Exception)r r" �es r �run_tracker_in_backgroundr* sm � � �'�3�:�N��
�7��u�=�>��$�$��$�6� ��� � 3�
�-�a�S�1�2�2� ���3�� ��s& �16 � A�A�
A �A�A � A z
/stream/startzCCTV Stream Management)�tags�requestc � �F K � t r@t j r0t j � t rt j d�� t j t | j | j fd�� at j � dd| j � d�iS �w)z�11 Starts the video tracking background process for a given RTSP/HTTP URL.12 If a stream is already running, stops it and starts the new one.13 g @)�timeoutT)�target�args�daemon�messagez"Successfully launched tracker for z in background) r �14is_running�stopr
�join� threading�Threadr* r r �start)r, s r � start_streamr9 * s� � �� � �.�3�3���� � � � � �� � ,� �%�%�(��k�k�7�-�-�
.���N�15 ���� �;�G�K�K�=��W� X�X�s �B B!z /stream/stopc � �n K � t r(t j rt j � ddiS ddiS �w)z>16 Stops the currently running video processing stream.17 r2 z Active stream tracking stopped.z No stream currently tracking.)r r3 r4 r r r � stop_streamr; B s7 � �� � �.�3�3���� ��<�=�=� �6� 7�7�s �35z/stream/status� Analytics)�response_modelr+ c � �B K � t r�t j rzt j } t dt t j18 � | j
dd� | j
dd� | j
dd� | j
dd� z �� S t d�� S �w)zc19 Retrieves the actual live counting metrics from the actively running CCTV stream tracker.20 �running�personr �vehicle)r r r r r �idle)r )r r3 �countsr r r$ �get)rC s r �get_stream_statusrE M s� � �� � �.�3�3� �&�&��$� � �n� 7� 7�8��J�J�x��+��Z�Z� �1�-��*�*�X�q�)�F�J�J�y�!�,D�D� 21� 22� !�� /�/�s �B B z/forecast/peaksc � �T K � t dd�� } | j �[| j � sKt j j d� r | j
� }|st dd�� �t dd�� �| j � }|�
t dd �� �|j d23� � S �w) zp24 Retrieves the predicted weekly peak hour timings from our ML model,25 using historical traffic data.26 z data/historical_traffic_log.csvzmodels/rf_model.pkl)� data_path�27model_pathi� z4Failed to train model from existing historical data.)� status_code�detaili� zfHistorical data not found. Run generate_synthetic_data.py to collect base patterns before forecasting.z Peak prediction returned None�records)�orient)28r �model�29load_model�os�path�exists� train_modelr �predict_peak_timings�to_dict)� predictor�success�df_peakss r �get_peak_forecastrX ` s� � �� � �(I�Vk�l�I� ��� �y�';�';�'=�
�7�7�>�>�;� <� �+�+�-�G��"�s�;q�r�r� � ���
�30 �-�-�/�H�����4S�T�T� � � �9� � -�-�s �B&B(�__main__zapi:appz0.0.0.0i@ T)�host�port�reload) �fastapir r r �pydanticr r6 rO �typingr �31cv_counterr �peak_predictorr �appr r r
r7 r r r r* �postr9 r; rD rE rX r �uvicorn�runr r r �<module>rf sE �� ;� ;� � � � � %� (�
�323�[��� ,0��� �(� /�-1����)�)�*� 1�6�� 6��I� �33 �3� 34 �C� 35 � ���/�!9� :��;�Y� 2� Y� <�Y�. ���.� 8� 9��:�8� ;�8� ��� �*?�{�m��T�0� U�0�$ ��� �+� ��/�.� 0�.�4 �z����G�K�K� � ��T�B� r