Update app/main.py

app/main.py

@@ -1,7 +1,7 @@
 from pydantic import BaseModel
 from fastapi import FastAPI, UploadFile, File
 from app.pdf_handler import process_pdf, process_image_pdf
-from app.rag_pipeline import generate_answer
+from app.rag_pipeline import generate_answer, generate_answer_am
 from app.test import query_test
 from fastapi import FastAPI, File, UploadFile
 from PIL import Image
@@ -21,6 +21,12 @@ async def chat(request: ChatRequest):
     return generate_answer(request.query, request.top_k)
 
 
+@app.post("/am/chat/")
+async def chat(request: ChatRequest):
+    return generate_answer_am(request.query, request.top_k)
+
+
+
 @app.post("/upload/")
 async def upload_pdf(file: UploadFile = File(...)):
     return process_pdf(file)