Update app/main.py

app/main.py

@@ -1,37 +1,35 @@
-# app/main.py
 from fastapi import FastAPI, Request, Form
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-import os
 
-from app.rag import load_data, retrieve_chunks
-from app.llm import generate_answer
+from rag import load_data, retrieve_chunks, ask_model  # your RAG functions
 
 app = FastAPI()
 
-app.mount("/static", StaticFiles(directory="static"), name="static")
-templates = Jinja2Templates(directory="templates")
+# Static files and templates
+app.mount("/static", StaticFiles(directory="app/static"), name="static")
+templates = Jinja2Templates(directory="app/templates")
+
+# Load your RAG data once
+documents, embeddings, model = load_data()
 
-documents, embeddings = load_data() 
 
 @app.get("/", response_class=HTMLResponse)
-async def home(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request, "question": "", "answer": ""})
+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(...)):
-    # 1️⃣ Retrieve relevant chunks
+    # Retrieve relevant chunks
     retrieved = retrieve_chunks(query, documents, embeddings)
     context = "\n\n".join(retrieved)
 
-    answer = generate_answer(context, query)
+    # Ask the model
+    answer = ask_model(context, query)
 
     return templates.TemplateResponse(
         "index.html",
-        {
-            "request": request,
-            "question": query,
-            "answer": answer
-        }
+        {"request": request, "answer": answer, "query": query}
     )