Update app.py

app.py

@@ -8,7 +8,7 @@ load_dotenv()
 if not os.path.isdir("low_partners_chatbot_data"):
     load_dataset()
 
-from low_partners_chatbot_data.utils import run_conversation
+from low_partners_chatbot_data.agent import Agent
 
 
 import gradio as gr
@@ -17,6 +17,8 @@ from PIL import Image
 import io
 import os
 
+agent = Agent()
+
 
 def extract_pages_from_multiple_pdfs(pdf_info):
     images = []
@@ -47,7 +49,7 @@ def extract_pages_from_multiple_pdfs(pdf_info):
 
 
 def get_answer(query):
-    answer, pdf_info = run_conversation(query)
+    answer, pdf_info = agent.ask(query)
     print(f"Answer: {answer}")
     print(f"PDF Info: {pdf_info}")
     images = []
@@ -75,3 +77,4 @@ iface = gr.Interface(
 )
 
 iface.launch()
+