Update app.py

app.py

@@ -194,6 +194,32 @@ agent_app = workflow.compile()
 flask_app = Flask(__name__)
 socketio = SocketIO(flask_app, cors_allowed_origins="*")
 
+# Set up an uploads directory
+UPLOAD_FOLDER = os.path.join(os.getcwd(), "uploads")
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
+
+# Create a global agent_app using the default DATABASE_URI
+agent_app = create_agent_app(DATABASE_URI)
+
+# Endpoint for uploading a DB file
+@flask_app.route("/upload", methods=["GET", "POST"])
+def upload():
+    if request.method == "POST":
+        file = request.files.get("file")
+        if not file:
+            return "No file uploaded", 400
+        file_path = os.path.join(UPLOAD_FOLDER, file.filename)
+        file.save(file_path)
+        # Build a new URI (for SQLite, use absolute path)
+        new_db_uri = f"sqlite:///{file_path}"
+        # Reinitialize the agent_app with the new DB
+        global agent_app
+        agent_app = create_agent_app(new_db_uri)
+        socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
+        return redirect(url_for("index"))
+    return render_template("upload.html")
+
 # Function to run the agent in a separate thread
 def run_agent(prompt):
     try:
@@ -205,7 +231,7 @@ def run_agent(prompt):
     except Exception as e:
         socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
         socketio.emit("final", {"message": "Generation failed."})
-        
+
 @flask_app.route("/")
 def index():
     return render_template("index.html")
@@ -220,5 +246,8 @@ def generate():
     thread.start()
     return "OK", 200
 
+# Assign the Flask app to "app" for gunicorn
+app = flask_app
+
 if __name__ == "__main__":
     socketio.run(app, debug=True)