Upload app.py

app.py

@@ -121,19 +121,32 @@ def analyze_video(video_file, persona_choice, reference_video_choice):
         # Process video and get pose analysis
         video_path = video_file
         
-        # Reference video mapping
-        ref_mapping = {
-            "Squat": "demo_squat.mp4",
-            "Pushup": "demo_pushup.mp4", 
-            "Pullup": "demo_pullup.mp4"
+        # Download reference data from HF Dataset
+        from huggingface_hub import hf_hub_download
+        
+        # Map exercise to reference
+        exercise_map = {
+            "Squat": "squat",
+            "Pushup": "pushup",
+            "Pullup": "pullup"
         }
-        ref_video = ref_mapping.get(reference_video_choice, "demo_squat.mp4")
-        ref_path = PROJECT_ROOT / "reference_videos" / ref_video
+        exercise_id = exercise_map.get(reference_video_choice, "pushup")
+        
+        # Download reference keypoints from dataset
+        try:
+            ref_3d_path = hf_hub_download(
+                repo_id="rlogh/fitness-coach-references",
+                filename=f"{exercise_id}/keypoints_3D.npz",
+                repo_type="dataset"
+            )
+            ref_results = {'keypoints_3d': np.load(ref_3d_path)['reconstruction']}
+        except Exception as e:
+            print(f"Warning: Could not load reference: {e}")
+            ref_results = None
         
-        # Process videos
+        # Process user video
         print(f"Processing {video_path}...")
         user_results = process_video(video_path)
-        ref_results = process_video(str(ref_path)) if ref_path.exists() else None
         
         if not user_results or 'keypoints_3d' not in user_results:
             return "Failed to process video - no pose detected", "", ""