Update app/main.py

app/main.py

@@ -2,8 +2,8 @@ from fastapi import FastAPI, Request, Form
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-
-from app.rag import load_data, retrieve_chunks, ask_model  # your RAG functions
+from app.llm import generate_answer
+from app.rag import load_data, retrieve_chunks
 
 app = FastAPI()
 
@@ -27,7 +27,7 @@ async def ask(request: Request, query: str = Form(...)):
     context = "\n\n".join(retrieved)
 
     # Ask the model
-    answer = ask_model(context, query)
+    answer = generate_model(context, query)
 
     return templates.TemplateResponse(
         "index.html",