Update app.py

app.py

@@ -246,14 +246,16 @@ def create_app():
                 try:
                     agent_app = create_agent_app(abs_file_path)
                     socketio.emit("log", {"message": "[INFO]: Agent initialized."})
-                    flash("Agent initialized.", "info")
+                    with app.app_context():
+                        flash("Agent initialized.", "info")
                 except Exception as e:
                     error_message = f"Agent initialization failed: {str(e)}"
                     socketio.emit("log", {"message": f"[ERROR]: {error_message}"})
                     socketio.emit("final", {"message": "Agent initialization failed."})
-                    flash(error_message, "error")
+                    with app.app_context():
+                        flash(error_message, "error")
                     return
-
+    
             query = {"messages": [("user", prompt)]}
             result = agent_app.invoke(query)
             try:
@@ -262,21 +264,25 @@ def create_app():
                     result = result["messages"][-1].tool_calls[0]["args"]["final_answer"]
                 else:
                     result = "Query execution did not return a valid final answer."
-                    flash("Query execution did not return a valid final answer.", "warning")
+                    with app.app_context():
+                        flash("Query execution did not return a valid final answer.", "warning")
             except KeyError:
                 result = "The agent's response did not contain the expected 'final_answer' key."
-                flash("Unexpected agent response format.", "warning")
+                with app.app_context():
+                    flash("Unexpected agent response format.", "warning")
             except Exception as e:
                 result = f"An error occurred while processing the agent's response: {str(e)}"
-                flash("Error processing agent response.", "error")
-
+                with app.app_context():
+                    flash("Error processing agent response.", "error")
+    
             print("final_answer------>", result)
             socketio.emit("final", {"message": result})
         except Exception as e:
             error_message = f"Generation failed: {str(e)}"
             socketio.emit("log", {"message": f"[ERROR]: {error_message}"})
             socketio.emit("final", {"message": "Generation failed."})
-            flash(error_message, "error")
+            with app.app_context():
+                flash(error_message, "error")
             traceback.print_exc()
 
     @flask_app.route("/")