Update app.py

app.py

@@ -14,6 +14,7 @@ import pprint
 from urllib.error import HTTPError
 import subprocess
 import gradio as gr
+import traceback
 pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token="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")
 
 
@@ -105,6 +106,21 @@ def Transcribe(audio="temp_audio.wav"):
     except OSError:
         pass
     return t_text
+# subprocess.call(['ffmpeg', '-i', 'audio.mp3',
+#                    'audio.wav'])
+
+def AudioTranscribe(audio, retries=5):
+    if retries:
+        try:
+            subprocess.call(['ffmpeg', '-i', audio,'temp_audio.wav'])
+        except Exception as ex:
+            traceback.print_exc()
+            return AudioTranscribe(audio, retries-1)
+        if not (os.path.exist("temp_audio.wav")):
+            return AudioTranscribe(audio, retries-1)
+        return Transcribe()
+    else:
+        raise gr.Error("There is some issue ith Audio Transcriber. Please try again later!")
 
 def VideoTranscribe(video):
     command = f"ffmpeg -i {video} -ab 160k -ac 2 -ar 44100 -vn temp_audio.wav"
@@ -143,16 +159,38 @@ def YoutubeTranscribe(URL, retries = 5):
     else:
         raise gr.Error(f"Unable to get video from {URL}")
 
-with gr.Blocks() as i:
-    with gr.Column():
-        video = gr.Video()
-        text = gr.Textbox(label="Youtube Link", placeholder="https://www.youtube.com/watch?v=GECcjrYHH8w")
-        btn = gr.Button("Run")
-        with gr.Column():
-            output = gr.Textbox(label="Transcribed Text", lines=10)
-        if not video and not text:
-            raise gr.Error("Either input url or video (not both)")
-        else:
-            btn.click(fn=YoutubeTranscribe, inputs=text, outputs=output)
-i.launch()
+# with gr.Blocks() as i:
+#     with gr.Row():
+#         with gr.Column():
+#             with gr.Row():
+#                 video = gr.Video()
+#                 audio = gr.Audio()
+#             text = gr.Textbox(label="Youtube Link", placeholder="https://www.youtube.com/watch?v=GECcjrYHH8w")
+#             btn = gr.Button("Run")
+#         with gr.Row():
+#             output = gr.Textbox(label="Transcribed Text", lines=15)
+#         if not video and not text:
+#             raise gr.Error("Either input url or video (not both)")
+#         else:
+#             btn.click(fn=YoutubeTranscribe, inputs=text, outputs=output)
+# i.launch()
+
+ut = gr.Interface(
+    fn=YoutubeTranscribe,
+    inputs=gr.Textbox(label="Youtube Link", placeholder="https://www.youtube.com/watch?v=GECcjrYHH8w"),
+    outputs=gr.Textbox(label="Transcribed Text", lines=15)
+)
+vt = gr.Interface(
+    fn=VideoTranscribe,
+    inputs='video',
+    outputs=gr.Textbox(label="Transcribed Text", lines=15)
+)
+at = gr.Interface(
+    fn=AudioTranscribe,
+    inputs='audio',
+    outputs=gr.Textbox(label="Transcribed Text", lines=15)
+)
+
+demo = gr.TabbedInterface([ut, vt, at], ["Youtube URL", "Video", "Audio"])
+demo.launch()
 # YoutubeTranscribe('https://www.youtube.com/watch?v=GECcjrYHH8w')