Update app.py

app.py

@@ -89,7 +89,7 @@ def process_audio(audio_path):
 def handle_audio_input(audio_path=None, youtube_url=None):
     if youtube_url:
         audio_path = download_youtube_to_mp3(youtube_url, output_dir="downloads", filename="youtube_audio")
-    
+        return process_audio(audio_path)
     if audio_path:
         return process_audio(audio_path)
     else:
@@ -112,16 +112,20 @@ def create_gradio_interface():
             submit_btn.click(fn=on_text_submit, inputs=text_input, outputs=[lang_output, output_text, time_taken])
 
         with gr.Tab("Audio Input"):
-            audio_input = gr.Audio(label="Upload Audio", type="filepath")
-            youtube_input = gr.Textbox(label="YT URL (optional)")
-            audio_output = gr.Textbox(label="Detected Topics from Audio")
-            audio_submit_btn = gr.Button("Process Audio")
-
-            def on_audio_submit(audio, youtube_url):
-                return handle_audio_input(audio_path=audio, youtube_url=youtube_url)
-
-            audio_submit_btn.click(fn=on_audio_submit, inputs=[audio_input, youtube_input], outputs=audio_output)
-
+    audio_input = gr.Audio(label="Upload Audio", type="filepath")
+    youtube_input = gr.Textbox(label="YouTube URL")
+    audio_output = gr.Textbox(label="Detected Topics from Audio")
+    audio_submit_btn = gr.Button("Process Audio")
+
+    def on_audio_submit(audio, youtube_url):
+        if youtube_url:
+            return handle_audio_input(youtube_url=youtube_url)
+        elif audio:
+            return handle_audio_input(audio_path=audio)
+        else:
+            return "Please provide either an audio file or a YouTube URL."
+
+    audio_submit_btn.click(fn=on_audio_submit, inputs=[audio_input, youtube_input], outputs=audio_output)
     iface.launch()
 
 if __name__ == "__main__":