added api calls

app/main.py

@@ -24,6 +24,24 @@ IMAGE_TYPES = {"image/jpeg", "image/jpg", "image/png", "image/webp"}
 VIDEO_TYPES = {"video/mp4", "video/quicktime", "video/webm", "video/x-matroska"}
 GIF_TYPES = {"image/gif"}
 
+_EXT_MIME = {
+    ".jpg": "image/jpeg", ".jpeg": "image/jpeg",
+    ".png": "image/png",
+    ".webp": "image/webp",
+    ".gif": "image/gif",
+    ".mp4": "video/mp4",
+    ".mov": "video/quicktime",
+    ".webm": "video/webm",
+    ".mkv": "video/x-matroska",
+}
+
+
+def _resolve_content_type(content_type: str, filename: str | None) -> str:
+    if content_type not in ("", "application/octet-stream") or not filename:
+        return content_type
+    suffix = Path(filename).suffix.lower()
+    return _EXT_MIME.get(suffix) or mimetypes.guess_type(filename)[0] or content_type
+
 HF_REPORT_REPO = os.environ.get("HF_REPORT_REPO", "ComplexDataLab/openfake-reports")
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
@@ -83,8 +101,7 @@ def _predict_with_preprocess(image: Image.Image) -> dict:
 @app.post("/api/predict")
 async def predict(file: UploadFile = File(...)):
     content_type = (file.content_type or "").lower()
-    if content_type in ("", "application/octet-stream") and file.filename:
-        content_type = mimetypes.guess_type(file.filename)[0] or content_type
+    content_type = _resolve_content_type(content_type, file.filename)
     raw = await file.read()
     size_mb = len(raw) / (1024 * 1024)
 
@@ -174,8 +191,7 @@ async def report(
     # Read the uploaded file
     raw = await file.read()
     content_type = (file.content_type or "").lower()
-    if content_type in ("", "application/octet-stream") and file.filename:
-        content_type = mimetypes.guess_type(file.filename)[0] or content_type
+    content_type = _resolve_content_type(content_type, file.filename)
     if content_type not in IMAGE_TYPES | VIDEO_TYPES | GIF_TYPES:
         raise HTTPException(415, "Unsupported file type for reporting.")