Update app.py

app.py

@@ -9,6 +9,7 @@ import google.generativeai as genai
 # Load environment variables
 load_dotenv()
 
+# Flask app setup
 app = Flask(__name__)
 app.config['SECRET_KEY'] = os.getenv("FLASK_SECRET_KEY", "insecure-key")
 app.config['SESSION_TYPE'] = 'filesystem'
@@ -16,6 +17,7 @@ app.config['SESSION_PERMANENT'] = False
 app.config['SESSION_FILE_DIR'] = './flask_session'
 Session(app)
 
+# Folder setup
 UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@@ -27,7 +29,7 @@ gemini_key = os.getenv("GEMINI_API_KEY")
 if gemini_key:
     try:
         genai.configure(api_key=gemini_key)
-        model = genai.GenerativeModel('gemini-2.0-flash')
+        model = genai.GenerativeModel('gemini-2.0-pro')
     except Exception as e:
         print(f"Gemini setup failed: {e}")
 
@@ -63,42 +65,49 @@ def upload():
     if not file.filename.endswith('.zip'):
         return jsonify({'error': 'Only .zip supported'}), 400
 
+    # Extract project
     project_id = session['project_id']
     project_path = os.path.join(app.config['UPLOAD_FOLDER'], project_id)
-
-    # Clear existing files
     if os.path.exists(project_path):
         for root, dirs, files in os.walk(project_path, topdown=False):
-            for f in files:
-                os.remove(os.path.join(root, f))
-            for d in dirs:
-                os.rmdir(os.path.join(root, d))
+            for f in files: os.remove(os.path.join(root, f))
+            for d in dirs: os.rmdir(os.path.join(root, d))
     os.makedirs(project_path, exist_ok=True)
 
-    # Save and extract zip
     zip_path = os.path.join(project_path, file.filename)
     file.save(zip_path)
     with zipfile.ZipFile(zip_path, 'r') as z:
         z.extractall(project_path)
     os.remove(zip_path)
 
+    # Read and prepare context
     files = get_project_files(project_path)
-    context = "User uploaded project:\n"
+    prompt = "**Analyze this project and give suggestions for improvements, refactoring, or bugs.**\n\n"
     for path, content in files.items():
-        context += f"**File:** `{path}`\n```\n{content[:2000]}\n```\n\n"
+        prompt += f"File: `{path}`\n```\n{content[:1500]}\n```\n\n"
 
+    # Ask Gemini
+    try:
+        ai_response = "Gemini model is not configured."
+        if model:
+            reply = model.generate_content(prompt)
+            ai_response = reply.text
+    except Exception as e:
+        ai_response = f"Error talking to Gemini: {e}"
+
+    # Save chat history
     session['chat_history'] = [
-        {"role": "user", "parts": [context]},
-        {"role": "model", "parts": ["Project loaded. How can I help you?"]}
+        {"role": "user", "parts": [prompt]},
+        {"role": "model", "parts": [ai_response]}
     ]
     session.modified = True
 
     return jsonify({
-        "message": "Project uploaded.",
+        "message": "Project uploaded and analyzed by Gemini.",
         "file_tree": list(files.keys()),
         "chat_history": [
             {"role": "user", "content": "Project uploaded."},
-            {"role": "assistant", "content": "Project loaded. How can I help you?"}
+            {"role": "assistant", "content": ai_response}
         ]
     })
 
@@ -113,6 +122,7 @@ def chat():
     if 'project_id' not in session:
         session['project_id'] = str(uuid.uuid4())
         session['chat_history'] = []
+
     session['chat_history'].append({"role": "user", "parts": [msg]})
     try:
         chat = model.start_chat(history=session['chat_history'])