Update app.py

app.py

@@ -26,21 +26,25 @@ async def text_to_speech(text, voice, rate, pitch):
         await communicate.save(tmp_path)
     return tmp_path, None
 
-# Generate SRT file with 1 line of subtitles containing 10-14 words
+# Generate SRT file with 2 lines of subtitles containing 14-16 words total
 def generate_srt(words, audio_duration, srt_path):
     with open(srt_path, 'w', encoding='utf-8') as srt_file:
-        segment_duration = audio_duration / (len(words) // 14)  # Average duration for each 14 words
+        segment_duration = audio_duration / (len(words) // 16)  # Average duration for each 16 words
         current_time = 0
         
-        for i in range(0, len(words), 14):  # Every 14 words for one line
-            line = ' '.join(words[i:i + 14])  # Up to 14 words
+        for i in range(0, len(words), 16):  # Every 16 words for two lines
+            line1 = ' '.join(words[i:i + 8])  # First line with up to 8 words
+            line2 = ' '.join(words[i + 8:i + 16])  # Second line with the next 8 words
             
+            if line2.strip() == '':
+                line2 = ''  # Handle case where there aren't enough words for the second line
+
             start_time = current_time
             end_time = start_time + segment_duration  # Adjust duration for one line
             
             start_time_str = format_srt_time(start_time)
             end_time_str = format_srt_time(end_time)
-            srt_file.write(f"{i // 14 + 1}\n{start_time_str} --> {end_time_str}\n{line}\n\n")
+            srt_file.write(f"{i // 16 + 1}\n{start_time_str} --> {end_time_str}\n{line1}\n{line2}\n\n")
             
             current_time += segment_duration  # Update current time for the next segment