Update app.py

app.py

@@ -57,6 +57,7 @@ print(traceback.format_exc())
 # =============================================================================
 # create_agent_app: Given a database path, initialize the agent workflow.
 # =============================================================================
+
 def create_agent_app(db_path: str):
     # Use ChatGroq as our LLM here; you can swap to ChatMistralAI if preferred.
     from langchain_groq import ChatGroq
@@ -141,6 +142,7 @@ def create_agent_app(db_path: str):
     # -------------------------------------------------------------------------
     # Update database URI and file path, create SQLDatabase connection.
     # -------------------------------------------------------------------------
+    
     abs_db_path_local = os.path.abspath(db_path)
     global DATABASE_URI
     DATABASE_URI = abs_db_path_local
@@ -157,6 +159,7 @@ def create_agent_app(db_path: str):
     # -------------------------------------------------------------------------
     # Create SQL toolkit.
     # -------------------------------------------------------------------------
+    
     from langchain_community.agent_toolkits import SQLDatabaseToolkit
     toolkit_instance = SQLDatabaseToolkit(db=db_instance, llm=llm)
     tools_instance = toolkit_instance.get_tools()
@@ -164,6 +167,7 @@ def create_agent_app(db_path: str):
     # -------------------------------------------------------------------------
     # 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,6 +209,7 @@ def create_agent_app(db_path: str):
     # -------------------------------------------------------------------------
     # Get tools for listing tables and fetching schema.
     # -------------------------------------------------------------------------
+    
     list_tables_tool = next((tool for tool in tools_instance if tool.name == "sql_db_list_tables"), None)
     get_schema_tool = next((tool for tool in tools_instance if tool.name == "sql_db_schema"), None)
 
@@ -234,6 +239,7 @@ def create_agent_app(db_path: str):
 # =============================================================================
 # create_app: The application factory.
 # =============================================================================
+    
 def create_app():
     flask_app = Flask(__name__, static_url_path='/uploads', static_folder='uploads')
     socketio = SocketIO(flask_app, cors_allowed_origins="*")
@@ -246,6 +252,7 @@ def create_app():
     # -------------------------------------------------------------------------
     # Serve uploaded files via a custom route.
     # -------------------------------------------------------------------------
+    
     @flask_app.route("/files/<path:filename>")
     def uploaded_file(filename):
         return send_from_directory(flask_app.config['UPLOAD_FOLDER'], filename)
@@ -253,6 +260,7 @@ def create_app():
     # -------------------------------------------------------------------------
     # Helper: run_agent runs the agent with the given prompt.
     # -------------------------------------------------------------------------
+    
     def run_agent(prompt, socketio):
         global agent_app, abs_file_path, db_path
         if not abs_file_path:
@@ -284,6 +292,7 @@ def create_app():
     # -------------------------------------------------------------------------
     # Route: index page.
     # -------------------------------------------------------------------------
+    
     @flask_app.route("/")
     def index():
         return render_template("index.html")
@@ -291,6 +300,7 @@ def create_app():
     # -------------------------------------------------------------------------
     # Route: generate (POST) – receives a prompt and runs the agent.
     # -------------------------------------------------------------------------
+    
     @flask_app.route("/generate", methods=["POST"])
     def generate():
         try:
@@ -310,6 +320,7 @@ def create_app():
     # -------------------------------------------------------------------------
     # Route: upload (GET/POST) – handles uploading the SQLite DB file.
     # -------------------------------------------------------------------------
+    
     @flask_app.route("/upload", methods=["GET", "POST"])
     def upload():
         global abs_file_path, agent_app, db_path
@@ -339,6 +350,7 @@ def create_app():
 # =============================================================================
 # Create the app for Gunicorn compatibility.
 # =============================================================================
+
 app, socketio_instance = create_app()
 
 if __name__ == "__main__":