Update app.py

app.py

@@ -254,12 +254,11 @@ def run(audio_files, file_m, file_index):
     return result, result[0]  # Assuming result is a list of file paths
 
 
-def process_audio(audio_files, file_m, file_index):
-
-    if not isinstance(audio_files, list):
-        audio_files = [audio_files]
-        
-    result, first_audio = run(audio_files, file_m, file_index)
+def process_audio(audio_file, uploaded_files, file_m, file_index):
+    if audio_file is not None:
+        result, first_audio = run([audio_file], file_m, file_index)
+    elif uploaded_files is not None:
+        result, first_audio = run(uploaded_files, file_m, file_index)
     
     # Ensure first_audio is correctly formatted as a valid file path
     if not os.path.exists(first_audio):
@@ -268,6 +267,13 @@ def process_audio(audio_files, file_m, file_index):
     return result, gr.update(value=first_audio, visible=True)
 
 
+def switch_input(input_type):
+    if input_type == "Record Audio":
+        return gr.update(visible=True), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True)
+
+
 def model_conf():
     model_files = [f for f in os.listdir(model_dir) if f.endswith(".pth")]
     return gr.Dropdown(
@@ -307,7 +313,14 @@ def get_gui(theme):
     with gr.Blocks(theme=theme, delete_cache=(3200, 3200)) as app:
         gr.Markdown(title)
 
-        aud = audio_conf()
+
+        input_type = gr.Radio(["Record Audio", "Upload Files"], label="Select Input Method", value="Record Audio")
+    
+        audio = gr.Audio(label="Record Audio", source="microphone", type="filepath", visible=True)
+        files = gr.Files(label="Upload Audio Files", type="file", file_count="multiple", visible=False)
+        
+        input_type.change(switch_input, inputs=[input_type], outputs=[audio, files])
+
         model = model_conf()
         indx = index_conf()
         button_base = button_conf()