Divided by zero problem resolved

app.py

@@ -47,6 +47,12 @@ def generate_srt(words, audio_duration, srt_path, num_lines):
         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):
             lines = []
             for j in range(num_lines):