Update app.py

app.py

@@ -4,35 +4,36 @@ import os
 import tempfile
 import subprocess
 import srt
+import datetime
 
-
-
-# Load Whisper model
+# Load Whisper model (using cache for performance)
 @st.cache_resource
 def load_model():
-    return whisper.load_model("small")  # Use "small" model for free usage
+    return whisper.load_model("small")  # Use the small model for free usage
 
-# Function to transcribe video
+# Transcribe video using Whisper
 def transcribe_video(video_path):
     model = load_model()
     result = model.transcribe(video_path)
     return result["segments"]
 
-# Function to create SRT file
+# Create an SRT file from the transcription segments
 def create_srt(subtitles, srt_path):
     subs = []
     for i, segment in enumerate(subtitles):
+        # Convert float seconds to timedelta objects
+        start_time = datetime.timedelta(seconds=segment["start"])
+        end_time = datetime.timedelta(seconds=segment["end"])
         subs.append(srt.Subtitle(
             index=i + 1,
-            start=srt.srt_timestamp_to_timedelta(segment["start"]),
-            end=srt.srt_timestamp_to_timedelta(segment["end"]),
+            start=start_time,
+            end=end_time,
             content=segment["text"]
         ))
-    
     with open(srt_path, "w", encoding="utf-8") as f:
         f.write(srt.compose(subs))
 
-# Function to burn captions into video
+# Burn subtitles into the video using FFmpeg
 def burn_captions(video_path, srt_path, output_path):
     command = [
         "ffmpeg",
@@ -43,13 +44,13 @@ def burn_captions(video_path, srt_path, output_path):
     ]
     subprocess.run(command, check=True)
 
-# Streamlit UI
+# Streamlit app UI
 st.title("🎥 AI Video Captioning App")
 
 uploaded_file = st.file_uploader("Upload a video file", type=["mp4", "mkv", "avi", "mov"])
 
 if uploaded_file:
-    # Save uploaded video
+    # Save the uploaded video to a temporary file
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video:
         temp_video.write(uploaded_file.read())
         video_path = temp_video.name