Create app.py

app.py

@@ -0,0 +1,44 @@
+import whisper
+model = whisper.load_model("base")
+
+import gradio as gr
+from moviepy.editor import VideoFileClip
+import os
+
+def generate_subtitles(video_path):
+    # Extract audio from video
+    video = VideoFileClip(video_path)
+    audio_path = "temp_audio.wav"
+    video.audio.write_audiofile(audio_path)
+    
+    # Transcribe audio using Whisper
+    result = model.transcribe(audio_path)
+    
+    # Clean up temporary audio file
+    os.remove(audio_path)
+    
+    # Format the subtitles in .srt format
+    srt_content = ""
+    for index, segment in enumerate(result['segments']):
+        start_time = segment['start']
+        end_time = segment['end']
+        text = segment['text']
+        
+        # Convert seconds to SRT time format (HH:MM:SS,mmm)
+        def format_time(seconds):
+            hours = int(seconds // 3600)
+            minutes = int((seconds % 3600) // 60)
+            seconds = seconds % 60
+            milliseconds = int((seconds - int(seconds)) * 1000)
+            return f"{hours:02}:{minutes:02}:{int(seconds):02},{milliseconds:03}"
+        
+        srt_content += f"{index + 1}\n"
+        srt_content += f"{format_time(start_time)} --> {format_time(end_time)}\n"
+        srt_content += f"{text}\n\n"
+    
+    # Save the .srt file
+    srt_filename = "subtitles.srt"
+    with open(srt_filename, "w", encoding="utf-8") as srt_file:
+        srt_file.write(srt_content)
+    
+    return srt_content, srt_filename