Update app.py

app.py

@@ -258,22 +258,31 @@ def translate_audio(audio_file_path, prompt, model):
 # subtitles maker
 
 # helper function convert json transcription to srt
+from datetime import timedelta
+
 def create_srt_from_text(transcription_text):
-    """Converts Groq text transcription to SRT format (assuming no timestamps)."""
     srt_lines = []
-    duration = timedelta(seconds=2)  # Adjust the duration as needed
-    text_parts = transcription_text.split(".")
     start_time = timedelta(seconds=0)
 
+    # Define a function to calculate the duration based on text length
+    def calculate_duration(text):
+        words_per_minute = 150  # Average reading speed
+        words = len(text.split())
+        duration_seconds = (words / words_per_minute) * 60
+        return timedelta(seconds=duration_seconds)
+
+    text_parts = transcription_text.split(".")
+    
     for i, text_part in enumerate(text_parts):
         text_part = text_part.strip()
         if text_part:
+            duration = calculate_duration(text_part)
             end_time = start_time + duration
-            # Corrected Timestamp Formatting
-            start_timestamp = f"{start_time.seconds:02d}:{start_time.microseconds // 1000:03d}"
-            end_timestamp = f"{end_time.seconds:02d}:{end_time.microseconds // 1000:03d}"
+            start_timestamp = str(start_time).split('.')[0] + ',' + str(start_time.microseconds // 1000).zfill(3)
+            end_timestamp = str(end_time).split('.')[0] + ',' + str(end_time.microseconds // 1000).zfill(3)
             srt_lines.append(f"{i + 1}\n{start_timestamp} --> {end_timestamp}\n{text_part.strip()}\n\n")
             start_time = end_time  # Move to the next time slot
+    
     return "".join(srt_lines)
 
 
@@ -310,29 +319,15 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language,
     # Generate subtitles and add to video if input is video
     if audio_file_path.lower().endswith((".mp4", ".webm")):
         try:
-            # Use ffmpeg to add subtitles to the video
+            # Use ffmpeg to burn subtitles into the video
             output_file_path = audio_file_path.replace(os.path.splitext(audio_file_path)[1], "_with_subs" + os.path.splitext(audio_file_path)[1])
             subprocess.run(
                 [
                     "ffmpeg",
                     "-i",
                     audio_file_path,
-                    "-i",
-                    temp_srt_path,
-                    "-map",
-                    "0:v",
-                    "-map",
-                    "0:a",
-                    "-map",
-                    "1",
-                    "-c:v",
-                    "copy",
-                    "-c:a",
-                    "copy",
-                    "-c:s",
-                    "mov_text",
-                    "-metadata:s:s:0",
-                    "language=eng",
+                    "-vf",
+                    f"subtitles={temp_srt_path}",
                     output_file_path,
                 ],
                 check=True,