Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import streamlit as st
 import time
 from metrics_tracker import MetricsTracker
-from tool_handler import agent  # PySpur agent handles tool routing
+from tool_handler import run_tool  # Use PySpur-integrated run_tool function
 
 # Load vector DB from ZIP on startup
 import embedding  
@@ -80,9 +80,9 @@ with st.container():
             start_time = time.time()
             st.session_state.chat_history.append({"user": user_query})
 
-            # Use PySpur agent to handle routing
+            # Use PySpur-compatible tool runner
             try:
-                response = agent.run(user_query, model=selected_model, use_rag=rag_toggle)
+                response = run_tool(user_query, model=selected_model, use_rag=rag_toggle)
                 routed_correctly = True
             except Exception as e:
                 response = f"Error handling query: {str(e)}"