Update demo.py

demo.py

@@ -140,9 +140,23 @@ with gr.Blocks(title="🔊", theme=gr.themes.Base(primary_hue="blue", neutral_hu
                         model_choices, index_choices = change_choices()
                         
                         audio_paths = get_audio_paths('audios')
-                        default_audio = audio_paths[0] if audio_paths else ""
-                        default_model = model_choices[0] if model_choices else ""
-                        default_index = index_choices[0] if index_choices else ""
+
+                        # Helper to safely get the first item from list or dict
+                        # Fixes KeyError: 0 when change_choices returns a dictionary
+                        def safe_first(data):
+                            if not data: return ""
+                            if isinstance(data, dict):
+                                # If it's a dict, Gradio uses values as the actual data/paths
+                                return next(iter(data.values()))
+                            try:
+                                # Assume it's a list or sequence
+                                return data[0]
+                            except (IndexError, KeyError):
+                                return ""
+
+                        default_audio = safe_first(audio_paths)
+                        default_model = safe_first(model_choices)
+                        default_index = safe_first(index_choices)
 
                         return (
                             gr.update(choices=model_choices, value=default_model), # voice_model