Add application file

app.py

@@ -0,0 +1,74 @@
+
+import gradio as gr 
+import os
+import sys
+import subprocess
+#from moviepy.editor import VideoFileClip
+
+import whisper
+from whisper.utils import write_vtt
+
+model = whisper.load_model("medium")
+
+title = "Add Text/Caption to your YouTube Shorts - MultiLingual"
+
+def video2mp3(video_file, output_ext="mp3"):
+    filename, ext = os.path.splitext(video_file)
+    subprocess.call(["ffmpeg", "-y", "-i", video_file, f"{filename}.{output_ext}"], 
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.STDOUT)
+    return f"{filename}.{output_ext}"
+
+
+def translate(input_video):
+
+    audio_file = video2mp3(input_video)
+    
+    options = dict(beam_size=5, best_of=5, fp16 = False)
+    translate_options = dict(task="translate", **options)
+    result = model.transcribe(audio_file,**translate_options)
+
+    output_dir = ''
+    audio_path = audio_file.split(".")[0]
+
+    with open(os.path.join(output_dir, audio_path + ".vtt"), "w") as vtt:
+      write_vtt(result["segments"], file=vtt)
+
+    subtitle = audio_path + ".vtt"
+    output_video = audio_path + "_subtitled.mp4"
+
+    os.system(f"ffmpeg -i {input_video} -vf subtitles={subtitle} {output_video}")
+
+    return output_video
+
+block = gr.Blocks()
+with block:
+
+    with gr.Group():
+        with gr.Box():
+            with gr.Row().style():
+                inp_video = gr.Video(
+                    label="Input Video",
+                    type="filepath",
+                    mirror_webcam = False
+                )
+                op_video = gr.Video()
+            btn = gr.Button("Generate Subtitle Video")
+        
+        
+        
+
+
+        
+        btn.click(translate, inputs=[inp_video], outputs=[op_video])
+        gr.HTML('''
+        <h1>This app give english subtitle to your short video (less than 30sec) in english, in any major language </h1>
+        ''')
+        gr.HTML('''
+        <div class="footer">
+                    <p>Model by <a href="https://github.com/openai/whisper" style="text-decoration: underline;" target="_blank">OpenAI</a> - Gradio App by <a href="https://www.linkedin.com/in/oluwafunbiadeneye?lipi=urn%3Ali%3Apage%3Ad_flagship3_profile_view_base_contact_details%3BJhODLsMxQTeoJI7nYzDzNw%3D%3D" style="text-decoration: underline;" target="_blank">Oluwafunbi</a> inspired by <a href= href="https://twitter.com/1littlecoder" style="text-decoration: underline;" target="_blank">1littlecoder</a>
+                    </p>
+        </div>
+        ''')
+
+block.launch(debug = True)

requirements.txt

@@ -0,0 +1,2 @@
+tensorflow
+git+https://github.com/openai/whisper.git