Update response_modes.py

response_modes.py

@@ -409,7 +409,7 @@ def chat_stream(message, history, current_model, logged_in_username, user_chats,
 
     today_date = datetime.now().strftime("%B %d, %Y")
 
-    # === STRONG CONTEXT HANDLING (works in EVERY mode) ===
+    # Persistent chat context (instructions)
     context_instruction = ""
     if context and str(context).strip():
         ctx = str(context).strip()
@@ -421,93 +421,103 @@ Never mention this rule unless the user explicitly asks about it."""
 
     system_context = [{"role": "system", "content": context_instruction}] if context_instruction else []
 
-    # === CODING MODE: Inject all persistent files (ALWAYS define the variable) ===
-    coding_context = ""   # ← This line is the fix
+    # === CODING MODE: 2-AGENT EXPERIMENT (Coder + Reviewer) ===
     if mode == "💻 Coding Mode":
-        current_chat = None
-        for chat in (user_chats or []):
-            if chat.get("title") == current_chat_title:
-                current_chat = chat
-                break
+        # Gather all persistent files
+        coding_files_str = ""
+        current_chat = next((chat for chat in (user_chats or []) if chat.get("title") == current_chat_title), None)
         if current_chat and current_chat.get("coding_files"):
-            coding_context = "\n\n=== PERSISTENT PROJECT FILES (read these first) ===\n"
+            coding_files_str = "\n\n=== PERSISTENT PROJECT FILES (read these first) ===\n"
             for fname, data in current_chat["coding_files"].items():
-                coding_context += f"\n--- {fname} ({data.get('language','text')}) ---\n{data['content']}\n"
-
-    user_input = quote(message)
-    if file_text:
-        user_input += f"\n\n--- Attached files (up to 5) ---\n{file_text}"
-
-    # Inject persistent coding files (now safe in all modes)
-    if mode == "💻 Coding Mode" and coding_context.strip():
-        user_input = coding_context + "\n\n" + user_input
-
-    history.append({"role": "user", "content": user_input})
-    yield history
-    history.append({"role": "assistant", "content": ""})
-    yield history
-
-    if mode == "Deep Think":
-        # SPECIAL CASE: User just wants the sources from last response
-        lower_msg = message.lower().strip()
-        if any(phrase in lower_msg for phrase in ["show the sources", "show sources", "sources please", "the sources", "list sources"]):
-            # Walk backwards to find the last sources section
-            for i in range(len(history)-2, -1, -1):
-                if "**📌 Key Sources:**" in history[i]["content"]:
-                    sources_only = history[i]["content"].split("**📌 Key Sources:**")[-1].strip()
-                    history[-1]["content"] = f"**📌 Sources from your previous query:**\n\n{sources_only}"
-                    yield history
-                    return  # done
-
-        is_news = any(k in message.lower() for k in ["news", "today", "latest", "breaking", "headlines", "update", "philippines"])
-
-        thinking_steps = [
-            "🧠 Thinking step-by-step...", "🔍 Understanding the core question...",
-            "📋 Planning research approach...", "🔎 Gathering real-time data...",
-            "🧩 Analyzing & cross-checking...", "✍️ Synthesizing final answer..."
+                coding_files_str += f"\n--- {fname} ({data.get('language','text')}) ---\n{data['content']}\n"
+
+        user_input = quote(message)
+        if file_text:
+            user_input += f"\n\n--- Attached files ---\n{file_text}"
+
+        full_context = coding_files_str + "\n\nUser request: " + user_input
+
+        # ====================== AGENT 1: CODER ======================
+        history.append({"role": "assistant", "content": "🤖 **Coder Agent** is thinking and writing code..."})
+        yield history
+
+        coder_messages = system_context + [
+            {"role": "user", "content": f"""You are an expert senior software engineer.
+Use the project files above. {full_context}
+
+Write clean, correct, well-commented code or make the requested changes."""}
         ]
 
-        for step in thinking_steps:
-            history[-1]["content"] = step
-            yield history
+        coder_response = client.chat.completions.create(
+            model=current_model,
+            messages=coder_messages,
+            temperature=0.3,
+            max_tokens=4096,
+            stream=False
+        ).choices[0].message.content
+
+        history[-1]["content"] = f"🤖 **Coder Agent**:\n\n{coder_response}"
+        yield history
+
+        # ====================== AGENT 2: REVIEWER ======================
+        history.append({"role": "assistant", "content": "🕵️ **Reviewer Agent** is analyzing the code..."})
+        yield history
+
+        reviewer_prompt = f"""You are a strict senior code reviewer.
+Here is the latest code from the Coder:
 
-        queries = generate_queries(client, current_model, message, today_date)
-        queries = [enhance_query(q) for q in queries]
+{coder_response}
 
-        all_context = ""
-        raw_results = []
+Project files:
+{coding_files_str}
 
-        for i, q in enumerate(queries, 1):
-            history[-1]["content"] = f"🔎 Deep live search {i}/{len(queries)}: {q[:60]}..."
-            yield history
-            results = web_search(q, max_results=120)
+Critique it honestly:
+- Point out bugs, security issues, bad practices, performance problems
+- Suggest concrete improvements
+- Keep it constructive and specific
+- End with "OVERALL: GOOD / NEEDS_WORK" """
 
-            if is_news:
-                junk = ["wikivoyage", "wikipedia", "wikimedia", "tripadvisor", "booking.com", "airbnb", "lonelyplanet"]
-                results = [r for r in results if not any(j in r.lower() for j in junk)]
+        reviewer_response = client_qwen.chat.completions.create(   # ← uses your existing Groq Qwen (fast)
+            model="qwen/qwen3-32b",
+            messages=[{"role": "user", "content": reviewer_prompt}],
+            temperature=0.4,
+            max_completion_tokens=2048,
+            stream=False
+        ).choices[0].message.content
 
-            all_context += f"\n\n=== Search {i}: {q} ===\n" + "\n\n".join(results)
-            raw_results.extend(results[:8])
+        history[-1]["content"] = f"🕵️ **Reviewer Agent**:\n\n{reviewer_response}"
+        yield history
 
-        history[-1]["content"] = "✍️ Synthesizing final answer from live data..."
+        # ====================== AGENT 1: CODER FINAL REFINEMENT ======================
+        history.append({"role": "assistant", "content": "🤖 **Coder Agent** is refining based on feedback..."})
         yield history
 
-        refined_answer = qwen_reasoning(message, all_context, context)
+        final_prompt = f"""Previous code you wrote:
+{coder_response}
 
-        useful_sources = [s for s in raw_results if len(s.strip()) > 40 and "broad search used" not in s.lower()]
-        sources_section = format_sources(useful_sources, max_sources=7)
+Reviewer feedback:
+{reviewer_response}
 
-        full_response = refined_answer
-        if sources_section:
-            full_response += f"\n\n---\n\n**📌 Key Sources:**\n\n{sources_section}"
+Now produce the FINAL polished version.
+Incorporate all valid feedback.
+If everything is good, just output the clean final code with a short "✅ All good after review" note."""
+
+        final_response = client.chat.completions.create(
+            model=current_model,
+            messages=system_context + [{"role": "user", "content": final_prompt}],
+            temperature=0.25,
+            max_tokens=4096,
+            stream=False
+        ).choices[0].message.content
 
-        history[-1]["content"] = full_response
+        history[-1]["content"] = f"🤖 **Final Code (after review)**:\n\n{final_response}"
         yield history
 
-        # Auto-save
+        # Auto-save the conversation
         if logged_in_username and user_chats:
             for chat in user_chats:
                 if chat["title"] == current_chat_title:
                     chat["history"] = history
                     break
-            save_chats(logged_in_username, user_chats, auth_globals)
+            save_chats(logged_in_username, user_chats, auth_globals)
+        return