Update app.py

app.py

@@ -1,111 +1,46 @@
 import os
-import uuid
-from pathlib import Path
-import requests
 import gradio as gr
 from TTS.api import TTS
 
-# ---------------------------
-# Model directory
-# ---------------------------
-MODEL_DIR = Path("models/xtts_v2")
-MODEL_DIR.mkdir(parents=True, exist_ok=True)
-
-# ---------------------------
-# Files to download
-# ---------------------------
-FILES = {
-    "model.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/main/model.pth",
-    "config.json": "https://huggingface.co/coqui/XTTS-v2/resolve/main/config.json",
-    "vocab.json": "https://huggingface.co/coqui/XTTS-v2/resolve/main/vocab.json",
-    "speakers_xtts.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/main/speakers_xtts.pth",
-}
-
-# ---------------------------
-# Download files function
-# ---------------------------
-def download_file(url, path):
-    if path.exists():
-        print(f"✅ Already exists: {path}")
-        return
-    print(f"⬇️ Downloading {path.name}...")
-    response = requests.get(url, stream=True)
-    total = int(response.headers.get('content-length', 0))
-    with open(path, 'wb') as f:
-        downloaded = 0
-        for data in response.iter_content(chunk_size=1024*1024):
-            f.write(data)
-            downloaded += len(data)
-            print(f"\r{path.name}: {downloaded/1024/1024:.2f}/{total/1024/1024:.2f} MB", end="")
-    print(f"\n✅ Downloaded: {path.name}")
-
-# Download all model files
-for fname, url in FILES.items():
-    download_file(url, MODEL_DIR / fname)
-
-# ---------------------------
-# Load TTS model (once)
-# ---------------------------
-print("⬇️ Loading XTTS-v2 model...")
+MODEL_PATH = "/models/xtts/model.pth"
+CONFIG_PATH = "/models/xtts/config.json"
 
+print("🔁 Loading XTTS model once...")
 
 tts = TTS(
-    model_path="models/xtts_v2",      # 👈 directory, NOT model.pth
-    config_path="models/xtts_v2/config.json",
+    model_path=MODEL_PATH,
+    config_path=CONFIG_PATH,
     gpu=False
 )
 
-tts.to("cpu")
-
+print("✅ Model loaded successfully")
 
+def generate_tts(text, speaker_wav):
+    if not text or speaker_wav is None:
+        return None
 
-print("✅ Model loaded successfully!")
+    out_path = "output.wav"
 
-# ---------------------------
-# Outputs directory
-# ---------------------------
-OUTPUT_DIR = Path("outputs")
-OUTPUT_DIR.mkdir(exist_ok=True)
-
-# ---------------------------
-# TTS generation function
-# ---------------------------
-def generate_tts(text, voice_file):
-    if not text or voice_file is None:
-        return None, "❌ Please provide both text and voice sample."
-
-    # Generate unique output filename
-    out_name = f"{uuid.uuid4().hex}.wav"
-    out_path = OUTPUT_DIR / out_name
-
-    # Generate audio
     tts.tts_to_file(
         text=text,
-        speaker_wav=voice_file,
-        language="hi",  # You can change language if needed
-        file_path=str(out_path)
+        speaker_wav=speaker_wav,
+        language="hi",
+        file_path=out_path
     )
 
-    return str(out_path), "✅ Audio generated!"
+    return out_path
 
-# ---------------------------
-# Gradio UI
-# ---------------------------
-demo = gr.Interface(
+
+app = gr.Interface(
     fn=generate_tts,
     inputs=[
-        gr.Textbox(label="Text", placeholder="Enter text here..."),
-        gr.Audio(type="filepath", label="Upload Voice Sample (WAV)")
-    ],
-    outputs=[
-        gr.Audio(label="Generated Audio"),
-        gr.Textbox(label="Status")
+        gr.Textbox(label="Text"),
+        gr.Audio(type="filepath", label="Sample Voice (WAV)")
     ],
-    title="XTTS Voice Cloning (CPU, HF Space)",
-    description="Upload a voice sample + text → get cloned voice audio. Model loads once and is reused."
+    outputs=gr.Audio(type="filepath", label="Generated Voice"),
+    title="XTTS Voice Cloning (CPU)",
+    description="Upload voice + text → get cloned speech"
 )
 
-# Launch Gradio app
 if __name__ == "__main__":
-    print("🚀 Launching Gradio UI...")
-    demo.launch(share=True)  # HF Space automatically creates URL
+    app.launch(server_name="0.0.0.0", server_port=7860)