Update app.py

app.py

@@ -229,12 +229,17 @@ def create_app():
         os.makedirs(UPLOAD_FOLDER)
 
     flask_app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
+    agent_app = None
     # Create a global agent_app using the default DATABASE_URI
     ###############################################################################
     # Helper function to run the agent; uses the global agent_app.
     ###############################################################################
     def run_agent(prompt, socketio):
+        global agent_app
+        if agent_app is None:
+            socketio.emit("log", {"message": "[ERROR]: No database has been uploaded. Please upload a database file first."})
+            socketio.emit("final", {"message": "No database available. Upload a database and try again."})
+            return
         try:
             query = {"messages": [("user", prompt)]}
             result = agent_app.invoke(query)
@@ -278,7 +283,7 @@ def create_app():
                 if file and file.filename.endswith('.db'):
                     # Use flask_app.config instead of app.config
                     db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], 'uploaded.db')
-                    print("file uploaded")
+                    print("Saving file to:", db_path)
                     file.save(db_path)
                     
                     # Convert the file path to an absolute path and reinitialize the agent_app
@@ -286,10 +291,8 @@ def create_app():
                     global agent_app
                     agent_app = create_agent_app(abs_file_path)
                     
-                    # Use an f-string properly to log the uploaded file name
                     print(f"[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
                     socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
-                    print("Received file:", file.filename)
                     return redirect(url_for("index"))  # Go back to index page
             # For GET requests, simply render the upload form.
             return render_template("upload.html")