Revert to simple clean UI

app.py

@@ -4,7 +4,6 @@ import torch
 import torchaudio
 import tempfile
 import warnings
-import os
 warnings.filterwarnings("ignore")
 
 from sam_audio import SAMAudio, SAMAudioProcessor
@@ -49,150 +48,48 @@ def separate_audio(audio_path, video_path, text_prompt):
         target_path = save_audio(result.target[0].unsqueeze(0).cpu(), sample_rate)
         residual_path = save_audio(result.residual[0].unsqueeze(0).cpu(), sample_rate)
 
-        return target_path, residual_path, f"✅ Successfully isolated **'{text_prompt}'**"
+        return target_path, residual_path, f"✅ Successfully isolated '{text_prompt}'"
     
     except Exception as e:
         import traceback
         traceback.print_exc()
         return None, None, f"❌ Error: {str(e)}"
 
-# Custom CSS
-custom_css = """
-.gradio-container {
-    background: #0d0d0d !important;
-    max-width: 1400px !important;
-}
-
-.gradio-container *, .gradio-container p, .gradio-container span, 
-.gradio-container h1, .gradio-container h2, .gradio-container h3,
-.gradio-container label, .gradio-container .markdown-text {
-    color: #ffffff !important;
-}
-
-input, textarea {
-    background: #1a1a1a !important;
-    border: 1px solid #444 !important;
-    color: #ffffff !important;
-}
-
-input::placeholder, textarea::placeholder {
-    color: #888 !important;
-}
-
-.primary-btn {
-    background: linear-gradient(135deg, #e91e8c, #9c27b0) !important;
-    border: none !important;
-    border-radius: 24px !important;
-    color: #ffffff !important;
-    font-weight: 600 !important;
-}
-
-.example-btn {
-    background: #2a2a2a !important;
-    border: 1px solid #444 !important;
-    color: #ffffff !important;
-    border-radius: 8px !important;
-}
-
-.example-btn:hover {
-    background: #3a3a3a !important;
-    border-color: #e91e8c !important;
-}
-
-hr {
-    border-color: #333 !important;
-}
-"""
-
-# Build Gradio Interface
-with gr.Blocks(
-    title="SAM-Audio - Isolate Sounds",
-    theme=gr.themes.Default(
-        primary_hue="pink",
-        secondary_hue="purple",
-        neutral_hue="gray",
-    ),
-    css=custom_css
-) as demo:
+# Build Gradio Interface - Simple and clean
+with gr.Blocks(title="SAM-Audio Demo") as demo:
+    gr.Markdown(
+        """
+        # 🎵 SAM-Audio: Segment Anything for Audio
+        
+        Isolate specific sounds from an audio or video file using natural language prompts.
+        
+        **Model:** [facebook/sam-audio-small](https://huggingface.co/facebook/sam-audio-small)
+        """
+    )
     
     with gr.Row():
-        # Sidebar
-        with gr.Column(scale=1, min_width=260):
-            gr.Markdown("# 🎵 Isolate Sounds")
-            gr.Markdown("Extract and isolate any sound from audio or video using AI.")
-            
-            gr.Markdown("---")
-            gr.Markdown("### How it works")
-            gr.Markdown("**1.** Add audio or video")
-            gr.Markdown("**2.** Describe the sound")
-            gr.Markdown("**3.** Get separated tracks")
+        with gr.Column():
+            gr.Markdown("### Upload Audio or Video")
+            input_audio = gr.Audio(label="Audio File", type="filepath")
+            input_video = gr.Video(label="Video File (MP4)")
             
-            gr.Markdown("---")
-            gr.Markdown("**Model:** SAM-Audio Small")
-        
-        # Main content area
-        with gr.Column(scale=4):
-            gr.Markdown("## 📤 Upload Audio or Video")
-            
-            with gr.Tabs():
-                with gr.TabItem("� Audio"):
-                    input_audio = gr.Audio(
-                        label="Upload audio file (MP3, WAV, FLAC, etc.)",
-                        type="filepath"
-                    )
-                
-                with gr.TabItem("🎬 Video"):
-                    input_video = gr.Video(
-                        label="Upload video file (MP4, MKV, AVI, etc.)"
-                    )
-            
-            gr.Markdown("## 💬 Describe the Sound")
             text_prompt = gr.Textbox(
-                label="",
+                label="Text Prompt", 
                 placeholder="e.g., 'A man speaking', 'Piano melody', 'Dog barking'",
-                lines=1
-            )
-            
-            run_btn = gr.Button(
-                "🎯 Isolate Sound", 
-                variant="primary",
-                size="lg",
-                elem_classes=["primary-btn"]
-            )
-            
-            status_output = gr.Markdown(
-                value="Upload a file and describe what sound you want to isolate."
+                info="Describe the sound you want to isolate."
             )
             
-            gr.Markdown("---")
-            gr.Markdown("## 🎧 Results")
-            
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("**🎯 Isolated Sound (Target)**")
-                    output_target = gr.Audio(label="", show_label=False)
-                    
-                with gr.Column():
-                    gr.Markdown("**🔇 Background (Residual)**")
-                    output_residual = gr.Audio(label="", show_label=False)
-            
-            gr.Markdown("---")
-            gr.Markdown("## 💡 Example Prompts")
-            
-            with gr.Row():
-                btn1 = gr.Button("A man speaking", elem_classes=["example-btn"])
-                btn2 = gr.Button("A woman singing", elem_classes=["example-btn"])
-                btn3 = gr.Button("Piano", elem_classes=["example-btn"])
-                btn4 = gr.Button("Drums", elem_classes=["example-btn"])
-                btn5 = gr.Button("Guitar", elem_classes=["example-btn"])
-                btn6 = gr.Button("Dog barking", elem_classes=["example-btn"])
-            
-            btn1.click(fn=lambda: "A man speaking", outputs=[text_prompt])
-            btn2.click(fn=lambda: "A woman singing", outputs=[text_prompt])
-            btn3.click(fn=lambda: "Piano", outputs=[text_prompt])
-            btn4.click(fn=lambda: "Drums", outputs=[text_prompt])
-            btn5.click(fn=lambda: "Guitar", outputs=[text_prompt])
-            btn6.click(fn=lambda: "Dog barking", outputs=[text_prompt])
+            run_btn = gr.Button("🎯 Isolate Sound", variant="primary")
+            status_output = gr.Markdown(value="")
+        
+        with gr.Column():
+            gr.Markdown("### Results")
+            output_target = gr.Audio(label="Isolated Sound (Target)")
+            output_residual = gr.Audio(label="Background (Residual)")
+    
+    gr.Markdown("---")
+    gr.Markdown("### Example Prompts")
+    gr.Markdown("- A man speaking\n- A woman singing\n- Piano\n- Drums\n- Guitar\n- Dog barking\n- Car engine")
 
     run_btn.click(
         fn=separate_audio,