Generic Fix for Prompt and API (v8)

main.py

@@ -124,6 +124,19 @@ else:
 async def read_root(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
+from pydantic import BaseModel
+
+class QueryRequest(BaseModel):
+    query: str
+
+@app.post("/query")
+async def query_endpoint(request: QueryRequest):
+    if not rag_chain:
+        return {"answer": "System Error: RAG chain not initialized."}
+    
+    answer = rag_chain.invoke(request.query)
+    return {"answer": answer}
+
 @app.post("/get_response")
 async def get_response(request: Request, query: str = Form(...)):
     if not rag_chain: