Juartaurus/poi_Engineering
0
1U
2 ~
eV � @ s� d dl Z e j�d� d dlZd dlZd dlZd dlmZmZm Z m3Z4 d dl m Z d dl
mZ d dlmZ d dlmZ e� Ze� Zejded d5�d d � ed d6�Zejd
e d�e7d�dd��Ze�d�ed�fe d�dd��Zedkr�ej dddd� dS ) � Nz./)�FastAPI�File�8UploadFile�Request)� HTMLResponse)�Jinja2Templates)� StaticFiles)�SignboardTrackerz/static�static)� directory)�name� templatesz/upload)�response_class��requestc � s t �dd| i�S )Nz tracker.htmlr )r
�TemplateResponser � r �9/home/kienvs/sources/poi_engineering_api/./api/tracker.py�upload s r z /uploadfile.)�filec � s | j �d�d dkr$| j| j dd�S d�| j�d�d d� �}tj�d| �r\t�d| � t� d| � t� d| d � t9d|� d| j� �d10��}t� | j |� W 5 Q R X d|� dd�| j�d�d d� �� d | j�d�d � �}d }t
�d|� d| j� �||d| d �}d
|iS d S )N�/r �videozNeed a video file instead.)� file_name� file_type�warning�.�����z./static/videos/z/frames�wbz_traked.�mp4v�results)� content_type�split�filename�join�os�path�exists�shutil�rmtree�mkdir�open� copyfileobjr �signboardTracker�inference_signboard)r �
output_folder�buffer�output�
output_formatr r r r �create_upload_file s �8&r2 �__main__zapi.tracker:appTi@ )�reload�port) �sysr% �appendr$ �uvicornr' �fastapir r r r �fastapi.responsesr �fastapi.templatingr �fastapi.staticfilesr �src.tracker.signboard_trackr r, �app�mountr
�getr �postr2 �__name__�runr r r r �<module> s& 11