Update app.py

app.py

@@ -47,6 +47,7 @@ os.environ["MISTRAL_API_KEY"] = MISTRAL_API_KEY
 # Global variables for dynamic agent and DB file path; initially None.
 agent_app = None
 abs_file_path = None
+db_path = None
 
 # =============================================================================
 # create_agent_app: Given a database path, initialize the agent workflow.
@@ -129,11 +130,12 @@ def create_agent_app(db_path: str):
     DATABASE_URI = abs_db_path_local
     db_uri = f"sqlite:///{abs_db_path_local}"
     print("db_uri", db_uri)
-
+    flash(f"db_uri:{db_uri}", "warning")
     # Create SQLDatabase connection using langchain utility.
     from langchain_community.utilities import SQLDatabase
     db_instance = SQLDatabase.from_uri(db_uri)
     print("db_instance----->", db_instance)
+    flash(f"db_instance:{db_instance}", "warning")
 
     # Create SQL toolkit.
     from langchain_community.agent_toolkits import SQLDatabaseToolkit
@@ -230,19 +232,25 @@ def create_app():
         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."})
             return
         try:
+            abs_file_path = os.path.abspath(db_path)
+            agent_app = create_agent_app(abs_file_path)
+            
             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."})
 
@@ -269,6 +277,7 @@ def create_app():
             return "OK", 200
         except Exception as e:
             print(f"[ERROR]: {str(e)}")
+            flash(f"[ERROR]: {str(e)}", "error")
             socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
             return "ERROR", 500
 
@@ -277,7 +286,7 @@ def create_app():
     # -------------------------------------------------------------------------
     @flask_app.route("/upload", methods=["GET", "POST"])
     def upload():
-        global abs_file_path, agent_app
+        global abs_file_path, agent_app, db_path
         try:
             if request.method == "POST":
                 file = request.files.get("file")
@@ -290,14 +299,15 @@ def create_app():
                     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)
+                    #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."})
                     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")