Update main.py

Author: fatimaalizaidi

app/main.py

@@ -4,6 +4,7 @@
 from app.classifier_model import load_models, classify_text
 import os
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import JSONResponse
 
 
 print("svc_texpose - starting")
@@ -25,16 +26,23 @@
 async def root(request: Request):
     return templates.TemplateResponse("index.html", {"request": request, "prediction": None, "input_text": ""})
 
-@app.post("/classify", response_class=HTMLResponse)
-async def classify(request: Request, input_text: str = Form(default="")):
-    if input_text.strip():  # Ensure input is not empty
+
+
+
+@app.post("/api/classify")
+async def classify_api(request: Request):
+    data = await request.json()
+    input_text = data.get("input_text", "")
+
+    if input_text.strip():
         result = classify_text(input_text, model_ai_hum, model_llm, tokenizer)
         prediction = result["type"]
 
-        # Ensure "llm" exists and is not empty
         if result.get("llm"):
             prediction += f" Using {result['llm']}"
 
-        return templates.TemplateResponse("index.html", {"request": request, "prediction": prediction, "input_text": input_text})
+        return JSONResponse(content={"prediction": prediction})
+    
+    return JSONResponse(content={"prediction": "No text provided."})
 
-    return templates.TemplateResponse("index.html", {"request": request, "prediction": "No text provided.", "input_text": ""})
+