Update app.py

app.py

@@ -164,10 +164,94 @@ async def cancel_job(job_id: str):
 
 # Initialize Gradio interface
 with gr.Blocks() as gradio_app:
-    # ...existing gradio interface code...
+    gr.Markdown("# Advanced TTS with Configurable SRT Generation")
+    gr.Markdown("Generate perfectly synchronized audio and subtitles with natural speech patterns.")
+    
+    with gr.Row():
+        with gr.Column(scale=3):
+            text_input = gr.Textbox(label="Enter Text", lines=10, placeholder="Enter your text here...")
+        
+        with gr.Column(scale=2):
+            voice_dropdown = gr.Dropdown(
+                label="Select Voice", 
+                choices=list(voice_options.keys()), 
+                value="Jenny Female"
+            )
+            pitch_slider = gr.Slider(
+                label="Pitch Adjustment (Hz)", 
+                minimum=-10, 
+                maximum=10, 
+                value=0, 
+                step=1
+            )
+            rate_slider = gr.Slider(
+                label="Rate Adjustment (%)", 
+                minimum=-25, 
+                maximum=25, 
+                value=0, 
+                step=1
+            )
+    
+    with gr.Row():
+        with gr.Column():
+            words_per_line = gr.Slider(
+                label="Words per Line", 
+                minimum=3, 
+                maximum=12, 
+                value=6, 
+                step=1,
+                info="Controls how many words appear on each line of the subtitle"
+            )
+        with gr.Column():
+            lines_per_segment = gr.Slider(
+                label="Lines per Segment", 
+                minimum=1, 
+                maximum=4, 
+                value=2, 
+                step=1,
+                info="Controls how many lines appear in each subtitle segment"
+            )
+        with gr.Column():
+            parallel_processing = gr.Checkbox(
+                label="Enable Parallel Processing", 
+                value=True,
+                info="Process multiple segments simultaneously for faster conversion"
+            )
+    
+    submit_btn = gr.Button("Generate Audio & Subtitles")
+    error_output = gr.Textbox(label="Status", visible=False)
+    
+    with gr.Row():
+        with gr.Column():
+            audio_output = gr.Audio(label="Preview Audio")
+        with gr.Column():
+            srt_file = gr.File(label="Download SRT")
+            audio_file = gr.File(label="Download Audio")
+    
+    submit_btn.click(
+        fn=process_text_with_progress,
+        inputs=[
+            text_input,
+            pitch_slider,
+            rate_slider,
+            voice_dropdown,
+            words_per_line,
+            lines_per_segment,
+            parallel_processing
+        ],
+        outputs=[
+            srt_file,
+            audio_file,
+            audio_output,
+            error_output,
+            error_output
+        ],
+        api_name="generate"
+    )
 
-# Mount Gradio app
+# Mount Gradio app to FastAPI
 app = gr.mount_gradio_app(app, gradio_app, path="/")
 
+# Start the FastAPI server
 if __name__ == "__main__":
     uvicorn.run("fastapi_app:app", host="0.0.0.0", port=8000, reload=True)