Update app.py

app.py

@@ -281,41 +281,40 @@ class FileManager:
         except Exception:
             pass  # Ignore if directory isn't empty or can't be removed
 
-def create_download_link(file_path, label="Download"):
-    """Create a styled download link for a file"""
+
+    def create_download_link(file_path, label="Download"):
     if not file_path:
         return None
-
     filename = os.path.basename(file_path)
     return f"""
-    <div class="download-btn" style="margin: 10px 0;">
-        <a href="file={file_path}"
-           download="{filename}"
-           target="_blank"
-           rel="noopener noreferrer"
-           style="display: flex; align-items: center; justify-content: center; padding: 12px 24px; background: linear-gradient(135deg, #4776E6, #8E54E9); color: white; text-decoration: none; border-radius: 8px; font-weight: 600; transition: all 0.3s ease; text-align: center; width: 100%; gap: 8px;"
-           onmouseover="this.style.transform='translateY(-2px)'; this.style.boxShadow='0 5px 15px rgba(71, 118, 230, 0.3)';"
-           onmouseout="this.style.transform='translateY(0)'; this.style.boxShadow='none';"
-           onclick="event.preventDefault(); fetch(this.href).then(resp => resp.blob()).then(blob => {{
-               const url = window.URL.createObjectURL(blob);
-               const a = document.createElement('a');
-               a.style.display = 'none';
-               a.href = url;
-               a.download = '{filename}';
-               document.body.appendChild(a);
-               a.click();
-               window.URL.revokeObjectURL(url);
-               document.body.removeChild(a);
-           }});">
-           <span style="font-size: 20px;">⬇️</span>
-           <span>{label}</span>
-        </a>
-    </div>
+    <a href="file={file_path}"
+       download="{filename}"
+       target="_blank"
+       rel="noopener noreferrer"
+       style="display: inline-block; padding: 10px 20px; background: linear-gradient(135deg, #4776E6, #8E54E9); color: white; text-decoration: none; border-radius: 8px; font-weight: 600; transition: all 0.3s ease; text-align: center; width: 100%; margin: 5px 0;"
+       onmouseover="this.style.transform='translateY(-2px)'; this.style.boxShadow='0 5px 15px rgba(71, 118, 230, 0.3)';"
+       onmouseout="this.style.transform='translateY(0)'; this.style.boxShadow='none';"
+       onclick="event.preventDefault(); fetch(this.href).then(resp => resp.blob()).then(blob => {{
+           const url = window.URL.createObjectURL(blob);
+           const a = document.createElement('a');
+           a.style.display = 'none';
+           a.href = url;
+           a.download = '{filename}';
+           document.body.appendChild(a);
+           a.click();
+           window.URL.revokeObjectURL(url);
+           document.body.removeChild(a);
+       }});">
+        ⬇️ {label}
+    </a>   
+    
     """
 
 # Create global file manager
 file_manager = FileManager()
 
+
+
 # IMPROVEMENT 3: Parallel Processing for Segments
 async def generate_accurate_srt(
     text: str, 
@@ -462,7 +461,7 @@ async def process_text_with_progress(
 ):
     # Input validation
     if not text or text.strip() == "":
-        return None, None, None, "Please enter some text to convert to speech."
+        return None, None, None, True, "Please enter some text to convert to speech."
     
     # Format pitch and rate strings
     pitch_str = f"{pitch:+d}Hz" if pitch != 0 else "+0Hz"
@@ -486,28 +485,14 @@ async def process_text_with_progress(
             parallel=parallel_processing
         )
         
-        # Create download links with descriptive labels
-        srt_download_link = create_download_link(srt_path, "Download Subtitles (.srt)")
-        audio_download_link = create_download_link(audio_path, "Download Audio (.mp3)")
-        
-        # Return exactly 4 values in the expected order:
-        # 1. Audio preview path
-        # 2. SRT download button HTML
-        # 3. Audio download button HTML
-        # 4. Status message
-        return (
-            audio_path,                     # Audio preview
-            srt_download_link,              # SRT download button
-            audio_download_link,            # Audio download button
-            "Generation completed successfully!"  # Status message
-        )
-    
+        # If successful, return results and hide error
+        return srt_path, audio_path, audio_path, False, ""
     except TTSError as e:
-        # Return error state with proper number of outputs
-        return None, None, None, f"TTS Error: {str(e)}"
+        # Return specific TTS error
+        return None, None, None, True, f"TTS Error: {str(e)}"
     except Exception as e:
-        # Return error state with proper number of outputs
-        return None, None, None, f"Unexpected error: {str(e)}"
+        # Return any other error
+        return None, None, None, True, f"Unexpected error: {str(e)}"
 
 # Voice options dictionary
 voice_options = {
@@ -557,41 +542,33 @@ import atexit
 atexit.register(file_manager.cleanup_all)
 
 # Create Gradio interface
-with gr.Blocks(title="Advanced TTS with Configurable SRT Generation", analytics_enabled=False) as app:
+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...",
-                elem_id="text_input"
-            )
+            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",
-                elem_id="voice_dropdown"
+                value="Jenny Female"
             )
             pitch_slider = gr.Slider(
                 label="Pitch Adjustment (Hz)", 
                 minimum=-10, 
                 maximum=10, 
                 value=0, 
-                step=1,
-                elem_id="pitch_slider"
+                step=1
             )
             rate_slider = gr.Slider(
                 label="Rate Adjustment (%)", 
                 minimum=-25, 
                 maximum=25, 
                 value=0, 
-                step=1,
-                elem_id="rate_slider"
+                step=1
             )
     
     with gr.Row():
@@ -602,8 +579,7 @@ with gr.Blocks(title="Advanced TTS with Configurable SRT Generation", analytics_
                 maximum=12, 
                 value=6, 
                 step=1,
-                info="Controls how many words appear on each line of the subtitle",
-                elem_id="words_per_line"
+                info="Controls how many words appear on each line of the subtitle"
             )
         with gr.Column():
             lines_per_segment = gr.Slider(
@@ -612,34 +588,30 @@ with gr.Blocks(title="Advanced TTS with Configurable SRT Generation", analytics_
                 maximum=4, 
                 value=2, 
                 step=1,
-                info="Controls how many lines appear in each subtitle segment",
-                elem_id="lines_per_segment"
+                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)",
-                elem_id="parallel_processing"
+                info="Process multiple segments simultaneously for faster conversion (recommended for longer texts)"
             )
     
-    submit_btn = gr.Button("Generate Audio & Subtitles", elem_id="submit_btn")
-      # Output components
-    with gr.Column(variant="panel") as output_col:
-        gr.Markdown("### Generated Output")
-        # Main output containers
-        with gr.Row():
-            audio_preview = gr.Audio(label="Preview Audio", elem_id="audio_preview")
-            
-        with gr.Row():
-            with gr.Column(scale=1):
-                srt_download = gr.HTML(label="", elem_id="srt_download")
-            with gr.Column(scale=1):
-                audio_download = gr.HTML(label="", elem_id="audio_download")
-        
-        error_output = gr.Textbox(label="Status", elem_id="error_output")
+    submit_btn = gr.Button("Generate Audio & Subtitles")
     
-    # Handle button click
+    # 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")
+            srt_download = gr.HTML(elem_classes="download-btn")
+            audio_download = gr.HTML(elem_classes="download-btn")
+    
+    # Handle button click with manual error handling instead of .catch()
     submit_btn.click(
         fn=process_text_with_progress,
         inputs=[
@@ -652,11 +624,15 @@ with gr.Blocks(title="Advanced TTS with Configurable SRT Generation", analytics_
             parallel_processing
         ],
         outputs=[
-            audio_preview,  # Audio preview
-            srt_download,   # SRT download link
-            audio_download, # Audio download link
-            error_output   # Status/error message
-        ]
+            srt_file,
+            audio_file,
+            audio_output,
+            error_output,
+            error_output,
+            srt_download, 
+            audio_download
+        ],
+        api_name="generate"
     )
 
 if __name__ == "__main__":