Update response_modes.py

response_modes.py

@@ -480,21 +480,23 @@ Write clean, well-commented, production-ready code."""
         history.append({"role": "assistant", "content": "🧠 Cerebras Reviewer is doing deep analysis..."})
         yield history
 
-        reviewer_prompt = f"""You are an extremely strict senior code reviewer with high reasoning ability.
+        reviewer_prompt = f"""You are an extremely strict senior code reviewer.
 
 Code to review:
-{coder_response}
+{coder_response[:3000]}  # ← truncate to avoid overload
 
-Project files:
-{coding_files_str}
+User request:
+{user_input[:500]}
 
-Research:
-{"\n\n".join(research_results[:6]) if research_results else ""}
+Review strictly:
+- Bugs / logic errors
+- Security risks
+- Performance issues
+- Bad practices
+- Suggest fixes
+Then give the FINAL polished code.
 
-Perform a deep review:
-- Find bugs, security issues, performance problems, bad practices
-- Suggest concrete improvements
-- Give the final polished version at the end"""
+If the code is already good, say "✅ Looks solid" and return it unchanged."""
 
         try:
             cerebras_response = cerebras_client.chat.completions.create(
@@ -504,16 +506,16 @@ Perform a deep review:
                 temperature=0.3,
                 max_completion_tokens=8192,
                 reasoning_effort="high"
-            ).choices[0].message.content
+            ).choices[0].message.content or "✅ Reviewer found no major issues."
         except Exception as e:
-            cerebras_response = f"Reviewer error: {str(e)[:300]}"
+            cerebras_response = f"Reviewer failed: {str(e)[:200]}\n\nOriginal code kept."
 
         history[-1]["content"] = f"🧠 **Cerebras Reviewer**:\n\n{cerebras_response}"
-        history[-1]["tokens"] = len(history[-1]["content"]) // 4 + 50
         yield history
 
-        # Final output
-        history.append({"role": "assistant", "content": f"✅ **Final Polished Code**\n\n{cerebras_response}"})
+        # Final output (show review + code)
+        final_text = f"✅ **Final Polished Code**\n\n{cerebras_response}\n\n**Original Coder version for reference:**\n{coder_response}"
+        history.append({"role": "assistant", "content": final_text})
         yield history
 
         # Auto-save