Update app.py

app.py

@@ -207,25 +207,7 @@ def create_agent_app(db_path: str):
     # Compile and return the agent application workflow.
     return workflow.compile()
 
-###############################################################################
-# Helper function to run the agent; uses the global agent_app.
-###############################################################################
-def run_agent(prompt, socketio):
-    try:
-        query = {"messages": [("user", prompt)]}
-        result = agent_app.invoke(query)
-        try:
-            result = result["messages"][-1].tool_calls[0]["args"]["final_answer"]
-        except Exception:
-            result = "Query failed or no valid answer found."
-
-        print("final_answer------>", result)
-        socketio.emit("final", {"message": f"{result}"})
-    except Exception as e:
-        socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
-        socketio.emit("final", {"message": "Generation failed."})
-
-###############################################################################
+    ###############################################################################
 # Application Factory: create_app()
 #
 # This function sets up the Flask application, SocketIO, routes, and initializes
@@ -243,8 +225,23 @@ def create_app():
     flask_app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
     # Create a global agent_app using the default DATABASE_URI
-    #global agent_app
-    #agent_app = create_agent_app(DATABASE_URI)
+    ###############################################################################
+    # Helper function to run the agent; uses the global agent_app.
+    ###############################################################################
+    def run_agent(prompt, socketio):
+        try:
+            query = {"messages": [("user", prompt)]}
+            result = agent_app.invoke(query)
+            try:
+                result = result["messages"][-1].tool_calls[0]["args"]["final_answer"]
+            except Exception:
+                result = "Query failed or no valid answer found."
+    
+            print("final_answer------>", result)
+            socketio.emit("final", {"message": f"{result}"})
+        except Exception as e:
+            socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
+            socketio.emit("final", {"message": "Generation failed."})
 
     @flask_app.route("/")
     def index():
@@ -276,7 +273,8 @@ def create_app():
             abs_file_path = os.path.abspath(db_path)
             global agent_app
             agent_app = create_agent_app(abs_file_path)
-    
+            print("agent app",agent_app)
+            print("[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
             socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
             return redirect(url_for("index"))  # Go back to index page
         return render_template("upload.html")