Update app.py

app.py

@@ -1,18 +1,17 @@
 import os
-import requests
 from fastapi import FastAPI, Form, HTTPException
 from fastapi.responses import HTMLResponse, FileResponse
+from huggingface_hub import InferenceClient
 
 app = FastAPI()
 
-# একটি অত্যন্ত ফাস্ট এবং স্টেবল টেক্সট-টু-স্পিচ মডেল
-API_URL = "https://api-inference.huggingface.co/models/facebook/mms-tts-eng"
-
-# হাগিংফেস স্পেসের নিজস্ব পরিবেশ থেকে টোকেন নেওয়া
+# হাগিংফেস ক্লায়েন্ট ইনিশিয়ালাইজ করা
+# স্পেসের Secrets থেকে টোকেন থাকলে নেবে, না থাকলে ফ্রি টিয়ারে কাজ করবে
 HF_TOKEN = os.getenv("HF_TOKEN", "")
-headers = {}
-if HF_TOKEN:
-    headers["Authorization"] = f"Bearer {HF_TOKEN}"
+client = InferenceClient(token=HF_TOKEN if HF_TOKEN else None)
+
+# আপনি চাইলে Kokoro বা Facebook এর যেকোনো মডেল এখানে দিতে পারেন
+MODEL_ID = "facebook/mms-tts-eng"
 
 @app.get("/", response_class=HTMLResponse)
 def index():
@@ -24,10 +23,10 @@ def index():
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
         <title>Raybil Voice AI Engine</title>
         <style>
-            body { font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; max-width: 600px; margin: 40px auto; padding: 30px; background: #0f172a; color: #f8fafc; border-radius: 12px; box-shadow: 0 10px 15px -3px rgba(0,0,0,0.3); }
+            body { font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; max-width: 600px; margin: 40px auto; padding: 30px; background: #0f172a; color: #f8fafc; border-radius: 12px; }
             h2 { color: #38bdf8; text-align: center; }
             textarea { width: 100%; height: 120px; padding: 12px; margin: 15px 0; background: #1e293b; color: #fff; border: 1px solid #475569; border-radius: 6px; box-sizing: border-box; resize: none; }
-            button { width: 100%; padding: 14px; background: #0284c7; color: white; border: none; border-radius: 6px; font-size: 16px; font-weight: bold; cursor: pointer; transition: 0.2s; }
+            button { width: 100%; padding: 14px; background: #0284c7; color: white; border: none; border-radius: 6px; font-size: 16px; font-weight: bold; cursor: pointer; }
             button:hover { background: #0369a1; }
         </style>
     </head>
@@ -44,29 +43,18 @@ def index():
 
 @app.post("/generate")
 def generate_voice(text: str = Form(...)):
-    payload = {"inputs": text}
-    
     try:
-        # হাগিংফেস সার্ভারকে রেসপন্স করার জন্য পর্যাপ্ত সময় (৬০ সেকেন্ড) দেওয়া হলো
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Network Timeout/Error: {str(e)}")
-    
-    if response.status_code == 200:
-        # ফেসবুকের এই মডেলগুলো .wav ফরম্যাটে অডিও দেয়, তাই আমরা সেফ ফরম্যাট ব্যবহার করছি
+        # ক্লায়েন্ট অটোমেটিকলি সঠিক এন্ডপয়েন্ট এবং নেটওয়ার্ক পাথ হ্যান্ডেল করবে
+        audio_bytes = client.text_to_speech(text, model=MODEL_ID)
+        
         output_path = "output.wav"
-        try:
-            with open(output_path, "wb") as f:
-                f.write(response.content)
-            return FileResponse(output_path, media_type="audio/wav", filename="voice.wav")
-        except Exception as e:
-            raise HTTPException(status_code=500, detail=f"File System Error: {str(e)}")
+        with open(output_path, "wb") as f:
+            f.write(audio_bytes)
             
-    # যদি মডেলটি হাগিংফেস সার্ভারে লোড হতে থাকে, তবে এটি ৫৩৩ বা ২০২ কোড দেয়
-    elif response.status_code in [202, 503]:
-        raise HTTPException(status_code=response.status_code, detail="মডেলটি হাগিংফেস সার্ভারে লোড হচ্ছে। অনুগ্রহ করে ৩০ সেকেন্ড পর আবার চেষ্টা করুন।")
-    else:
-        raise HTTPException(status_code=response.status_code, detail=f"HF Server returned error: {response.text}")
+        return FileResponse(output_path, media_type="audio/wav", filename="voice.wav")
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"AI Engine Error: {str(e)}")
 
 if __name__ == "__main__":
     import uvicorn