change download mechanism

app.py

@@ -22,7 +22,12 @@ def install_madmom():
 
 install_madmom()
 
+# Add current directory to Python path for ml_models
+sys.path.insert(0, '.')
+sys.path.insert(0, './ml_models')
 
+# Import inference - any error will bubble up
+from inference import inference
 
 def download_data_from_hub():
     base_dir = Path(".")
@@ -48,10 +53,14 @@ def download_data_from_hub():
     if not all_folders_exist:
         print(f"Downloading data folders from dataset: {data_repo_id}")
         
+        # Download to a temporary directory first
+        temp_dir = base_dir / "temp_download"
+        temp_dir.mkdir(exist_ok=True)
+        
         downloaded_path = snapshot_download(
             repo_id=data_repo_id,
             repo_type="dataset",
-            local_dir=str(base_dir),
+            local_dir=str(temp_dir),
             local_dir_use_symlinks=False,
             token=token,
             ignore_patterns=["*.md", "*.txt", ".gitattributes", "README.md"]
@@ -59,15 +68,33 @@ def download_data_from_hub():
         
         print("Dataset downloaded successfully")
         
+        # Move folders from temp_download to current directory
+        import shutil
+        
         for folder_name in folders_to_check:
-            folder_path = base_dir / folder_name
-            if folder_path.exists():
-                file_count = len([f for f in folder_path.rglob("*") if f.is_file()])
+            temp_folder_path = temp_dir / folder_name
+            target_folder_path = base_dir / folder_name
+            
+            if temp_folder_path.exists():
+                if target_folder_path.exists():
+                    shutil.rmtree(target_folder_path)
+                shutil.move(str(temp_folder_path), str(target_folder_path))
+                print(f"Moved {folder_name} to current directory")
+                
+                file_count = len([f for f in target_folder_path.rglob("*") if f.is_file()])
                 print(f"{folder_name}: {file_count:,} files")
-                downloaded_folders[folder_name] = str(folder_path)
+                downloaded_folders[folder_name] = str(target_folder_path)
             else:
-                print(f"{folder_name} folder not found in downloaded data")
+                print(f"{folder_name} not found in downloaded data")
+                # Check what's actually in temp directory
+                print(f"Contents of temp directory:")
+                for item in temp_dir.iterdir():
+                    print(f"  {item.name}")
                 downloaded_folders[folder_name] = None
+        
+        # Clean up temp directory
+        shutil.rmtree(temp_dir)
+        
     else:
         print("Data folders already exist locally")
         for folder_name in folders_to_check:
@@ -329,8 +356,7 @@ h1 {
 print("Starting Music Plagiarism Detection App...")
 print("Downloading covers80 and ml_models folders...")
 folders = download_data_from_hub()
-# Import inference - any error will bubble up
-from inference import inference
+
 if folders.get("covers80") and folders.get("ml_models"):
     print("All required data folders are ready")
 elif folders.get("covers80") or folders.get("ml_models"):