Update app.py

app.py

@@ -15,6 +15,7 @@ import multiprocessing
 import psutil
 import concurrent.futures
 import gc
+from gradio.themes import Monochrome
 
 class TimingManager:
     def __init__(self):
@@ -523,92 +524,189 @@ voice_options = {
 import atexit
 atexit.register(file_manager.cleanup_all)
 
-# Create Gradio interface
-with gr.Blocks(title="Advanced TTS with Configurable SRT Generation") as app:
-    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...")
+# Create custom theme
+theme = Monochrome().set(
+    primary_hue="blue",
+    secondary_hue="slate",
+    neutral_hue="zinc",
+)
+
+# Create Gradio interface with modern UI
+with gr.Blocks(
+    title="Text to Speech Studio", 
+    theme=theme,
+    css="""
+        .container { max-width: 1200px; margin: auto; }
+        .title { text-align: center; margin-bottom: 2rem; }
+        .input-group { margin-bottom: 1.5rem; }
+        .help-text { font-size: 0.9em; color: #666; }
+        .status-area { margin: 1rem 0; }
+    """
+) as app:
+    with gr.Group(elem_classes="container"):
+        gr.Markdown(
+            """
+            # 🎙️ Text to Speech Studio
+            ### Generate professional quality audio with synchronized subtitles
+            """
+        , elem_classes="title")
         
-        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 (recommended for longer texts)"
+        with gr.Tabs():
+            with gr.TabItem("📝 Text Input"):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        text_input = gr.Textbox(
+                            label="Your Text",
+                            lines=10,
+                            placeholder="Enter your text here. The AI will automatically segment it into natural phrases...",
+                            elem_classes="input-group"
+                        )
+                        gr.Markdown(
+                            "💡 **Tip:** For best results, ensure proper punctuation in your text.",
+                            elem_classes="help-text"
+                        )
+                    
+                    with gr.Column(scale=2):
+                        with gr.Group():
+                            gr.Markdown("### Voice Settings")
+                            voice_dropdown = gr.Dropdown(
+                                label="Voice",
+                                choices=list(voice_options.keys()),
+                                value="Jenny Female",
+                                elem_classes="input-group"
+                            )
+                            with gr.Row():
+                                with gr.Column():
+                                    pitch_slider = gr.Slider(
+                                        label="Pitch",
+                                        minimum=-10,
+                                        maximum=10,
+                                        value=0,
+                                        step=1,
+                                        elem_classes="input-group"
+                                    )
+                                with gr.Column():
+                                    rate_slider = gr.Slider(
+                                        label="Speed",
+                                        minimum=-25,
+                                        maximum=25,
+                                        value=0,
+                                        step=1,
+                                        elem_classes="input-group"
+                                    )
+            
+            with gr.TabItem("⚙️ Advanced Settings"):
+                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 subtitle line length",
+                            elem_classes="input-group"
+                        )
+                    with gr.Column():
+                        lines_per_segment = gr.Slider(
+                            label="Lines per Segment",
+                            minimum=1,
+                            maximum=4,
+                            value=2,
+                            step=1,
+                            info="📑 Controls subtitle block size",
+                            elem_classes="input-group"
+                        )
+                    with gr.Column():
+                        parallel_processing = gr.Checkbox(
+                            label="Parallel Processing",
+                            value=True,
+                            info="⚡ Faster processing for longer texts",
+                            elem_classes="input-group"
+                        )
+        
+        with gr.Row():
+            with gr.Column(scale=2):
+                submit_btn = gr.Button(
+                    "🎯 Generate Audio & Subtitles",
+                    variant="primary",
+                    scale=2
+                )
+            with gr.Column():
+                clear_btn = gr.Button("🔄 Clear All", variant="secondary")
+        
+        with gr.Group(elem_classes="status-area"):
+            error_output = gr.Textbox(
+                label="Status",
+                visible=False,
+                elem_classes="error-message"
             )
+            
+        with gr.Tabs():
+            with gr.TabItem("🎧 Preview"):
+                audio_output = gr.Audio(
+                    label="Generated Audio",
+                    elem_classes="preview-audio"
+                )
+            with gr.TabItem("📥 Downloads"):
+                with gr.Row():
+                    with gr.Column():
+                        srt_file = gr.File(
+                            label="📄 Subtitle File (SRT)",
+                            elem_classes="download-file"
+                        )
+                    with gr.Column():
+                        audio_file = gr.File(
+                            label="🎵 Audio File (MP3)",
+                            elem_classes="download-file"
+                        )
+
+        gr.Markdown(
+            """
+            ### 📌 Features
+            - Professional-quality text-to-speech conversion
+            - Automatic natural speech segmentation
+            - Perfectly synchronized subtitles
+            - Multiple voice options and customization
+            """,
+            elem_classes="help-text"
+        )
+
+    # Clear button functionality
+    def clear_inputs():
+        return {
+            text_input: "",
+            pitch_slider: 0,
+            rate_slider: 0,
+            voice_dropdown: "Jenny Female",
+            words_per_line: 6,
+            lines_per_segment: 2,
+            parallel_processing: True,
+            error_output: gr.update(visible=False),
+            audio_output: None,
+            srt_file: None,
+            audio_file: None
+        }
     
-    submit_btn = gr.Button("Generate Audio & Subtitles")
-    
-    # Add error message component
-    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")
-    
-    # Handle button click with manual error handling instead of .catch()
+    clear_btn.click(
+        fn=clear_inputs,
+        inputs=[],
+        outputs=[
+            text_input, pitch_slider, rate_slider, voice_dropdown,
+            words_per_line, lines_per_segment, parallel_processing,
+            error_output, audio_output, srt_file, audio_file
+        ]
+    )
+
+    # Existing button click handler
     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
+            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
+            srt_file, audio_file, audio_output, error_output, error_output
         ],
         api_name="generate"
     )