Update app.py

app.py

@@ -11,17 +11,6 @@ import gradio as gr
 # Core pipeline function
 # -----------------------
 def analyze_pushup_video(video_path: str, save_annotated: bool = True, annotated_out_path: str | None = None):
-    """
-    Returns:
-    {
-      "ok": bool,
-      "error": str | None,
-      "rep_count": int,
-      "avg_rep_prob": float,
-      "rep_events": list[dict],
-      "annotated_video_path": str | None
-    }
-    """
     if not os.path.exists(video_path):
         return {
             "ok": False,
@@ -243,32 +232,33 @@ def analyze_pushup_video(video_path: str, save_annotated: bool = True, annotated
 
 
 # -----------------------
-# API function (Space endpoint)
+# API function
 # -----------------------
 def api_analyze(uploaded_file):
-    """
-    uploaded_file is a dict-like object from gr.File in many gradio versions:
-    - sometimes it is {"path": "..."}
-    - sometimes it is an object with .name
-    We'll support both.
-    """
     workdir = tempfile.mkdtemp()
     in_path = os.path.join(workdir, "input.mp4")
 
-    # Resolve input path
+    if uploaded_file is None:
+        return {"ok": False, "error": "No file received.", "rep_count": 0, "avg_rep_prob": 0.0, "rep_events": []}, None
+
+    # Gradio can pass:
+    # - dict with "path"
+    # - tempfile.NamedTemporaryFile-like with .name
+    # - string path
     if isinstance(uploaded_file, dict) and "path" in uploaded_file:
         src_path = uploaded_file["path"]
     else:
-        # gradio can pass a file object with .name
         src_path = getattr(uploaded_file, "name", None) or str(uploaded_file)
 
+    if not src_path or not os.path.exists(src_path):
+        return {"ok": False, "error": f"Upload path missing or not found: {src_path}", "rep_count": 0, "avg_rep_prob": 0.0, "rep_events": []}, None
+
     shutil.copy(src_path, in_path)
 
     out_path = os.path.join(workdir, "annotated.mp4")
     result = analyze_pushup_video(in_path, save_annotated=True, annotated_out_path=out_path)
 
     if not result["ok"]:
-        # Return placeholders so frontend won't crash
         return {"ok": False, "error": result["error"], "rep_count": 0, "avg_rep_prob": 0.0, "rep_events": []}, None
 
     summary = {
@@ -284,7 +274,8 @@ def api_analyze(uploaded_file):
 with gr.Blocks(title="Pushup API") as demo:
     gr.Markdown("# Pushup Analyzer API\nUpload a video, get rep count + confidence.\n")
 
-    video_file = gr.File(label="Upload .mp4", file_types=["video"])
+    # FIX: file_types should be extensions, not ["video"]
+    video_file = gr.File(label="Upload video", file_types=[".mp4", ".mov", ".webm", ".mkv"])
     btn = gr.Button("Analyze")
 
     out_json = gr.JSON(label="Results JSON")
@@ -294,7 +285,7 @@ with gr.Blocks(title="Pushup API") as demo:
         fn=api_analyze,
         inputs=[video_file],
         outputs=[out_json, out_video],
-        api_name="analyze",  # IMPORTANT: gives you a stable endpoint name
+        api_name="analyze",
     )
 
 if __name__ == "__main__":