Update app.py

app.py

@@ -13,7 +13,7 @@ from langchain.agents import Tool
 from langchain.agents.format_scratchpad import format_log_to_str
 from langchain.agents.output_parsers import ReActJsonSingleInputOutputParser
 from langchain_core.callbacks import CallbackManager, BaseCallbackHandler
-from langchain_community.agent_toolkits.load_tools import load_tools 
+from langchain_community.agent_toolkits.load_tools import load_tools
 from langchain_core.tools import tool
 from langchain_community.agent_toolkits import SQLDatabaseToolkit
 from langchain.chains import LLMMathChain
@@ -255,7 +255,7 @@ def create_app():
                     with app.app_context():
                         flash(error_message, "error")
                     return
-    
+
             query = {"messages": [("user", prompt)]}
             result = agent_app.invoke(query)
             try:
@@ -274,7 +274,7 @@ def create_app():
                 result = f"An error occurred while processing the agent's response: {str(e)}"
                 with app.app_context():
                     flash("Error processing agent response.", "error")
-    
+
             print("final_answer------>", result)
             socketio.emit("final", {"message": result})
         except Exception as e:
@@ -299,12 +299,14 @@ def create_app():
             thread = threading.Thread(target=run_agent, args=(prompt, socketio))
             socketio.emit("log", {"message": f"[INFO]: Starting thread: {thread}"})
             thread.start()
-            flash("Query submitted successfully.", "info")
+            with flask_app.app_context():
+                flash("Query submitted successfully.", "info")
             return "OK", 200
         except Exception as e:
             error_message = f"[ERROR]: {str(e)}"
             socketio.emit("log", {"message": error_message})
-            flash(error_message, "error")
+            with flask_app.app_context():
+                flash(error_message, "error")
             return "ERROR", 500
 
     def is_sqlite_db(file_path):