Update app.py

app.py

@@ -1,7 +1,9 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from fastapi.responses import PlainTextResponse
+from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import uvicorn
 
 app = FastAPI()
 
@@ -13,6 +15,10 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 model.eval()
 
 
+class Message(BaseModel):
+    message: str
+
+
 def generate_ai(message: str):
 
     prompt = f"User: {message}\nAssistant:"
@@ -39,31 +45,23 @@ async def root():
     return PlainTextResponse("AI server работает")
 
 
-@app.api_route("/", methods=["GET","POST","PUT","PATCH","DELETE","HEAD"])
-async def universal(request: Request):
+# POST (для docs)
+@app.post("/")
+async def receive(data: Message):
 
-    try:
+    reply = generate_ai(data.message)
 
-        # пробуем получить JSON
-        try:
-            data = await request.json()
-            message = data.get("message", "")
-        except:
-            # если не JSON — читаем обычный текст
-            body = await request.body()
-            message = body.decode("utf-8")
+    return PlainTextResponse(reply)
 
-        if not message:
-            message = "Hello"
 
-        reply = generate_ai(message)
+# GET (для Minecraft)
+@app.get("/ask")
+async def ask(message: str):
 
-        return PlainTextResponse(reply)
+    reply = generate_ai(message)
 
-    except Exception as e:
-        return PlainTextResponse(f"ERROR: {str(e)}")
+    return PlainTextResponse(reply)
 
 
 if __name__ == "__main__":
-    import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)