Update app.py

app.py

@@ -1,50 +1,58 @@
-from fastapi import FastAPI, HTTPException, Request
+import os
+from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 from huggingface_hub import InferenceClient
-import os
 
+# --- Model & client setup ---
 MODEL_ID = "microsoft/Phi-3-mini-4k-instruct"
-HF_TOKEN = os.getenv("HF_TOKEN")
+HF_TOKEN = os.getenv("HF_TOKEN")  # set in Space secrets
+
 client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
 
 app = FastAPI()
 
+# --- CORS ---
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # later restrict to your domain
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-@app.post("/api/humanize")
-async def generate_humanized_text(request: Request):
-    # 1. Read JSON body
-    try:
-        body = await request.json()
-    except:
-        raise HTTPException(status_code=400, detail="Invalid JSON")
+# --- Request body schema ---
+class HumanizeRequest(BaseModel):
+    text: str
 
-    # 2. Extract field
-    text = body.get("text")
-    if not text:
-        raise HTTPException(status_code=400, detail="Missing 'text' field")
+# --- Endpoint ---
+@app.post("/api/humanize")
+async def generate_humanized_text(body: HumanizeRequest):
+    text = body.text
+    if not text.strip():
+        raise HTTPException(status_code=400, detail="Text cannot be empty")
 
-    # 3. Build prompt
-    prompt = f"Humanize the following text:\n\n{text}"
+    prompt = (
+        "Humanize the following text, making it sound natural and engaging:\n\n"
+        f"{text}"
+    )
 
-    # 4. Generate output
     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:
-        raise HTTPException(status_code=500, detail=str(e))
+        # This will show up in Space logs
+        print("HF error:", e)
+        raise HTTPException(status_code=500, detail=f"HF error: {e}")
 
     return {"result": response}
 
+# --- Health check ---
 @app.get("/")
 def health():
     return {"status": "ok"}