Update app.py

app.py

@@ -77,17 +77,26 @@ if os.path.exists("/content/drive/MyDrive/combined"):
     DOG_AUDIO_BASE_PATH = '/content/drive/MyDrive/combined'
     HUMAN_AUDIO_BASE_PATH = '/content/drive/MyDrive/human'
     print("Using Google Drive audio paths")
+elif os.path.exists("combined") and os.path.exists("human"):
+    # Hugging Face Spaces with audio in repo root
+    DOG_AUDIO_BASE_PATH = 'combined'
+    HUMAN_AUDIO_BASE_PATH = 'human'
+    print("Using Hugging Face Spaces audio paths (repo root)")
 elif os.path.exists("audio/combined"):
-    # Hugging Face Spaces with audio in repo
+    # Alternative Hugging Face Spaces location
     DOG_AUDIO_BASE_PATH = 'audio/combined'
     HUMAN_AUDIO_BASE_PATH = 'audio/human'
-    print("Using Hugging Face Spaces audio paths")
+    print("Using Hugging Face Spaces audio paths (audio subdir)")
 else:
     # Fallback to local dummy paths
     DOG_AUDIO_BASE_PATH = DOG_DIR
     HUMAN_AUDIO_BASE_PATH = HUMAN_DIR
     print("Using local dummy audio paths")
 
+print(f"Audio base paths configured:")
+print(f"- Dog audio base: {DOG_AUDIO_BASE_PATH}")
+print(f"- Human audio base: {HUMAN_AUDIO_BASE_PATH}")
+
 
 # ---------------------------------------------------------------
 # Cross-Species Analysis Functions
@@ -291,32 +300,42 @@ def resolve_audio_path(row: pd.Series) -> str:
     source = row.get("source", "")
     label = row.get("label", "")
 
+    # Debug information
+    print(f"🔍 Resolving audio path for: {basename} ({source}, {label})")
+
     # For "Dog" data, the structure is: combined/{label}/{filename}
     if source == "Dog":
+        # Try with label subdirectory first
         expected_path = os.path.join(DOG_AUDIO_BASE_PATH, label, basename)
+        print(f"  Trying dog path: {expected_path} (exists: {os.path.exists(expected_path)})")
         if os.path.exists(expected_path):
             return expected_path
         
         # Try without subdirectory in case files are flat
         expected_path = os.path.join(DOG_AUDIO_BASE_PATH, basename)
+        print(f"  Trying flat dog path: {expected_path} (exists: {os.path.exists(expected_path)})")
         if os.path.exists(expected_path):
             return expected_path
 
     # For "Human" data, search within all "Actor_XX" subfolders
     elif source == "Human":
         if os.path.isdir(HUMAN_AUDIO_BASE_PATH):
+            print(f"  Searching in human base: {HUMAN_AUDIO_BASE_PATH}")
             for actor_folder in os.listdir(HUMAN_AUDIO_BASE_PATH):
                 if actor_folder.startswith("Actor_"):
                     expected_path = os.path.join(HUMAN_AUDIO_BASE_PATH, actor_folder, basename)
+                    print(f"    Trying: {expected_path} (exists: {os.path.exists(expected_path)})")
                     if os.path.exists(expected_path):
                         return expected_path
         
         # Try without subdirectory in case files are flat
         expected_path = os.path.join(HUMAN_AUDIO_BASE_PATH, basename)
+        print(f"  Trying flat human path: {expected_path} (exists: {os.path.exists(expected_path)})")
         if os.path.exists(expected_path):
             return expected_path
     
     # Fallback for dummy data or other cases
+    print(f"  Trying basename directly: {basename} (exists: {os.path.exists(basename)})")
     if os.path.exists(basename):
         return basename
     
@@ -324,14 +343,17 @@ def resolve_audio_path(row: pd.Series) -> str:
     if source == "Dog":
         for label_dir in ["bark", "growl", "whine", "pant"]:
             local_path = os.path.join(DOG_DIR, label_dir, basename)
+            print(f"  Trying local dog path: {local_path} (exists: {os.path.exists(local_path)})")
             if os.path.exists(local_path):
                 return local_path
     elif source == "Human":
         local_path = os.path.join(HUMAN_DIR, "Actor_01", basename)
+        print(f"  Trying local human path: {local_path} (exists: {os.path.exists(local_path)})")
         if os.path.exists(local_path):
             return local_path
     
     # If all else fails, return the original basename
+    print(f"  ❌ No valid path found, returning basename: {basename}")
     return basename
 
 def get_cmt_data(filepath: str, lens: str):
@@ -786,7 +808,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) a
                 
                 # Audio files
                 primary_audio = primary_fp if primary_fp and os.path.exists(primary_fp) else None
-                neighbor_audio = neighbor_fp if neighbor_row and neighbor_fp and os.path.exists(neighbor_fp) else None
+                neighbor_audio = neighbor_fp if neighbor_row is not None and neighbor_fp and os.path.exists(neighbor_fp) else None
 
                 return (dual_holo_fig, dual_diag_fig, primary_info, neighbor_info, 
                         primary_audio, neighbor_audio)