app.py

app.py

@@ -35,37 +35,35 @@ def get_summary_from_file(file):
     b = summarizer(article)
     return b[0]['summary_text']
 
+def process_url(url):
+    transcription = get_text_from_url(url)
+    summary = get_summary_from_url(url)
+    return summary, transcription
+
+def process_file(file):
+    transcription = get_text_from_file(file)
+    summary = get_summary_from_file(file)
+    return summary, transcription
+
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Youtube video transcription with OpenAI's Whisper</center></h1>")
     gr.Markdown("<center>Enter the link of any youtube video or upload an MP4 file to get the transcription and a summary in text form.</center>")
     
-    with gr.Tab('Get the transcription of any Youtube video'):
-        with gr.Row():
-            input_text_1 = gr.Textbox(placeholder='Enter the Youtube video URL', label='URL')
-            output_text_1 = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
-        result_button_1 = gr.Button('Get Transcription from URL')
-        
-    with gr.Tab('Summary of Youtube video'):
+    with gr.Tab('Youtube Video'):
         with gr.Row():
             input_text = gr.Textbox(placeholder='Enter the Youtube video URL', label='URL')
-            output_text = gr.Textbox(placeholder='Summary text of the Youtube Video', label='Summary')
-        result_button = gr.Button('Get Summary from URL')
+            output_summary = gr.Textbox(placeholder='Summary text of the Youtube Video', label='Summary')
+            output_transcription = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
+        result_button = gr.Button('Process Youtube Video')
         
-    with gr.Tab('Upload MP4 for Transcription'):
+    with gr.Tab('Uploaded MP4'):
         with gr.Row():
             input_file = gr.File(label='Upload MP4')
-            output_text_file = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
-        result_button_file = gr.Button('Get Transcription from File')
-        
-    with gr.Tab('Summary from Uploaded MP4'):
-        with gr.Row():
-            input_file_summary = gr.File(label='Upload MP4')
-            output_text_file_summary = gr.Textbox(placeholder='Summary text of the video', label='Summary')
-        result_button_file_summary = gr.Button('Get Summary from File')
+            output_file_summary = gr.Textbox(placeholder='Summary text of the video', label='Summary')
+            output_file_transcription = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
+        result_button_file = gr.Button('Process Uploaded MP4')
 
-    result_button.click(get_summary_from_url, inputs=input_text, outputs=output_text)
-    result_button_1.click(get_text_from_url, inputs=input_text_1, outputs=output_text_1)
-    result_button_file.click(get_text_from_file, inputs=input_file, outputs=output_text_file)
-    result_button_file_summary.click(get_summary_from_file, inputs=input_file_summary, outputs=output_text_file_summary)
+    result_button.click(process_url, inputs=input_text, outputs=[output_summary, output_transcription])
+    result_button_file.click(process_file, inputs=input_file, outputs=[output_file_summary, output_file_transcription])
     
 demo.launch(debug=True)