Update app/main.py

app/main.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, Request, Form
-from fastapi.responses import HTMLResponse
+from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from app.llm import generate_answer
@@ -20,16 +20,28 @@ async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request, "answer": ""})
 
 
-@app.post("/ask", response_class=HTMLResponse)
-async def ask(request: Request, query: str = Form(...)):
-    # Retrieve relevant chunks
+# @app.post("/ask", response_class=HTMLResponse)
+# async def ask(request: Request, query: str = Form(...)):
+#     # Retrieve relevant chunks
+#     retrieved = retrieve_chunks(query, documents, embeddings)
+#     context = "\n\n".join(retrieved)
+
+#     # Ask the model
+#     answer = generate_answer(context, query)
+
+#     return templates.TemplateResponse(
+#         "index.html",
+#         {"request": request, "answer": answer, "query": query}
+#     )
+
+
+
+@app.post("/ask")
+async def ask(query: str = Form(...)):
     retrieved = retrieve_chunks(query, documents, embeddings)
     context = "\n\n".join(retrieved)
 
-    # Ask the model
     answer = generate_answer(context, query)
 
-    return templates.TemplateResponse(
-        "index.html",
-        {"request": request, "answer": answer, "query": query}
-    )
+    return {"answer": answer}
+