Update main.py

main.py

@@ -4,6 +4,7 @@ from fastapi.openapi.utils import get_openapi
 from fastapi.openapi.docs import get_swagger_ui_html
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
+from model import answer_only
 from fastapi import Query
 from transformers import pipeline
 import os, logging, traceback
@@ -169,3 +170,24 @@ async def bulk_analyze(data: BulkReviewInput, token: str = Query(None)):
     except Exception as e:
         logging.error(f"🔥 Bulk processing failed: {traceback.format_exc()}")
         raise HTTPException(status_code=500, detail="Failed to analyze bulk reviews")
+from model import answer_only
+
+class FollowUpRequest(BaseModel):
+    text: str
+    question: str
+    verbosity: Optional[str] = "brief"
+
+@app.post("/followup/")
+async def followup(request: FollowUpRequest, x_api_key: str = Header(None)):
+    if x_api_key and x_api_key != VALID_API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
+    if not request.question or len(request.text.split()) < 10:
+        raise HTTPException(status_code=400, detail="Question or text is too short.")
+
+    try:
+        answer = answer_only(request.text, request.question)
+        return {"answer": answer}
+    except Exception as e:
+        logging.error(f"❌ Follow-up failed: {traceback.format_exc()}")
+        raise HTTPException(status_code=500, detail="Internal Server Error during follow-up.")