Update app.py

app.py

@@ -31,6 +31,10 @@ from langgraph.prebuilt import ToolNode
 # Load environment variables
 load_dotenv()
 
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
 # Set up the DB URI using an environment variable.
 # In your .env file, ensure you have:
 # DATABASE_URI=sqlite:///employee.db
@@ -46,16 +50,6 @@ os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 from langchain_groq import ChatGroq
 llm = ChatGroq(model="llama3-70b-8192")
 
-# Connect to the provided database URI using SQLDatabase (which expects a URI)
-#from langchain_community.utilities import SQLDatabase
-#db = SQLDatabase.from_uri(DATABASE_URI)
-
-# Create SQL toolkit and get the tools
-#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:
     """
@@ -63,7 +57,7 @@ def db_query_tool(query: str) -> str:
     If the query is invalid or returns no result, an error message will be returned.
     In case of an error, the user is advised to rewrite the query and try again.
     """
-    result = db.run_no_throw(query)
+    result = db_instance.run_no_throw(query)
     if not result:
         return "Error: Query failed. Please rewrite your query and try again."
     return result
@@ -213,6 +207,24 @@ def create_agent_app(db_path: str):
     # Compile and return the agent application workflow.
     return workflow.compile()
 
+###############################################################################
+# Helper function to run the agent; uses the global agent_app.
+###############################################################################
+def run_agent(prompt, socketio):
+    try:
+        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."
+
+        print("final_answer------>", result)
+        socketio.emit("final", {"message": f"{result}"})
+    except Exception as e:
+        socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
+        socketio.emit("final", {"message": "Generation failed."})
+
 ###############################################################################
 # Application Factory: create_app()
 #
@@ -238,6 +250,7 @@ def create_app():
 
     @flask_app.route("/generate", methods=["POST"])
     def generate():
+        socketio.emit("log", {"message": "[STEP]: Entering query_gen..."})
         data = request.json
         prompt = data.get("prompt", "")
         socketio.emit("log", {"message": f"[INFO]: Received prompt: {prompt}\n"})
@@ -267,20 +280,6 @@ def create_app():
 
     return flask_app, socketio
 
-###############################################################################
-# Helper function to run the agent; uses the global agent_app.
-###############################################################################
-def run_agent(prompt, socketio):
-    try:
-        query = {"messages": [("user", prompt)]}
-        result = agent_app.invoke(query)
-        result = result["messages"][-1].tool_calls[0]["args"]["final_answer"]
-        print("final_answer------>", result)
-        socketio.emit("final", {"message": f"{result}"})
-    except Exception as e:
-        socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
-        socketio.emit("final", {"message": "Generation failed."})
-
 # Create the app and assign to "app" for Gunicorn compatibility.
 app, socketio_instance = create_app()