Update app.py

app.py

@@ -219,21 +219,24 @@ def create_agent_app(db_path: str):
 # This function sets up the Flask application, SocketIO, routes, and initializes
 # the global agent_app using the default DATABASE_URI. It returns the Flask app.
 ###############################################################################
+# --- Application Factory ---
 def create_app():
-    flask_app = Flask(__name__)
+    # Option: configure static files from uploads folder as well.
+    flask_app = Flask(__name__, static_url_path='/uploads', static_folder='uploads')
     socketio = SocketIO(flask_app, cors_allowed_origins="*")
+    
+    # Set up uploads folder
+    UPLOAD_FOLDER_LOCAL = os.path.join(os.getcwd(), "uploads")
+    if not os.path.exists(UPLOAD_FOLDER_LOCAL):
+        os.makedirs(UPLOAD_FOLDER_LOCAL)
+    flask_app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER_LOCAL
 
-    # Set up an uploads directory (for DB file uploads)
-    UPLOAD_FOLDER = os.path.join(os.getcwd(), "uploads")
-    if not os.path.exists(UPLOAD_FOLDER):
-        os.makedirs(UPLOAD_FOLDER)
+    # Static route: option if you want a custom route to serve files:
+    @flask_app.route("/files/<path:filename>")
+    def uploaded_file(filename):
+        return send_from_directory(flask_app.config['UPLOAD_FOLDER'], filename)
 
-    flask_app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-    agent_app = None
-    # Create a global agent_app using the default DATABASE_URI
-    ###############################################################################
     # Helper function to run the agent; uses the global agent_app.
-    ###############################################################################
     def run_agent(prompt, socketio):
         global agent_app
         if agent_app is None:
@@ -281,20 +284,20 @@ def create_app():
                     print("No file uploaded")
                     return "No file uploaded", 400
                 if file and file.filename.endswith('.db'):
-                    # Use flask_app.config instead of app.config
+                    # Save file using flask_app.config
                     db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], 'uploaded.db')
                     print("Saving file to:", db_path)
                     file.save(db_path)
                     
-                    # Convert the file path to an absolute path and reinitialize the agent_app
+                    # Reinitialize the agent_app with the new database file
                     abs_file_path = os.path.abspath(db_path)
                     global agent_app
                     agent_app = create_agent_app(abs_file_path)
                     
                     print(f"[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
                     socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
-                    return redirect(url_for("index"))  # Go back to index page
-            # For GET requests, simply render the upload form.
+                    return redirect(url_for("index"))
+            # For GET, render upload form:
             return render_template("upload.html")
         except Exception as e:
             socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
@@ -302,7 +305,7 @@ def create_app():
 
     return flask_app, socketio
 
-# Create the app and assign to "app" for Gunicorn compatibility.
+# Create the app for Gunicorn compatibility.
 app, socketio_instance = create_app()
 
 if __name__ == "__main__":