Update app.py

app.py

@@ -1,15 +1,25 @@
 import io
-import torch
-import soundfile as sf
+import subprocess
+import tempfile
+import os
 from flask import Flask, request, send_file
-from chatterbox.tts import ChatterboxTTS
 
 app = Flask(__name__)
 
-print("⏳ Chatterbox मॉडल लोड हो रहा है (पहली बार थोड़ा समय लगेगा)...")
-# CPU पर चलाने के लिए device="cpu" करें
-model = ChatterboxTTS.from_pretrained(device="cpu")
-print("✅ Chatterbox आपकी सेवा में तैयार है!")
+# हिंदी के लिए Piper मॉडल डाउनलोड करें (पहली बार चलने पर)
+MODEL_URL = "https://huggingface.co/rhasspy/piper-voices/resolve/main/hi/hi_IN/harshit/low/hi_IN-harshit-low.onnx"
+MODEL_JSON_URL = "https://huggingface.co/rhasspy/piper-voices/resolve/main/hi/hi_IN/harshit/low/hi_IN-harshit-low.onnx.json"
+MODEL_PATH = "/tmp/hi_IN-harshit-low.onnx"
+MODEL_JSON_PATH = "/tmp/hi_IN-harshit-low.onnx.json"
+
+def download_model():
+    if not os.path.exists(MODEL_PATH):
+        print("⏳ मॉडल डाउनलोड हो रहा है...")
+        subprocess.run(["wget", "-q", "-O", MODEL_PATH, MODEL_URL])
+        subprocess.run(["wget", "-q", "-O", MODEL_JSON_PATH, MODEL_JSON_URL])
+        print("✅ मॉडल तैयार है!")
+
+download_model()
 
 @app.route('/tts', methods=['POST'])
 def tts_api():
@@ -17,21 +27,33 @@ def tts_api():
     if not text:
         return "Error: 'text' parameter is missing", 400
 
-    # ऑडियो को numpy array के रूप में जनरेट करें
-    audio_array = model.generate(text)
-    
-    # ऑडियो को बफर में सेव करें
-    audio_buffer = io.BytesIO()
-    sf.write(audio_buffer, audio_array, model.sr, format='WAV')
-    audio_buffer.seek(0)
+    # Piper TTS को कमांड लाइन से चलाएं
+    with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f:
+        output_file = f.name
+
+    try:
+        cmd = [
+            "piper",
+            "-m", MODEL_PATH,
+            "-c", MODEL_JSON_PATH,
+            "--output_file", output_file
+        ]
+        proc = subprocess.run(cmd, input=text.encode('utf-8'), capture_output=True)
+        if proc.returncode != 0:
+            return f"Piper error: {proc.stderr.decode()}", 500
+
+        with open(output_file, 'rb') as audio_file:
+            audio_data = audio_file.read()
+    finally:
+        os.unlink(output_file)
 
     return send_file(
-        audio_buffer,
+        io.BytesIO(audio_data),
         mimetype='audio/wav',
         as_attachment=True,
-        download_name='chatterbox_output.wav'
+        download_name='output.wav'
     )
 
 @app.route('/')
 def home():
-    return "🎙️ Chatterbox TTS हिंदी API चल रही है!"
+    return "🎙️ Piper TTS हिंदी API चल रही है!"