Update app.py

app.py

@@ -1,40 +1,53 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
 from huggingface_hub import InferenceClient
+import os
 
-# Simple hosted model
 MODEL_ID = "microsoft/Phi-3-mini-4k-instruct"
-client = InferenceClient(model=MODEL_ID)
+HF_TOKEN = os.getenv("HF_TOKEN")
+client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
 
 app = FastAPI()
 
-class HumanizeRequest(BaseModel):
-    text: str
-
-# CORS so your Vercel frontend can call this API
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # later: ["https://formatai.live"]
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
 @app.post("/api/humanize")
-async def generate_humanized_text(request: HumanizeRequest):
+async def generate_humanized_text(request: Request):
+    # 1. Read raw JSON body
+    try:
+        body = await request.json()
+        print("Incoming JSON:", body)  # shows in Space logs
+    except Exception:
+        raise HTTPException(status_code=400, detail="Invalid JSON")
+
+    # 2. Extract "text"
+    text = body.get("text")
+    if not isinstance(text, str) or not text.strip():
+        raise HTTPException(status_code=400, detail="Field 'text' is required")
+
+    # 3. Build prompt and call model
     prompt = (
         "Humanize the following text, making it sound natural and engaging:\n\n"
-        f"{request.text}"
+        f"{text}"
     )
 
-    response = client.text_generation(
-        prompt,
-        max_new_tokens=512,
-        temperature=0.7,
-        top_p=0.9,
-        repetition_penalty=1.1,
-    )
+    try:
+        response = client.text_generation(
+            prompt,
+            max_new_tokens=512,
+            temperature=0.7,
+            top_p=0.9,
+            repetition_penalty=1.1,
+        )
+    except Exception as e:
+        print("HF error:", e)
+        raise HTTPException(status_code=500, detail=f"HF error: {e}")
 
     return {"result": response}