Update app.py

app.py

@@ -26,19 +26,23 @@ async def text_to_speech(text, voice, rate, pitch):
         await communicate.save(tmp_path)
     return tmp_path, None
 
-# 生成SRT文件
+# 生成SRT文件，支持两行字幕
 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)  # Average duration per word
+        segment_duration = audio_duration / (len(words) // 2)  # Average duration for each two words
         current_time = 0
         
-        for i, word in enumerate(words):
+        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 ""
+            
             start_time = current_time
             end_time = start_time + segment_duration
             
             start_time_str = format_srt_time(start_time)
             end_time_str = format_srt_time(end_time)
-            srt_file.write(f"{i + 1}\n{start_time_str} --> {end_time_str}\n{word}\n\n")
+            srt_file.write(f"{i // 2 + 1}\n{start_time_str} --> {end_time_str}\n{line1} {line2}\n\n")
             
             current_time += segment_duration  # Update current time