Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from langchain_core.messages import HumanMessage
-from veryfinal import main
+from veryfinal import run_query
 
 
 
@@ -25,7 +25,7 @@ class BasicAgent:
         print(f"Agent received question: {question}")
         
         # Get the full response from your agent
-        full_response = main(question)
+        full_response = run_query(question)
         
         # Extract just the answer part after "FINAL ANSWER: "
         if "FINAL ANSWER: " in full_response: