Update app.py

app.py

@@ -26,25 +26,24 @@ async def text_to_speech(text, voice, rate, pitch):
         await communicate.save(tmp_path)
     return tmp_path, None
 
-# 生成SRT文件，支持两行字幕
+# 生成SRT文件，支持每行5个单词和2行字幕
 def generate_srt(words, audio_duration, srt_path, fps=24):
     with open(srt_path, 'w', encoding='utf-8') as srt_file:
-        segment_duration = audio_duration / (len(words) // 2)  # Average duration for each two words
+        segment_duration = audio_duration / (len(words) // 5)  # Average duration for each 5 words
         current_time = 0
         
-        for i in range(0, len(words), 2):
-            # Get two words for each subtitle entry
-            line1 = words[i]
-            line2 = words[i + 1] if (i + 1) < len(words) else ""
+        for i in range(0, len(words), 10):  # Every 10 words for two lines
+            line1 = ' '.join(words[i:i + 5])  # First 5 words
+            line2 = ' '.join(words[i + 5:i + 10]) if (i + 5 < len(words)) else ""  # Next 5 words
             
             start_time = current_time
-            end_time = start_time + segment_duration
+            end_time = start_time + segment_duration * 2  # Adjust duration for two lines
             
             start_time_str = format_srt_time(start_time)
             end_time_str = format_srt_time(end_time)
-            srt_file.write(f"{i // 2 + 1}\n{start_time_str} --> {end_time_str}\n{line1} {line2}\n\n")
+            srt_file.write(f"{i // 10 + 1}\n{start_time_str} --> {end_time_str}\n{line1} {line2}\n\n")
             
-            current_time += segment_duration  # Update current time
+            current_time += segment_duration * 2  # Update current time for the next segment
 
     return srt_path