Update app.py

app.py

@@ -161,12 +161,31 @@ async def send_image_for_captioning(image_path: Path, course_name: str, server:
 
 async def download_and_extract_zip(course_name: str) -> Optional[Path]:
     """Downloads the zip file for the course and extracts its contents."""
-    zip_filename = f"{course_name}.zip"
-    repo_file = f"frames/{zip_filename}"
-    
-    print(f"[{FLOW_ID}] Downloading {repo_file} from {HF_DATASET_ID}...")
+    print(f"[{FLOW_ID}] Looking for files starting with '{course_name}' in frames/ directory...")
     
     try:
+        api = HfApi(token=HF_TOKEN)
+        
+        # List all files in the frames directory
+        repo_files = api.list_repo_files(
+            repo_id=HF_DATASET_ID,
+            repo_type="dataset"
+        )
+        
+        # Find zip files that start with the course name
+        matching_files = [
+            f for f in repo_files 
+            if f.startswith(f"frames/{course_name}") and f.endswith('.zip')
+        ]
+        
+        if not matching_files:
+            print(f"[{FLOW_ID}] No zip files found starting with '{course_name}' in frames/ directory.")
+            return None
+            
+        # Use the first matching file (you could modify this to process all matches)
+        repo_file = matching_files[0]
+        print(f"[{FLOW_ID}] Found matching file: {repo_file}")
+        
         # Use hf_hub_download to get the file path
         zip_path = hf_hub_download(
             repo_id=HF_DATASET_ID,