Update app.py

app.py

@@ -79,9 +79,16 @@ def extract_key_frames(video_file: str, key_frames_response: str) -> list:
         print("Error: Could not open video file.")
         return extracted_frames
     
+    # Strip Markdown code block if present
+    cleaned_response = key_frames_response.strip()
+    if cleaned_response.startswith("```json") and cleaned_response.endswith("```"):
+        cleaned_response = cleaned_response[7:-3].strip()  # Remove ```json and ```
+    elif cleaned_response.startswith("```") and cleaned_response.endswith("```"):
+        cleaned_response = cleaned_response[3:-3].strip()  # Remove generic ```
+    
     try:
         # Try parsing as JSON
-        key_frames = json.loads(key_frames_response)
+        key_frames = json.loads(cleaned_response)
         if not isinstance(key_frames, list):
             raise ValueError("Response is not a list.")
     except json.JSONDecodeError as e: