Create main.py

main.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+import requests
+import base64
+import os
+
+app = FastAPI()
+
+HUGGINGFACE_API_URL = "https://api-inference.huggingface.co/models/Somali-tts/somali_tts_model"
+HUGGINGFACE_API_TOKEN = f"Bearer {os.environ['HF_TOKEN']}"  # Token-ka laga helayo environment
+headers = {"Authorization": HUGGINGFACE_API_TOKEN}
+
+class TextInput(BaseModel):
+    text: str
+
+@app.post("/tts/")
+def generate_audio(input: TextInput):
+    response = requests.post(HUGGINGFACE_API_URL, headers=headers, json={"inputs": input.text})
+    if response.status_code == 200:
+        audio_base64 = base64.b64encode(response.content).decode("utf-8")
+        return {"audio_base64": audio_base64}
+    else:
+        return {"error": response.json()}