Update app.py

app.py

@@ -77,23 +77,26 @@ def generate_summary(transcription):
 
 # Define the main function to handle transcription and summary generation
 
-def process_lecture(input_data, lesson_plan):
+def process_lecture(input_type, audio_input, pdf_input, youtube_input, lesson_plan):
     transcription = ""
-    if isinstance(input_data, tuple):  # Audio input
-        input_type, input_value = input_data
-        if input_type == "audio":
-            with gr.File(input_value) as f:
-                transcription = transcribe_audio(f)
-        elif input_type == "youtube":
-            audio_path = download_youtube_audio(input_value)
+    if input_type == "Audio File":
+        if audio_input is not None:
+            transcription = transcribe_audio(audio_input)
+    elif input_type == "YouTube URL":
+        if youtube_input:
+            audio_path = download_youtube_audio(youtube_input)
             with open(audio_path, "rb") as f:
                 transcription = transcribe_audio(f)
-        elif input_type == "pdf":
-            transcription = extract_text_from_pdf(input_value)
+    elif input_type == "PDF Document":
+        if pdf_input is not None:
+            transcription = extract_text_from_pdf(pdf_input)
     
-    transcription_text = "\n".join([f"{segment['start']:.2f}-{segment['end']:.2f}: {segment['text']}" for segment in transcription['segments']])
-    summary = generate_summary(transcription)
-    return transcription_text, summary
+    if transcription:
+        transcription_text = "\n".join([f"{segment['start']:.2f}-{segment['end']:.2f}: {segment['text']}" for segment in transcription['segments']])
+        summary = generate_summary(transcription)
+        return transcription_text, summary
+    else:
+        return "No transcription available.", "No summary available."
 
 # Set up Gradio interface
 with gr.Blocks() as demo:
@@ -117,6 +120,7 @@ with gr.Blocks() as demo:
             transcription_output = gr.Textbox(label="Lecture Transcription with Timestamps", interactive=False)
             summary_output = gr.Textbox(label="Summarized Lecture Notes", interactive=False)
 
-    submit_btn.click(fn=process_lecture, inputs=[(input_type, [audio_input, pdf_input, youtube_input]), lesson_plan_input], outputs=[transcription_output, summary_output])
+    submit_btn.click(fn=process_lecture, inputs=[input_type, audio_input, pdf_input, youtube_input, lesson_plan_input], outputs=[transcription_output, summary_output])
 
-# Launch the interface demo.launch(share=True)
+# Launch the interface
+demo.launch(share=True)