Update app.py

app.py

@@ -10,12 +10,17 @@ load_dotenv() # Load environment variables from .env file
 
 # Initialize the Flask app
 app = Flask(__name__)
-app.secret_key = os.environ.get("FLASK_SECRET_KEY", 'your_default_secret_key_change_me')
+app.secret_key = os.environ.get("FLASK_SECRET_KEY", 'default-insecure-secret-key')
 
 # Configure the Gemini API client
 try:
-    genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
-    model = genai.GenerativeModel('gemini-1.5-flash') # Or 'gemini-pro'
+    gemini_api_key = os.environ.get("GEMINI_API_KEY")
+    if not gemini_api_key:
+        print("GEMINI_API_KEY environment variable not found.")
+        model = None
+    else:
+        genai.configure(api_key=gemini_api_key)
+        model = genai.GenerativeModel('gemini-1.5-flash')
 except Exception as e:
     print(f"Error configuring Gemini API: {e}")
     model = None
@@ -23,7 +28,6 @@ except Exception as e:
 # Configure the upload folder
 UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
@@ -34,13 +38,11 @@ def get_project_files(project_path):
     file_data = {}
     for root, _, files in os.walk(project_path):
         for file in files:
-            # Skip .DS_Store and other unwanted files
-            if file == '.DS_Store':
-                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:
-                with open(file_path, 'r', encoding='utf-8') as f:
+                with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
                     content = f.read()
                 file_data[relative_path] = content
             except Exception as e:
@@ -61,7 +63,6 @@ def index():
 @app.route('/upload', methods=['POST'])
 def upload_project():
     """Handles the project zip file upload."""
-    # FIX: Ensure session is initialized to prevent KeyError
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
@@ -86,8 +87,6 @@ def upload_project():
         os.remove(zip_path)
 
         project_files = get_project_files(project_path)
-        session['project_files'] = project_files
-
         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"
@@ -107,7 +106,6 @@ def upload_project():
             "file_tree": list(project_files.keys()),
             "chat_history": frontend_chat_history
         })
-
     return jsonify({"error": "Invalid file type. Please upload a .zip file."}), 400
 
 
@@ -115,47 +113,30 @@ def upload_project():
 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
+        return jsonify({"error": "Gemini API is not configured. Please check your API key."}), 500
 
     user_message = request.json.get('message')
     if not user_message:
         return jsonify({"error": "Empty message"}), 400
 
     project_id = session.get('project_id')
-    project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
+    if not project_id:
+        return jsonify({"error": "Your session has expired or could not be found. Please upload your project again."}), 400
 
-    # Add user message to history
+    project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
     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:
-        system_instruction = """You are an expert programmer AI assistant. You must follow these rules:
-1.  When asked to modify a file, you MUST respond with the full file path on the first line, followed by the complete, updated code for that file inside a markdown code block. Example:
-    `src/app.js`
-    ```javascript
-    // new updated javascript code
-    ```
-2.  When asked to create a new file, respond with the new file path and its content in the same format.
-3.  When asked to delete a file, respond ONLY with 'DELETE: [file_path]'.
-4.  For general conversation, just respond naturally."""
-        
-        # The history already provides context. We send the latest user message.
-        # The system instruction is part of the model's configuration or initial prompt.
-        # For simplicity here, we prepend it to guide this specific turn.
-        # A more advanced setup might use the system_instruction parameter in genai.configure
-        prompt_with_instruction = f"{system_instruction}\n\n## Project Context:\n{get_project_files(project_path)}\n\n## Conversation History:\n{session['chat_history']}\n\n## User's Latest Message:\n{user_message}"
-        
-        response = chat_session.send_message(user_message) # Send only the new message
+        response = chat_session.send_message(user_message)
         ai_response = response.text
 
-        # Add AI response to chat history
         session['chat_history'].append({"role": "model", "parts": [ai_response]})
         session.modified = True
 
-        # --- AI File Operation Logic ---
-        # A more robust solution might involve structured JSON output from the LLM
         if "```" in ai_response:
             lines = ai_response.split('\n')
             file_path_line = lines[0].strip()
@@ -164,16 +145,13 @@ def chat():
                 try:
                     content_start = ai_response.find('```')
                     code_block = ai_response[content_start:]
-                    # Extract language and content
                     code_content = '\n'.join(code_block.split('\n')[1:-1])
-
                     full_path = os.path.join(project_path, file_path)
                     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:
-                    pass # Ignore malformed code blocks for now
-
+                    pass
         elif ai_response.strip().startswith("DELETE:"):
             file_to_delete = ai_response.strip().split(":", 1)[1].strip()
             full_path = os.path.join(project_path, file_to_delete)
@@ -184,7 +162,6 @@ def chat():
                     print(f"Error deleting file {full_path}: {e}")
 
         return jsonify({"reply": ai_response})
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -197,9 +174,8 @@ def get_file_tree():
         return jsonify({"error": "No project in session."}), 400
     project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
     if not os.path.exists(project_path):
-        return jsonify({"error": "Project files not found."}), 404
-    file_tree = list(get_project_files(project_path).keys())
-    return jsonify({"file_tree": file_tree})
+        return jsonify({"file_tree": []})
+    return jsonify({"file_tree": list(get_project_files(project_path).keys())})
 
 
 @app.route('/file_content')
@@ -209,12 +185,11 @@ def get_file_content():
     file_path = request.args.get('path')
     if not project_id or not file_path:
         return jsonify({"error": "Missing project or file path"}), 400
-
     full_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id, file_path)
     if not os.path.exists(full_path):
         return jsonify({"error": "File not found"}), 404
     try:
-        with open(full_path, 'r', encoding='utf-8') as f:
+        with open(full_path, 'r', encoding='utf-8', errors='ignore') as f:
             content = f.read()
         return jsonify({"path": file_path, "content": content})
     except Exception as e:
@@ -227,20 +202,17 @@ def download_project():
     project_id = session.get('project_id')
     if not project_id:
         return "No project to download.", 404
-
     project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
     zip_filename = f"project_{project_id}.zip"
     zip_path = os.path.join(app.config['UPLOAD_FOLDER'], zip_filename)
-
     with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
         for root, _, files in os.walk(project_path):
             for file in files:
                 file_path = os.path.join(root, file)
                 arcname = os.path.relpath(file_path, project_path)
                 zipf.write(file_path, arcname)
-
     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(debug=True, port=5001)