Update app.py

app.py

@@ -17,23 +17,27 @@ def run_pipeline(audio_input, remove_fillers, remove_stutters, remove_silences,
     os.makedirs(out_dir, exist_ok=True)
     
     try:
-        # 1. Clean (Supports WhatsApp via Denoiser pre-process)
+        # 1. Clean (Passes all 5 specific toggles to the API)
         res = denoiser.process(
             audio_input, out_dir, 
             fillers=remove_fillers, 
             stutters=remove_stutters,
-            long_silences=remove_silences
+            long_silences=remove_silences,
+            breath=remove_breaths,
+            mouth_sounds=remove_mouth
         )
         
         # 2. Transcribe
-        text = transcriber.transcribe(res['audio_path'])['text']
+        text_data = transcriber.transcribe(res['audio_path'])
+        text = text_data['text']
+        
         if remove_fillers:
             text = denoiser.clean_transcript_fillers(text)
             
         # 3. Translate
         translated = translator.translate(text, target_lang)
 
-        # 4. Final MP3 Export
+        # 4. Final MP3 Export (Guarantees .mp3 output)
         final_mp3 = os.path.join(out_dir, "ClearWave_Final.mp3")
         subprocess.run([
             "ffmpeg", "-y", "-i", res['audio_path'],
@@ -44,20 +48,43 @@ def run_pipeline(audio_input, remove_fillers, remove_stutters, remove_silences,
     except Exception as e:
         return None, f"Error: {e}", ""
 
-# UI Design
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎙️ ClearWave: Universal Audio Studio")
+# ══════════════════════════════════════════════════════════════════
+# GRADIO INTERFACE (The Golden Layout)
+# ══════════════════════════════════════════════════════════════════
+with gr.Blocks(title="ClearWave AI Studio") as demo:
+    gr.Markdown("# 🎙️ ClearWave AI Studio")
+    gr.Markdown("Professional MP3 Audio Cleaning, Transcription, and Translation.")
+    
     with gr.Row():
-        with gr.Column():
-            audio_in = gr.Audio(label="Upload Audio (Any Format)", type="filepath")
-            f = gr.Checkbox(label="Remove Fillers/Stutters", value=True)
-            lang = gr.Dropdown(choices=["Hindi", "Telugu", "Spanish"], value="Hindi", label="Translate to")
-            btn = gr.Button("🚀 Process to Studio MP3", variant="primary")
-        with gr.Column():
-            audio_out = gr.Audio(label="Final Studio MP3")
-            t_out = gr.Textbox(label="Transcript")
-            tr_out = gr.Textbox(label="Translation")
-
-    btn.click(run_pipeline, [audio_in, f, f, f, f, f, lang], [audio_out, t_out, tr_out])
-
-demo.launch(theme=gr.themes.Soft())
+        with gr.Column(scale=1):
+            audio_in = gr.Audio(label="Upload Audio (WhatsApp, AAC, WAV, MP3)", type="filepath")
+            
+            with gr.Group():
+                gr.Markdown("### 🛠️ Audio Cleaning Settings")
+                f_check = gr.Checkbox(label="Remove Fillers (um, ah)", value=True)
+                s_check = gr.Checkbox(label="Remove Stutters", value=True)
+                z_check = gr.Checkbox(label="Remove Long Silences", value=True)
+                b_check = gr.Checkbox(label="Remove Breaths", value=True)
+                m_check = gr.Checkbox(label="Remove Mouth Noises", value=True)
+            
+            target_lang = gr.Dropdown(
+                choices=["Hindi", "Telugu", "Spanish", "French", "German", "Japanese", "Arabic"],
+                value="Hindi",
+                label="Translation Target Language"
+            )
+            
+            process_btn = gr.Button("🚀 Generate Studio MP3", variant="primary")
+
+        with gr.Column(scale=1):
+            audio_out = gr.Audio(label="Final Cleaned MP3 Output")
+            transcript_output = gr.Textbox(label="Cleaned Transcript", lines=5)
+            translation_output = gr.Textbox(label="Translation", lines=5)
+
+    process_btn.click(
+        fn=run_pipeline,
+        inputs=[audio_in, f_check, s_check, z_check, b_check, m_check, target_lang],
+        outputs=[audio_out, transcript_output, translation_output]
+    )
+
+if __name__ == "__main__":
+    demo.launch(theme=gr.themes.Soft())