Update app.py

app.py

@@ -51,16 +51,23 @@ toolkit = SQLDatabaseToolkit(db=db, llm=llm)
 tools = toolkit.get_tools()
 
 
-def create_agent_app(db_uri: str):
-    # Create new SQLDatabase connection
+def create_agent_app(db_path: str):
+    # Construct the SQLite URI from the given file path.
+    db_uri = f"sqlite:///{db_path}"
+    # Create new SQLDatabase connection using the constructed URI.
     from langchain_community.utilities import SQLDatabase
     db_instance = SQLDatabase.from_uri(db_uri)
 
-    # Create SQL toolkit and get the tools
+    # Create SQL toolkit and get the tools.
     from langchain_community.agent_toolkits import SQLDatabaseToolkit
     toolkit_instance = SQLDatabaseToolkit(db=db_instance, llm=llm)
     tools_instance = toolkit_instance.get_tools()
 
+    # ... (rest of the logic remains unchanged)
+    # [Define custom query tool, prompt templates, workflow nodes, etc.]
+
+    # Compile and return the agent application workflow.
+    return workflow.compile()
     # Define a custom query tool for executing SQL queries
 
 # Define a custom query tool for executing SQL queries
@@ -244,10 +251,9 @@ def create_app():
                 return "No file uploaded", 400
             file_path = os.path.join(UPLOAD_FOLDER, file.filename)
             file.save(file_path)
-            # For SQLite, use the absolute file path in the URI
-            new_db_uri = f"sqlite:///{file_path}"
+            # Reinitialize the agent_app using the uploaded file's path.
             global agent_app
-            agent_app = create_agent_app(new_db_uri)
+            agent_app = create_agent_app(file_path)
             socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
             return redirect(url_for("index"))
         return render_template("upload.html")