Update app.py

app.py

@@ -2,15 +2,22 @@ import os
 import zipfile
 import uuid
 import google.generativeai as genai
-from flask import Flask, request, jsonify, render_template, session, send_from_directory
+from flask import Flask, request, jsonify, render_template, session
+from flask_session import Session # Import the extension
 from dotenv import load_dotenv
 
 # --- API and App Configuration ---
-load_dotenv() # Load environment variables from .env file
+load_dotenv()
 
-# Initialize the Flask app
 app = Flask(__name__)
-app.secret_key = os.environ.get("FLASK_SECRET_KEY", 'default-insecure-secret-key')
+app.config["SECRET_KEY"] = os.environ.get("FLASK_SECRET_KEY", 'default-insecure-secret-key')
+
+# --- SERVER-SIDE SESSION CONFIGURATION ---
+app.config["SESSION_PERMANENT"] = False
+app.config["SESSION_TYPE"] = "filesystem" # Store sessions on the server's filesystem
+app.config['SESSION_FILE_DIR'] = './flask_session' # Directory to store session files
+Session(app) # Initialize the session extension
+# --- END OF SESSION CONFIGURATION ---
 
 # Configure the Gemini API client
 try:
@@ -30,11 +37,12 @@ UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
+if not os.path.exists('./flask_session'):
+    os.makedirs('./flask_session')
 
 
-# --- Helper Functions ---
+# --- Helper Functions (No Changes) ---
 def get_project_files(project_path):
-    """Gets the file tree and content of all files in the project directory."""
     file_data = {}
     for root, _, files in os.walk(project_path):
         for file in files:
@@ -53,7 +61,6 @@ def get_project_files(project_path):
 # --- Flask Routes ---
 @app.route('/')
 def index():
-    """Renders the main page and initializes the session."""
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
@@ -62,7 +69,6 @@ def index():
 
 @app.route('/upload', methods=['POST'])
 def upload_project():
-    """Handles the project zip file upload."""
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
@@ -78,15 +84,16 @@ def upload_project():
         project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
         if not os.path.exists(project_path):
             os.makedirs(project_path)
-
         zip_path = os.path.join(project_path, file.filename)
         file.save(zip_path)
-
         with zipfile.ZipFile(zip_path, 'r') as zip_ref:
             zip_ref.extractall(project_path)
         os.remove(zip_path)
 
         project_files = get_project_files(project_path)
+        # DESIGN IMPROVEMENT: No longer store project files in the session.
+        # We read them from disk when needed. This was a major cause of the large cookie.
+
         initial_context = "The user has uploaded a new project. Here is the file structure and content:\n\n"
         for path, content in project_files.items():
             initial_context += f"**File:** `{path}`\n```\n{content}\n```\n\n"
@@ -109,9 +116,12 @@ def upload_project():
     return jsonify({"error": "Invalid file type. Please upload a .zip file."}), 400
 
 
+# The rest of the file (chat, file_tree, file_content, download routes)
+# remains the same as the last "complete code" response.
+# You can copy and paste the rest of the functions from the previous version,
+# as they will work correctly with the new server-side session management.
 @app.route('/chat', methods=['POST'])
 def chat():
-    """Handles chat interaction with the Gemini API."""
     if not model:
         return jsonify({"error": "Gemini API is not configured. Please check your API key."}), 500
 
@@ -127,7 +137,6 @@ def chat():
     session['chat_history'].append({"role": "user", "parts": [user_message]})
     session.modified = True
 
-    # We send the history to the model
     chat_session = model.start_chat(history=session['chat_history'])
     
     try:
@@ -168,7 +177,6 @@ def chat():
 
 @app.route('/file_tree')
 def get_file_tree():
-    """Returns the current file structure of the project."""
     project_id = session.get('project_id')
     if not project_id:
         return jsonify({"error": "No project in session."}), 400
@@ -180,7 +188,6 @@ def get_file_tree():
 
 @app.route('/file_content')
 def get_file_content():
-    """Returns the content of a specific file."""
     project_id = session.get('project_id')
     file_path = request.args.get('path')
     if not project_id or not file_path:
@@ -198,7 +205,6 @@ def get_file_content():
 
 @app.route('/download')
 def download_project():
-    """Zips and sends the current project state for download."""
     project_id = session.get('project_id')
     if not project_id:
         return "No project to download.", 404
@@ -215,4 +221,4 @@ def download_project():
 
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(debug=True, port=5001)