Update app/main.py

app/main.py

@@ -1,32 +1,32 @@
-from fastapi import FastAPI, Request
-from fastapi.templating import Jinja2Templates
-from fastapi.staticfiles import StaticFiles
-
-from rag import load_data, retrieve_chunks
-
-app = FastAPI()
-
-templates = Jinja2Templates(directory="templates")
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
-chunks, heads = load_data()
-
-@app.get("/")
-def home(request: Request):
-    return templates.TemplateResponse(
-        "index.html",
-        {"request": request}
-    )
-
-@app.post("/search")
-async def search(request: Request):
-    body = await request.json()
-    query = body["query"]
-
-    retrieved = retrieve_chunks(query, chunks, heads)
-    answer = "\n\n".join(retrieved[:2])
-
-    return {
-        "answer": answer,
-        "sources": retrieved
-    }
+from fastapi import FastAPI, Request
+from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
+
+from app.rag import load_data, retrieve_chunks
+
+app = FastAPI()
+
+templates = Jinja2Templates(directory="templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+chunks, heads = load_data()
+
+@app.get("/")
+def home(request: Request):
+    return templates.TemplateResponse(
+        "index.html",
+        {"request": request}
+    )
+
+@app.post("/search")
+async def search(request: Request):
+    body = await request.json()
+    query = body["query"]
+
+    retrieved = retrieve_chunks(query, chunks, heads)
+    answer = "\n\n".join(retrieved[:2])
+
+    return {
+        "answer": answer,
+        "sources": retrieved
+    }