change video file validation

docker-compose.yml

@@ -4,6 +4,6 @@ services:
       context: .
       dockerfile: Dockerfile
     ports:
-      - "8000:8000"
+      - "7680:7680"
     volumes:
       - .:/app

main.py

@@ -5,29 +5,31 @@ from utils.process_video import process_video
 from utils.zip_response import zip_response
 from utils.read_html import read_html
 
-from fastapi import FastAPI, UploadFile, HTTPException, Form, Depends
+from fastapi import FastAPI, UploadFile, HTTPException, Form, Depends, File
 from fastapi.responses import HTMLResponse, Response
 from fastapi.security import HTTPBasic
 from pydantic import BaseModel, field_validator
 
+# TODO: fix error with validation(possibly)
+
 app = FastAPI()
 security = HTTPBasic()
 
-class MP4Video(BaseModel):
-    video_file: UploadFile
+# class MP4Video(BaseModel):
+#     video_file: UploadFile
     
-    @property
-    def filename(self):
-        return self.video_file.filename
-    @property
-    def file(self):
-        return self.video_file.file
+#     @property
+#     def filename(self):
+#         return self.video_file.filename
+#     @property
+#     def file(self):
+#         return self.video_file.file
 
-    @field_validator('video_file')
-    def validate_video_file(cls, self):
-        if not self.filename.endswith('.mp4'):
-            raise HTTPException(status_code=500, detail='Invalid video file type. Please upload an MP4 file.')
-        return self
+#     # @field_validator('video_file')
+#     # def validate_video_file(cls, v):
+#     #     if "video/" not in v.content_type:
+#     #         raise HTTPException(status_code=500, detail='Invalid video file type. Please upload an MP4 file.')
+#     #     return v
 
 class SRTFile(BaseModel):
     srt_file: Optional[UploadFile] = None
@@ -71,7 +73,7 @@ async def get_temp_dir():
         del dir
 
 @app.post("/process_video/")
-async def process_video_api(video_file: MP4Video = Depends(),
+async def process_video_api(video_file: MP4Video = File(media_type="video"),
                             srt_file: SRTFile = Depends(),
                             task: Optional[str] = Form("transcribe"),
                             max_words_per_line: Optional[int] = Form(6),