Update app.py

app.py

@@ -6,19 +6,20 @@ from flask import Flask, request, jsonify, render_template, session, send_from_d
 from flask_session import Session
 from dotenv import load_dotenv
 
-# --- API and App Configuration ---
+# Load environment variables
 load_dotenv()
 
+# Initialize Flask app
 app = Flask(__name__)
 app.config["SECRET_KEY"] = os.environ.get("FLASK_SECRET_KEY", 'default-insecure-secret-key')
 
-# --- SERVER-SIDE SESSION CONFIGURATION ---
+# Configure server-side sessions
 app.config["SESSION_PERMANENT"] = False
 app.config["SESSION_TYPE"] = "filesystem"
 app.config['SESSION_FILE_DIR'] = './flask_session'
 Session(app)
 
-# Configure the Gemini API client
+# Configure Gemini AI
 try:
     gemini_api_key = os.environ.get("GEMINI_API_KEY")
     if not gemini_api_key:
@@ -26,24 +27,24 @@ try:
         model = None
     else:
         genai.configure(api_key=gemini_api_key)
-        model = genai.GenerativeModel('gemini-2.0-flash')
+        model = genai.GenerativeModel('gemini-1.5-flash')
 except Exception as e:
     print(f"Error configuring Gemini API: {e}")
     model = None
 
-# Configure the upload folder
+# Configure upload folder
 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')
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+os.makedirs('./flask_session', exist_ok=True)
 
 def get_project_files(project_path):
+    """Get all files in a project directory with their contents."""
     file_data = {}
     for root, _, files in os.walk(project_path):
         for file in files:
-            if file in ['.DS_Store', '__MACOSX']: continue
+            if file in ['.DS_Store', '__MACOSX']:
+                continue
             file_path = os.path.join(root, file)
             relative_path = os.path.relpath(file_path, project_path)
             try:
@@ -56,6 +57,7 @@ def get_project_files(project_path):
 
 @app.route('/')
 def index():
+    """Render the main application page."""
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
@@ -63,6 +65,7 @@ def index():
 
 @app.route('/upload', methods=['POST'])
 def upload_project():
+    """Handle project zip file upload."""
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
@@ -78,7 +81,7 @@ def upload_project():
         project_id = session['project_id']
         project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
         
-        # Clear existing project files if they exist
+        # Clear existing project files
         if os.path.exists(project_path):
             for root, dirs, files in os.walk(project_path, topdown=False):
                 for name in files:
@@ -96,30 +99,32 @@ def upload_project():
 
         project_files = get_project_files(project_path)
 
+        # Prepare initial context for AI
         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"
 
+        # Update session with chat history
         session['chat_history'] = [
             {"role": "user", "parts": [initial_context]},
             {"role": "model", "parts": ["I have analyzed the project. I am ready to help. How can I assist you with your code?"]}
         ]
         session.modified = True
 
-        frontend_chat_history = [
-            {"role": "user", "content": "Project context sent to AI."},
-            {"role": "assistant", "content": "I have analyzed the project. I am ready to help. How can I assist you with your code?"}
-        ]
-
+        # Prepare response for frontend
         return jsonify({
             "message": "Project uploaded and analyzed.",
             "file_tree": list(project_files.keys()),
-            "chat_history": frontend_chat_history
+            "chat_history": [
+                {"role": "user", "content": "Project context sent to AI."},
+                {"role": "assistant", "content": "I have analyzed the project. I am ready to help. How can I assist you with your code?"}
+            ]
         })
     return jsonify({"error": "Invalid file type. Please upload a .zip file."}), 400
 
 @app.route('/chat', methods=['POST'])
 def chat():
+    """Handle chat messages with the AI."""
     if not model:
         return jsonify({"error": "Gemini API is not configured. Please check your API key."}), 500
 
@@ -129,7 +134,7 @@ def chat():
 
     project_id = session.get('project_id')
     if not project_id:
-        return jsonify({"error": "Your session has expired or could not be found. Please upload your project again."}), 400
+        return jsonify({"error": "Session expired. Please upload your project again."}), 400
 
     project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
     session['chat_history'].append({"role": "user", "parts": [user_message]})
@@ -157,7 +162,7 @@ def chat():
                     os.makedirs(os.path.dirname(full_path), exist_ok=True)
                     with open(full_path, 'w', encoding='utf-8') as f:
                         f.write(code_content)
-                except IndexError:
+                except Exception:
                     pass
         elif ai_response.strip().startswith("DELETE:"):
             file_to_delete = ai_response.strip().split(":", 1)[1].strip()
@@ -174,9 +179,10 @@ def chat():
 
 @app.route('/file_tree')
 def get_file_tree():
+    """Get the current project's file tree."""
     project_id = session.get('project_id')
     if not project_id:
-        return jsonify({"error": "No project in session. Please upload a project first."}), 400
+        return jsonify({"file_tree": []})
     
     project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
     if not os.path.exists(project_path):
@@ -186,17 +192,16 @@ def get_file_tree():
         project_files = get_project_files(project_path)
         return jsonify({"file_tree": list(project_files.keys())})
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        return jsonify({"file_tree": []})
 
 @app.route('/file_content')
 def get_file_content():
+    """Get the content of a specific file."""
     project_id = session.get('project_id')
     file_path = request.args.get('path')
     
-    if not project_id:
-        return jsonify({"error": "No project in session. Please upload a project first."}), 400
-    if not file_path:
-        return jsonify({"error": "Missing file path"}), 400
+    if not project_id or not file_path:
+        return jsonify({"error": "No file selected"}), 400
     
     full_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id, file_path)
     if not os.path.exists(full_path):
@@ -211,6 +216,7 @@ def get_file_content():
 
 @app.route('/download')
 def download_project():
+    """Download the current project as a zip file."""
     project_id = session.get('project_id')
     if not project_id:
         return "No project to download. Please upload a project first.", 404
@@ -222,7 +228,6 @@ def download_project():
     zip_filename = f"project_{project_id}.zip"
     zip_path = os.path.join(app.config['UPLOAD_FOLDER'], zip_filename)
     
-    # Remove existing zip if it exists
     if os.path.exists(zip_path):
         os.remove(zip_path)
     
@@ -236,4 +241,4 @@ def download_project():
     return send_from_directory(app.config['UPLOAD_FOLDER'], zip_filename, as_attachment=True)
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860, debug=True)