Update app.py

app.py

@@ -1,58 +1,29 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse
+import gradio as gr
 import requests
 import os
 
-app = FastAPI(title="Chat FastAPI HuggingFace")
-
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL = os.getenv("HF_MODEL", "TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
-@app.get("/")
-async def root():
-    return {"message": "🚀 FastAPI Chat Model activo y listo"}
-
-@app.post("/chat")
-async def chat(request: Request):
-    try:
-        data = await request.json()
-        message = data.get("message", "")
-
-        if not HF_TOKEN:
-            return JSONResponse(
-                content={"error": "HF_TOKEN no configurado"},
-                status_code=500
-            )
-
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-        payload = {"inputs": message}
-
-        response = requests.post(
-            f"https://api-inference.huggingface.co/models/{MODEL}",
-            headers=headers,
-            json=payload
-        )
-
-        if response.status_code != 200:
-            return JSONResponse(
-                content={"error": "Error desde Hugging Face", "details": response.text},
-                status_code=500
-            )
+def chat(message):
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    payload = {"inputs": message}
 
-        result = response.json()
-        reply = ""
+    response = requests.post(
+        f"https://api-inference.huggingface.co/models/{MODEL}",
+        headers=headers,
+        json=payload
+    )
 
-        if isinstance(result, list) and len(result) > 0:
-            reply = result[0].get("generated_text", "")
-        elif isinstance(result, dict) and "generated_text" in result:
-            reply = result["generated_text"]
-        else:
-            reply = str(result)
+    if response.status_code != 200:
+        return f"⚠️ Error desde Hugging Face: {response.text}"
 
-        return {"reply": reply}
+    result = response.json()
+    if isinstance(result, list) and len(result) > 0:
+        return result[0].get("generated_text", "")
+    elif isinstance(result, dict) and "generated_text" in result:
+        return result["generated_text"]
+    else:
+        return str(result)
 
-    except Exception as e:
-        return JSONResponse(
-            content={"error": "Error interno", "details": str(e)},
-            status_code=500
-        )
+gr.ChatInterface(fn=chat, title="Chat Gradio con HF").launch()