Update app.py

divison by zero update function

app.py

@@ -29,27 +29,30 @@ async def text_to_speech(text, voice, rate, pitch):
 # Generate SRT file with specified lines of subtitles
 def generate_srt(words, audio_duration, srt_path, num_lines):
     with open(srt_path, 'w', encoding='utf-8') as srt_file:
-        segment_duration = audio_duration / (len(words) // (5 * num_lines))  # Average duration for each segment
-        current_time = 0
+        divisor = len(words) // (5 * num_lines)
+        if divisor == 0:
+            segment_duration = audio_duration  # Use full duration as fallback
+        else:
+            segment_duration = audio_duration / divisor  # Calculate duration per segment
         
-        for i in range(0, len(words), 5 * num_lines):  # Adjusting for the number of lines
+        current_time = 0
+        for i in range(0, len(words), 5 * num_lines):
             lines = []
             for j in range(num_lines):
-                line = ' '.join(words[i + j * 5:i + (j + 1) * 5])  # 5 words per line
+                line = ' '.join(words[i + j * 5:i + (j + 1) * 5])
                 if line:
                     lines.append(line)
-                    
-            start_time = current_time
-            end_time = start_time + segment_duration  # Adjust duration for the current segment
             
+            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 // (5 * num_lines) + 1}\n{start_time_str} --> {end_time_str}\n" + "\n".join(lines) + "\n\n")
-            
-            current_time += segment_duration  # Update current time for the next segment
+            current_time += segment_duration
 
     return srt_path
 
+
 def format_srt_time(seconds):
     millis = int((seconds - int(seconds)) * 1000)
     seconds = int(seconds)