Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from moviepy.editor import VideoFileClip
 import os
 import whisper
+import srt
+from datetime import timedelta
 
 def extract_audio(video_path):
     try:
@@ -15,23 +17,19 @@ def extract_audio(video_path):
     except Exception as e:
         return str(e)
 
-def transcribe_audio_to_srt(audio_path, srt_file="out_put.srt"):
+def transcribe_audio_to_srt(audio_path, srt_file="output.srt"):
     model = whisper.load_model("base.en")
     result = model.transcribe(audio_path)
-    # Create a list to hold subtitle entries
     subtitles = []
-    start_time = 0.0
     for i, segment in enumerate(result['segments']):
         start_time = segment['start']
         end_time = segment['end']
         content = segment['text'].strip()
-        # Create a subtitle entry
         subtitle = srt.Subtitle(index=i+1,
                                 start=timedelta(seconds=start_time),
                                 end=timedelta(seconds=end_time),
                                 content=content)
         subtitles.append(subtitle)
-    # Write the subtitles to an SRT file
     with open(srt_file, 'w', encoding='utf-8') as f:
         f.write(srt.compose(subtitles))
     return srt_file
@@ -39,10 +37,13 @@ def transcribe_audio_to_srt(audio_path, srt_file="out_put.srt"):
 def process_video(video):
     video_path = video
     audio_path = extract_audio(video_path)
-    processed_audio_path = transcribe_audio_to_srt(audio_path)
-    with open(processed_audio_path, "r") as f:
-        srt_content = f.read()
-    return srt_content
+    if audio_path.endswith('.mp3'):
+        processed_audio_path = transcribe_audio_to_srt(audio_path)
+        with open(processed_audio_path, "r") as f:
+            srt_content = f.read()
+        return srt_content
+    else:
+        return "Failed to extract audio."
 
 iface = gr.Interface(
     fn=process_video,