Update app.py

app.py

@@ -1,10 +1,10 @@
-import whisper
-model = whisper.load_model("base")
-
 import gradio as gr
 from moviepy.editor import VideoFileClip
 import os
 
+import whisper
+model = whisper.load_model("base")
+
 def generate_subtitles(video_path):
     # Extract audio from video
     video = VideoFileClip(video_path)
@@ -41,4 +41,22 @@ def generate_subtitles(video_path):
     with open(srt_filename, "w", encoding="utf-8") as srt_file:
         srt_file.write(srt_content)
     
-    return srt_content, srt_filename
+    return srt_content, srt_filename
+
+
+def video_to_subtitles(video):
+    subtitles, srt_filename = generate_subtitles(video)
+    return subtitles, srt_filename  # Just return the file path directly
+
+iface = gr.Interface(
+    fn=video_to_subtitles,
+    inputs=gr.Video(label="Upload Video"),
+    outputs=[
+        gr.Textbox(label="Subtitles", lines=10),
+        gr.DownloadButton(label="Download .srt File")  # No file_name needed
+    ],
+    title="Video to Subtitles",
+    description="Upload a video to generate subtitles using Whisper AI and download the .srt file."
+)
+
+iface.launch(share=True)