U od@s$ddlZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZmZddlmZmZmZe d d gd Zd Zed e ddedededfee eeeeeeedddZejdededddZ ejdededddZ!dS)N) AsyncResult)ListOptional) JSONResponse) APIRouter UploadFileFileDependsForm)analyse) VideoBase PredictionTask)generate_unique_filenameget_file_durationextract_title_and_durationz/taskszBased Content Tasks)prefixtags)z.mp4z.mkvz.aviz.movz.tsz/process)default)fileurl confidence categoriesc sz|s |s tdddddWSd\}}}|r|jr|s|jtrtj|j\}}t|}tjd|} t | d} t |j | W5QRX| }t | }d } ntd ddddWSnf|r |s |trd } |}t|\}}tj|\}} ntd ddddWSntd ddddWSz>|||||d } t| }tt|| d|ddddWWStk r}z"tt||ddddWYWSd}~XYnXWn<tk r}ztd|ddddWYSd}~XYnXdS)a Process the data from the request. Parameters: - request: ProcessDataRequest object containing the request data. - file (optional): The form content file (bytes). - url (optional): The URL (HTTP/HTTPS). - confidence (optional): The confidence value (float). - categories (optional): The list of categories (List[str]). Returns: A dict with the message "Data processed successfully". z(Either 'file' or 'url' must be provided.ERROR)messagestatusi status_code)NNNstaticwbz(File detected and processed successfullyzSInvalid file type. Only video files (.mp4, .avi, .mov) and .ts files are supported.z5Video/TS file URL detected and processed successfullyz(Either a file or a URL must be provided.)titledurationsourcerrz/api/v1/tasks/result/ PROCESSING)task_idrZ url_resultr)r$rrN)datarr)rfilenameendswith extentionsospathsplitextrjoinopenshutil copyfileobjrrrr delaystr Exception)rrrrr r!r" extensionZunique_filename file_pathfr_Z video_datar$exr9*/var/www/html/BCA_API/app/api/v1/videos.pyprocesss          "r;z/status/{task_id})response_model)r$cs"t|}tt||jddddS)Nr$rresultr)rrr2r)r$taskr9r9r:rsrz/result/{task_id}csZt|}|s*tt||jddddS|}|d}tt||d|dddS)Nr=r>r%rr?rr@)rreadyrr2rget)r$rAZ task_resultr?r9r9r:r?s$  r?)"r*r/Z celery.resultrtypingrrfastapi.responsesrfastapirrrr r Z app.tasksr Z app.schemasr r rZ app.utilsrrrrouterr)postr2floatr;rCrr?r9r9r9r:s2    y