Update app.py

app.py

@@ -28,24 +28,25 @@ async def text_to_speech(text, voice, rate, pitch):
 
 # Generate SRT file based on user preferences
 def generate_srt(words, audio_duration, srt_path, words_per_line, lines_per_paragraph, speech_rate):
-    segment_duration = audio_duration / (len(words) / lines_per_paragraph)  # Average duration for each segment
+    total_words = len(words)
+    segment_duration = audio_duration / (total_words / words_per_line)  # Average duration for each segment
     adjusted_duration = segment_duration * (60 / (100 + speech_rate))  # Adjust duration based on speech rate
     current_time = 0
 
     with open(srt_path, 'w', encoding='utf-8') as srt_file:
-        for i in range(0, len(words), words_per_line * lines_per_paragraph):  # Every segment according to specified preferences
+        for i in range(0, total_words, words_per_line):  # Increment by words per line
             segment_words = words[i:i + (words_per_line * lines_per_paragraph)]
             lines = [segment_words[j:j + words_per_line] for j in range(0, len(segment_words), words_per_line)]
             lines = [' '.join(line) for line in lines]
-            
+
             start_time = current_time
-            end_time = start_time + adjusted_duration
+            end_time = min(start_time + adjusted_duration, audio_duration)  # Ensure it doesn't exceed audio duration
             
             start_time_str = format_srt_time(start_time)
             end_time_str = format_srt_time(end_time)
-            srt_file.write(f"{i // (words_per_line * lines_per_paragraph) + 1}\n{start_time_str} --> {end_time_str}\n")
+            srt_file.write(f"{i // words_per_line + 1}\n{start_time_str} --> {end_time_str}\n")
             srt_file.write('\n'.join(lines) + '\n\n')
-            
+
             current_time += adjusted_duration  # Update current time for the next segment
 
     return srt_path