Update app.py

app.py

@@ -1,18 +1,18 @@
 import os
 import uuid
-import gradio as gr
-import requests
 from pathlib import Path
+import requests
+import gradio as gr
 from TTS.api import TTS
 
 # ---------------------------
-# Create models folder
+# Model directory
 # ---------------------------
 MODEL_DIR = Path("models/xtts_v2")
 MODEL_DIR.mkdir(parents=True, exist_ok=True)
 
 # ---------------------------
-# Model files URLs
+# Files to download
 # ---------------------------
 FILES = {
     "model.pth": "https://huggingface.co/coqui/XTTS-v2/resolve/main/model.pth",
@@ -22,7 +22,7 @@ FILES = {
 }
 
 # ---------------------------
-# Download files if not exist
+# Download files function
 # ---------------------------
 def download_file(url, path):
     if path.exists():
@@ -39,55 +39,50 @@ def download_file(url, path):
             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 XTTS Model (CPU)
+# Load TTS model (once)
 # ---------------------------
-
-
-
-# Load XTTS Model (CPU)
+print("⬇️ Loading XTTS-v2 model...")
 tts = TTS(
-    model_path=str(MODEL_DIR / "model.pth"),
+    model_path=str(MODEL_DIR),  # pass folder, NOT file
     config_path=str(MODEL_DIR / "config.json"),
-    gpu=False  # CPU in HF Space
+    gpu=False
 )
-
-
-
-
-
-print("✅ Model loaded and ready!")
+print("✅ Model loaded successfully!")
 
 # ---------------------------
-# Create outputs folder
+# Outputs directory
 # ---------------------------
 OUTPUT_DIR = Path("outputs")
 OUTPUT_DIR.mkdir(exist_ok=True)
 
 # ---------------------------
-# Generate TTS
+# 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",  # Change if needed
+        language="hi",  # You can change language if needed
         file_path=str(out_path)
     )
 
     return str(out_path), "✅ Audio generated!"
 
 # ---------------------------
-# Gradio interface
+# Gradio UI
 # ---------------------------
 demo = gr.Interface(
     fn=generate_tts,
@@ -103,5 +98,6 @@ demo = gr.Interface(
     description="Upload a voice sample + text → get cloned voice audio. Model loads once and is reused."
 )
 
+# Launch app
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)