Update app.py

app.py

@@ -229,31 +229,34 @@ def create_app():
     # Helper: run_agent runs the agent with the given prompt.
     # -------------------------------------------------------------------------
     def run_agent(prompt, socketio):
-        global agent_app
-        if agent_app is None:
-            socketio.emit("log", {"message": "[ERROR]: No database has been uploaded. Upload a database file first."})
-            flash(f"[ERROR]: No database has been uploaded. Upload a database file first.", "error")
-            socketio.emit("final", {"message": "No database available. Upload one and try again."})
+        global agent_app, abs_file_path
+        if not abs_file_path:
+            socketio.emit("log", {"message": "[ERROR]: No DB file uploaded."})
+            socketio.emit("final", {"message": "No database available. Please upload one and try again."})
             return
+    
         try:
-            abs_file_path = os.path.abspath(db_path)
-            agent_app = create_agent_app(abs_file_path)
-            
+            # Lazy agent init
+            if agent_app is None:
+                print("[INFO]: Initializing agent for the first time...")
+                agent_app = create_agent_app(abs_file_path)
+                socketio.emit("log", {"message": "[INFO]: Agent initialized."})
+    
             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."
-                flash(f"[ERROR]: Query failed or no valid answer found.", "error")
+    
             print("final_answer------>", result)
             socketio.emit("final", {"message": result})
         except Exception as e:
             print(f"[ERROR]: {str(e)}")
-            flash(f"[ERROR]: {str(e)}", "error")
             socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
             socketio.emit("final", {"message": "Generation failed."})
 
+
     # -------------------------------------------------------------------------
     # Route: index page
     # -------------------------------------------------------------------------
@@ -286,28 +289,23 @@ def create_app():
     # -------------------------------------------------------------------------
     @flask_app.route("/upload", methods=["GET", "POST"])
     def upload():
-        global abs_file_path, agent_app, db_path
+        global abs_file_path, db_path
         try:
             if request.method == "POST":
                 file = request.files.get("file")
                 if not file:
-                    print("No file uploaded")
                     return "No file uploaded", 400
-                # Secure the filename to avoid path traversal issues.
                 filename = secure_filename(file.filename)
                 if filename.endswith('.db'):
                     db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], "uploaded.db")
-                    print("Saving file to:", db_path)
                     file.save(db_path)
-                    #abs_file_path = os.path.abspath(db_path)
-                    #agent_app = create_agent_app(abs_file_path)
-                    print(f"[INFO]: Database file '{filename}' uploaded and loaded.")
-                    socketio.emit("log", {"message": f"[INFO]: Database file '{filename}' uploaded and loaded."})
+                    abs_file_path = os.path.abspath(db_path)  # Save it here, don't create agent
+                    print(f"[INFO]: File '{filename}' uploaded. Agent will be initialized on first query.")
+                    socketio.emit("log", {"message": f"[INFO]: Database file '{filename}' uploaded."})
                     return redirect(url_for("index"))
             return render_template("upload.html")
         except Exception as e:
             print(f"[ERROR]: {str(e)}")
-            flash(f"[ERROR]: {str(e)}", "error")
             socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
             return render_template("upload.html")