Update app.py

app.py

@@ -1,75 +1,104 @@
+import gradio as gr
+import os
+import shutil
+import tempfile
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import FileResponse
-import tempfile, os, shutil
-
 from core.cloner import KokoClone
 
-app = FastAPI(title="KokoClone API")
-
-# Charge les modèles une seule fois au démarrage
+# --- Chargement des modèles (une seule fois) ---
 print("Loading KokoClone models...")
 cloner = KokoClone()
 
+# --- FastAPI montée SUR l'app Gradio ---
+def clone_voice(text, lang, ref_audio_path):
+    if not text or not text.strip():
+        raise gr.Error("Please enter some text.")
+    if not ref_audio_path:
+        raise gr.Error("Please upload or record a reference audio file.")
+    output_file = "gradio_output.wav"
+    try:
+        cloner.generate(text=text, lang=lang, reference_audio=ref_audio_path, output_path=output_file)
+        return output_file
+    except Exception as e:
+        raise gr.Error(f"Error: {str(e)}")
+
+def convert_voice(source_audio_path, ref_audio_path):
+    if not source_audio_path:
+        raise gr.Error("Please upload a source audio file.")
+    if not ref_audio_path:
+        raise gr.Error("Please upload a reference audio file.")
+    output_file = "gradio_convert_output.wav"
+    try:
+        cloner.convert(source_audio=source_audio_path, reference_audio=ref_audio_path, output_path=output_file)
+        return output_file
+    except Exception as e:
+        raise gr.Error(f"Error: {str(e)}")
+
+# --- UI Gradio (inchangée) ---
+with gr.Blocks() as demo:
+    gr.Markdown("<div style='text-align:center'><h1>🎧 KokoClone</h1></div>")
+    with gr.Tabs():
+        with gr.Tab("🎤 Text → Clone"):
+            with gr.Row():
+                with gr.Column():
+                    text_input = gr.Textbox(label="Text", lines=4)
+                    lang_input = gr.Dropdown(
+                        choices=[("English","en"),("French","fr"),("Hindi","hi"),
+                                 ("Japanese","ja"),("Chinese","zh"),("Spanish","es")],
+                        value="en", label="Language"
+                    )
+                    ref_audio = gr.Audio(label="Reference Voice", type="filepath")
+                    btn = gr.Button("🚀 Generate", variant="primary")
+                with gr.Column():
+                    out_audio = gr.Audio(label="Output", interactive=False)
+            btn.click(clone_voice, inputs=[text_input, lang_input, ref_audio], outputs=out_audio)
 
-@app.get("/")
-def root():
-    return {"status": "ok", "service": "KokoClone API"}
+        with gr.Tab("🔁 Audio → Clone"):
+            with gr.Row():
+                with gr.Column():
+                    src_audio = gr.Audio(label="Source Audio", type="filepath")
+                    ref_audio2 = gr.Audio(label="Reference Voice", type="filepath")
+                    btn2 = gr.Button("🔁 Convert", variant="primary")
+                with gr.Column():
+                    out_audio2 = gr.Audio(label="Output", interactive=False)
+            btn2.click(convert_voice, inputs=[src_audio, ref_audio2], outputs=out_audio2)
 
+# --- Montage FastAPI ---
+app = gr.mount_gradio_app(FastAPI(), demo, path="/")
 
-@app.post("/clone")
-async def clone_voice(
+@app.post("/api/clone")
+async def api_clone(
     text: str = Form(...),
     lang: str = Form(default="en"),
     reference_audio: UploadFile = File(...)
 ):
-    """
-    Text + reference audio → cloned speech WAV
-    """
-    # Sauvegarde l'audio de référence dans un fichier temporaire
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as ref_tmp:
         shutil.copyfileobj(reference_audio.file, ref_tmp)
         ref_path = ref_tmp.name
-
-    out_path = ref_path.replace(".wav", "_output.wav")
-
+    out_path = ref_path.replace(".wav", "_out.wav")
     try:
-        cloner.generate(
-            text=text,
-            lang=lang,
-            reference_audio=ref_path,
-            output_path=out_path
-        )
+        cloner.generate(text=text, lang=lang, reference_audio=ref_path, output_path=out_path)
         return FileResponse(out_path, media_type="audio/wav", filename="cloned.wav")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
     finally:
         os.unlink(ref_path)
 
-
-@app.post("/convert")
-async def convert_voice(
+@app.post("/api/convert")
+async def api_convert(
     source_audio: UploadFile = File(...),
     reference_audio: UploadFile = File(...)
 ):
-    """
-    Source audio + reference audio → re-voiced WAV
-    """
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as src_tmp:
         shutil.copyfileobj(source_audio.file, src_tmp)
         src_path = src_tmp.name
-
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as ref_tmp:
         shutil.copyfileobj(reference_audio.file, ref_tmp)
         ref_path = ref_tmp.name
-
     out_path = src_path.replace(".wav", "_converted.wav")
-
     try:
-        cloner.convert(
-            source_audio=src_path,
-            reference_audio=ref_path,
-            output_path=out_path
-        )
+        cloner.convert(source_audio=src_path, reference_audio=ref_path, output_path=out_path)
         return FileResponse(out_path, media_type="audio/wav", filename="converted.wav")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))