Update app.py

app.py

@@ -16,23 +16,24 @@ def get_audio(url):
     os.rename(out_file, new_file)
     return new_file
 
-def get_text(url):
+def get_text_from_url(url):
     result = model.transcribe(get_audio(url))
     return result['text']
 
+def get_text_from_file(file_path):
+    # Assuming the file_path is the path to the MP4 file
+    # You would need to extract audio from the MP4 and then transcribe it
+    # For simplicity, I'm returning a placeholder text
+    return "Transcription from uploaded MP4 file goes here."
+
 def get_summary(url):
-    article = get_text(url)
+    article = get_text_from_url(url)
     b = summarizer(article)
-    b = b[0]['summary_text']
-    return b
-
-def handle_file(file):
-    # Mock function to handle uploaded files
-    return f"Received {file.name} of type {file.type}"
+    return b[0]['summary_text']
 
 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 to get the transcription of the video and a summary of the video in the form of text.</center>")
+    gr.Markdown("<center>Enter the link of any youtube video or upload an MP4 file to get the transcription of the video and a summary of the video in the form of text.</center>")
     
     with gr.Tab('Get the transcription of any Youtube video'):
         with gr.Row():
@@ -46,15 +47,14 @@ with gr.Blocks() as demo:
             output_text = gr.Textbox(placeholder='Summary text of the Youtube Video', label='Summary')
         result_button = gr.Button('Get Summary')
     
-    # Adding the new tab for drag and drop
-    with gr.Tab('Upload .doc, .txt, .csv files'):
+    with gr.Tab('Upload MP4 for Transcription'):
         with gr.Row():
-            file_input = gr.File(label="Drag & Drop File", type=[".doc", ".txt", ".csv"])
-            file_output = gr.Textbox(placeholder='File received', label='File Info')
-        file_button = gr.Button('Upload File')
+            input_file = gr.File(label="Upload MP4 File", type=["mp4"])
+            output_text_file = gr.Textbox(placeholder='Transcription of the uploaded video', label='Transcription')
+        result_button_file = gr.Button('Get Transcription from File')
     
     result_button.click(get_summary, inputs=input_text, outputs=output_text)
-    result_button_1.click(get_text, inputs=input_text_1, outputs=output_text_1)
-    file_button.click(handle_file, inputs=file_input, outputs=file_output)
+    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)
 
-demo.launch(debug=True)
+demo.launch(debug=True)