Update app.py

app.py

@@ -629,7 +629,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) a
             def update_file_choices(species):
                 """Update the primary file dropdown based on selected species."""
                 species_files = df_combined[df_combined["source"] == species]["filepath"].astype(str).tolist()
-                return gr.Dropdown.update(choices=species_files, value=species_files[0] if species_files else "")
+                return species_files
 
             def update_cross_species_view(species, primary_file, neighbor_file, lens, resolution, wavelength):
                 if not primary_file:
@@ -720,21 +720,34 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) a
                 neighbor_audio = neighbor_fp if neighbor_row 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, 
-                        gr.Dropdown.update(choices=neighbor_choices, value=neighbor_file if neighbor_row else ""))
+                        primary_audio, neighbor_audio)
 
             # Event handlers
+            def update_dropdowns_on_species_change(species):
+                """Update both primary and neighbor dropdowns when species changes."""
+                species_files = df_combined[df_combined["source"] == species]["filepath"].astype(str).tolist()
+                opposite_species = 'Human' if species == 'Dog' else 'Dog'
+                neighbor_files = df_combined[df_combined["source"] == opposite_species]["filepath"].astype(str).tolist()
+                
+                primary_value = species_files[0] if species_files else ""
+                neighbor_value = neighbor_files[0] if neighbor_files else ""
+                
+                return (
+                    gr.Dropdown(choices=species_files, value=primary_value),
+                    gr.Dropdown(choices=neighbor_files, value=neighbor_value)
+                )
+            
             species_dropdown.change(
-                update_file_choices,
+                update_dropdowns_on_species_change,
                 inputs=[species_dropdown],
-                outputs=[primary_dropdown]
+                outputs=[primary_dropdown, neighbor_dropdown]
             )
 
             cross_species_inputs = [species_dropdown, primary_dropdown, neighbor_dropdown, 
                                   holo_lens_dropdown, holo_resolution_slider, holo_wavelength_slider]
             cross_species_outputs = [dual_holography_plot, dual_diagnostic_plot, 
                                    primary_info_html, neighbor_info_html,
-                                   primary_audio_out, neighbor_audio_out, neighbor_dropdown]
+                                   primary_audio_out, neighbor_audio_out]
 
             for component in cross_species_inputs:
                 component.change(update_cross_species_view, 
@@ -742,7 +755,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="cyan")) a
                                outputs=cross_species_outputs)
 
             # Initialize on load
-            demo.load(lambda: update_file_choices("Dog"), outputs=[primary_dropdown])
+            demo.load(lambda: update_dropdowns_on_species_change("Dog"), 
+                     outputs=[primary_dropdown, neighbor_dropdown])
             demo.load(update_cross_species_view, 
                      inputs=cross_species_inputs, 
                      outputs=cross_species_outputs)