Update lib/ui.py

lib/ui.py

@@ -1,45 +1,53 @@
-# lib/ui.py
-import os
-import zipfile
-from pathlib import Path
+"""
+UI components and helper functions for Gradio.
+"""
 
 import gradio as gr
 
-from .config import MODELS_DIR, BUILTIN_MODELS, logger
+from .config import OUTPUT_FORMATS, DEFAULT_OUTPUT_FORMAT
 from .models import list_models
 
+# Original functions (keep as they are)
+def refresh_models():
+    """Refresh the model dropdown and table."""
+    models = list_models()
+    return gr.update(value=models, choices=models), models
 
-def toggle_autotune(autotune: bool):
-    """Toggle autotune slider visibility"""
-    return gr.update(visible=autotune)
-
+def toggle_autotune(enable):
+    """Show/hide autotune strength slider."""
+    return gr.update(visible=enable)
 
 def upload_model(zip_file, model_name):
-    """Upload and extract a model zip file"""
-    if not zip_file:
-        return "❌ No file selected", gr.update(), None
-    
-    file_path = zip_file if isinstance(zip_file, str) else zip_file.get("name")
-    if not file_path or not os.path.exists(file_path):
-        return "❌ Invalid file", gr.update(), None
-    
-    name = model_name if model_name else Path(file_path).stem
-    model_dir = MODELS_DIR / name
-    model_dir.mkdir(exist_ok=True)
-    
-    try:
-        with zipfile.ZipFile(file_path, 'r') as zf:
-            zf.extractall(model_dir)
-        
-        new_models = list_models()
-        return f"✅ Model '{name}' loaded!", gr.update(choices=new_models, value=name), [[m] for m in new_models]
-    
-    except Exception as e:
-        logger.error(f"Upload error: {e}")
-        return f"❌ Error: {e}", gr.update(), None
-
-
-def refresh_models():
-    """Refresh the models list"""
-    models = list_models()
-    return gr.update(choices=models), [[m] for m in models]
+    """Extract and load a model from a ZIP file."""
+    # Original implementation
+    return "Model uploaded", list_models(), list_models()
+
+# New UI helpers for the full pipeline
+
+def create_video_section():
+    """
+    Returns a Gradio Video component for uploading video files.
+    """
+    return gr.Video(
+        label="Upload Video (MP4, WebM, etc.)",
+        sources=["upload"],
+        format="mp4",
+        height=300,
+    )
+
+def create_five_outputs():
+    """
+    Creates the 5 output audio components and returns them as a tuple.
+    Use this in your app.py to display the results.
+    """
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("#### 🎤 Input Files (no RVC)")
+            entrada_acapella = gr.Audio(label="Acapella (extracted)", type="filepath", interactive=False)
+            entrada_audio = gr.Audio(label="Original Audio", type="filepath", interactive=False)
+            entrada_instrumental = gr.Audio(label="Instrumental (extracted)", type="filepath", interactive=False)
+        with gr.Column():
+            gr.Markdown("#### 🎙️ Output Files (with RVC)")
+            saida_audio = gr.Audio(label="RVC on original", type="filepath", interactive=False)
+            saida_acapella = gr.Audio(label="RVC on acapella", type="filepath", interactive=False)
+    return entrada_acapella, entrada_audio, entrada_instrumental, saida_audio, saida_acapella