Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, redirect, url_for
+from flask import Flask, render_template, request, redirect, url_for, flash
 from flask_socketio import SocketIO
 import os
 from dotenv import load_dotenv
@@ -114,7 +114,8 @@ def create_agent_app(db_path: str):
     workflow.add_node("first_tool_call", first_tool_call)
     workflow.add_node("list_tables_tool", create_tool_node_with_fallback([list_tool]))
     workflow.add_node("get_schema_tool", create_tool_node_with_fallback([schema_tool]))
-    workflow.add_node("model_get_schema", lambda s: {"messages": [model_get_schema.invoke(s["messages\])]})
+    # Corrected the unterminated string literal in the lambda function below:
+    workflow.add_node("model_get_schema", lambda s: {"messages": [model_get_schema.invoke(s["messages"])]})
     workflow.add_node("query_gen", query_gen_node)
     workflow.add_node("correct_query", model_check_query)
     workflow.add_node("execute_query", create_tool_node_with_fallback([db_query_tool]))
@@ -148,7 +149,7 @@ def upload():
                 save_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], "uploaded.db")
                 file.save(save_path)
                 abs_file_path = os.path.abspath(save_path)
-                agent_app = None
+                agent_app = None  # Reset agent; reinitialize on next query.
                 socketio.emit("log", {"message": f"Database '{filename}' uploaded."})
                 return redirect(url_for("index"))
         return render_template("upload.html")
@@ -185,7 +186,8 @@ def run_agent(prompt):
         socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
         socketio.emit("final", {"message": "Generation failed."})
 
+# Expose the Flask app as "app" for Gunicorn
 app = flask_app
 
 if __name__ == "__main__":
-    socketio.run(app, debug=True)
+    socketio.run(app, debug=True)