Update app.py

app.py

@@ -23,21 +23,17 @@ def format_time(seconds):
 # Function to split text into segments by punctuation or limit to 7-8 words
 def split_text_into_segments(text):
     segments = []
-    # Split by punctuation (., !, ?)
     raw_segments = re.split(r'([.!?])', text)
     for i in range(0, len(raw_segments) - 1, 2):
-        # Combine segment with following punctuation
         sentence = raw_segments[i].strip() + raw_segments[i + 1]
         words = sentence.split()
         
-        # If segment is longer than 8 words, split into 7-8 word chunks
         if len(words) > 8:
             for j in range(0, len(words), 8):
                 segments.append(" ".join(words[j:j + 8]))
         else:
             segments.append(sentence.strip())
     
-    # Handle remaining text after the last punctuation
     if len(raw_segments) % 2 == 1:
         remaining_text = raw_segments[-1].strip()
         words = remaining_text.split()
@@ -46,11 +42,11 @@ def split_text_into_segments(text):
     
     return segments
 
-# Function to generate SRT with accurate timing per batch and cross-check timing
+# Function to generate SRT with accurate timing per batch
 async def generate_accurate_srt(batch_text, batch_num, start_offset, pitch, rate, voice):
     audio_file = f"batch_{batch_num}_audio.wav"
     
-    # Generate the audio using edge-tts with pitch and rate adjustment
+    # Generate the audio using edge-tts
     tts = edge_tts.Communicate(batch_text, voice, rate=rate, pitch=f"{pitch}Hz")
     await tts.save(audio_file)
 
@@ -67,7 +63,6 @@ async def generate_accurate_srt(batch_text, batch_num, start_offset, pitch, rate
     for index, segment in enumerate(segments):
         end_time = start_time + segment_duration
         
-        # If end_time exceeds actual audio length of the batch, adjust it
         if end_time > start_offset + actual_length:
             end_time = start_offset + actual_length
 
@@ -75,35 +70,28 @@ async def generate_accurate_srt(batch_text, batch_num, start_offset, pitch, rate
         srt_content += f"{format_time(start_time)} --> {format_time(end_time)}\n"
         srt_content += segment + "\n\n"
         
-        # Update start time for next segment
         start_time = end_time
 
-    return srt_content, audio_file, start_time  # Return updated start time for cumulative tracking
+    return srt_content, audio_file, start_time
 
-# Batch processing function with cumulative timing, progress indicator, and final SRT validation
+# Batch processing function
 async def batch_process_srt_and_audio(script_text, pitch, rate, voice, progress=gr.Progress()):
     batches = [script_text[i:i + 500] for i in range(0, len(script_text), 500)]
     all_srt_content = ""
     combined_audio = AudioSegment.empty()
-    start_offset = 0.0  # Track cumulative time offset for SRT timing
+    start_offset = 0.0  
 
-    # Process each batch sequentially to ensure proper timing and cumulative offset tracking
     for batch_num, batch_text in enumerate(batches):
         srt_content, audio_file, end_offset = await generate_accurate_srt(batch_text, batch_num, start_offset, pitch, rate, voice)
         all_srt_content += srt_content
 
-        # Append the audio of each batch to the combined audio
         batch_audio = AudioSegment.from_file(audio_file)
         combined_audio += batch_audio
-        start_offset = end_offset  # Update the start offset for the next batch
+        start_offset = end_offset
 
-        # Clean up the individual batch audio file
         os.remove(audio_file)
-
-        # Update progress
         progress((batch_num + 1) / len(batches))
 
-    # Final cross-check: Adjust any subtitle that exceeds the total audio length
     total_audio_length = combined_audio.duration_seconds
     validated_srt_content = ""
     for line in all_srt_content.strip().splitlines():
@@ -116,15 +104,12 @@ async def batch_process_srt_and_audio(script_text, pitch, rate, voice, progress=
             line = f"{format_time(start_time)} --> {format_time(end_time)}"
         validated_srt_content += line + "\n"
 
-    # Generate unique names for the final files
     unique_id = uuid.uuid4()
-    final_audio_path = f"final_audio_{unique_id}.mp3"  # Set to MP3
+    final_audio_path = f"final_audio_{unique_id}.mp3"
     final_srt_path = f"final_subtitles_{unique_id}.srt"
 
-    # Export combined audio directly as MP3 with 320 kbps bitrate
     combined_audio.export(final_audio_path, format="mp3", bitrate="320k")
 
-    # Export validated SRT with unique names
     with open(final_srt_path, "w") as srt_file:
         srt_file.write(validated_srt_content)
 
@@ -132,12 +117,11 @@ async def batch_process_srt_and_audio(script_text, pitch, rate, voice, progress=
 
 # Gradio interface function
 async def process_script(script_text, pitch, rate, voice):
-    # Ensure rate is formatted correctly
-    formatted_rate = f"{'+' if rate > 0 else ''}{int(rate)}%"  # Format rate for edge_tts
-    srt_path, audio_path = await batch_process_srt_and_audio(script_text, pitch, formatted_rate, voice)
+    formatted_rate = f"{'+' if rate > 0 else ''}{int(rate)}%"  
+    srt_path, audio_path = await batch_process_srt_and_audio(script_text, pitch, formatted_rate, voice_options[voice])
     return srt_path, audio_path, audio_path
 
-# Gradio interface setup with pitch adjustment slider, rate adjustment slider, and voice selection
+# Gradio interface setup
 voice_options = {
     "Andrew": "en-US-AndrewNeural",
     "Jenny": "en-US-JennyNeural",
@@ -183,16 +167,17 @@ app = gr.Interface(
     fn=process_script,
     inputs=[
         gr.Textbox(label="Enter Script Text", lines=10),
-        gr.Slider(label="Pitch Adjustment (Hz)", minimum=-100, maximum=100, step=1, value=1),  # Default set to +1
-        gr.Slider(label="Rate Adjustment (%)", minimum=-100, maximum=100, step=1, value=1),  # Default set to +1
-        gr.Dropdown(label="Select Voice", choices=list(voice_options.keys()), value="Andrew")
+        gr.Slider(label="Pitch Adjustment (Hz)", minimum=-100, maximum=100, step=1, value=1),
+        gr.Slider(label="Rate Adjustment (%)", minimum=-100, maximum=100, step=1, value=1),
+        gr.Dropdown(label="Select Voice", choices=list(voice_options.keys()), value="Andrew"),
+        gr.Button(label="Submit")  # Add submit button
     ],
     outputs=[
         gr.File(label="Download SRT"),
         gr.File(label="Download Audio"),
         gr.Audio(label="Audio Playback")
     ],
-    live=True
+    live=False  # Changed to False to enable the Submit button functionality
 )
 
 app.launch()