Update app.py

app.py

@@ -92,5 +92,27 @@ demo = gr.Interface(
     description="Upload a video to classify it as real or fake."
 )
 
+from fastapi import UploadFile, File
+from fastapi.responses import JSONResponse
+import tempfile
+import os
+
+@demo.app.post("/api/predict")
+async def api_predict(file: UploadFile = File(...)):
+    """Upload + predict in one call — for ShieldSense app."""
+    if not file.filename or not any(file.filename.lower().endswith(ext) for ext in ('.mp4', '.mov', '.avi', '.webm', '.mkv', '.jpg', '.jpeg', '.png')):
+        return JSONResponse({"error": "Invalid file type. Use video or image."}, status_code=400)
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as f:
+            f.write(await file.read())
+            tmp_path = f.name
+        try:
+            result = predict_video(tmp_path)
+            return JSONResponse(result)
+        finally:
+            os.unlink(tmp_path)
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)