Update app.py

app.py

@@ -1,8 +1,9 @@
-from flask import Flask, render_template, request
+from flask import Flask, render_template, request, redirect, url_for
 from flask_socketio import SocketIO
 import threading
 import os
 from dotenv import load_dotenv
+import sqlite3  # You can keep this import if you use sqlite3 elsewhere
 
 # LangChain and agent imports
 from langchain_community.chat_models.huggingface import ChatHuggingFace  # if needed later
@@ -30,18 +31,19 @@ from langgraph.prebuilt import ToolNode
 # Load environment variables
 load_dotenv()
 
-# Instead of hardcoding the DB URI, get it from an environment variable.
-# This lets you plug in any single DB by changing the DATABASE_URI environment variable.
+# Set up the DB URI using an environment variable.
+# In your .env file, ensure you have:
+# DATABASE_URI=sqlite:///employee.db
 DATABASE_URI = os.getenv("DATABASE_URI", "sqlite:///employee.db") 
 
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-os.environ["GROQ_API_KEY"] =  GROQ_API_KEY
+os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 
 # Use ChatGroq LLM (which does not require a Hugging Face API token)
 from langchain_groq import ChatGroq
 llm = ChatGroq(model="llama3-70b-8192")
 
-# Connect to the provided database URI (works with any single DB)
+# Connect to the provided database URI using SQLDatabase (which expects a URI)
 from langchain_community.utilities import SQLDatabase
 db = SQLDatabase.from_uri(DATABASE_URI)
 
@@ -50,8 +52,6 @@ from langchain_community.agent_toolkits import SQLDatabaseToolkit
 toolkit = SQLDatabaseToolkit(db=db, llm=llm)
 tools = toolkit.get_tools()
 
-
-
 # Define a custom query tool for executing SQL queries
 @tool
 def db_query_tool(query: str) -> str:
@@ -122,7 +122,10 @@ query_gen = query_gen_prompt | llm.bind_tools([SubmitFinalAnswer])
 
 def create_agent_app(db_path: str):
     # Construct the SQLite URI from the given file path.
-    db_uri = f"sqlite:///{db_path}"
+    # Ensure the db_path is absolute so that SQLAlchemy can locate the file.
+    abs_db_path = os.path.abspath(db_path)
+    db_uri = f"sqlite:///{abs_db_path}"
+    
     # Create new SQLDatabase connection using the constructed URI.
     from langchain_community.utilities import SQLDatabase
     db_instance = SQLDatabase.from_uri(db_uri)
@@ -132,9 +135,6 @@ def create_agent_app(db_path: str):
     toolkit_instance = SQLDatabaseToolkit(db=db_instance, llm=llm)
     tools_instance = toolkit_instance.get_tools()
 
-    # ... (rest of the logic remains unchanged)
-        # Define a custom query tool for executing SQL queries
-
     # Define workflow nodes and fallback functions
     def first_tool_call(state: State) -> dict[str, list[AIMessage]]:
         return {"messages": [AIMessage(content="", tool_calls=[{"name": "sql_db_list_tables", "args": {}, "id": "tool_abcd123"}])]}
@@ -205,7 +205,6 @@ def create_agent_app(db_path: str):
     workflow.add_edge("correct_query", "execute_query")
     workflow.add_edge("execute_query", "query_gen")
 
-
     # Compile and return the agent application workflow.
     return workflow.compile()
 
@@ -249,15 +248,15 @@ def create_app():
             return "No file uploaded", 400
         file_path = os.path.join(UPLOAD_FOLDER, file.filename)
         file.save(file_path)
-    
-        # Initialize new agent app using file path
+
+        # Convert the file path to an absolute path and reinitialize the agent_app
+        abs_file_path = os.path.abspath(file_path)
         global agent_app
-        agent_app = create_agent_app(file_path)
+        agent_app = create_agent_app(abs_file_path)
     
         socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
         return redirect(url_for("index"))  # Go back to index page
 
-
     return flask_app, socketio
 
 ###############################################################################
@@ -278,4 +277,4 @@ def run_agent(prompt, socketio):
 app, socketio_instance = create_app()
 
 if __name__ == "__main__":
-    socketio_instance.run(app, debug=True)
+    socketio_instance.run(app, debug=True)