Adding a "best answer"

App.py

@@ -8,6 +8,10 @@ import google.generativeai as genai
 from anthropic import Anthropic
 import openai
 from typing import List, Dict, Any, Optional
+from dotenv import load_dotenv
+
+# Load environment variables from .env file if it exists
+load_dotenv()
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -472,7 +476,28 @@ class PolyThinkOrchestrator:
                 <h3 class="problem-title">Problem Statement</h3>
                 <div class="problem-content">{problem}</div>
             </div>
+        """
+
+        # Add best answer section if there's agreement
+        last_judgment = next((step.get("judgment", "") for step in reversed(history) if "judgment" in step), "")
+        if "AGREEMENT: YES" in last_judgment.upper():
+            # Get the last solutions before agreement
+            last_solutions = next((step["solutions"] for step in reversed(history) if "solutions" in step), None)
+            if last_solutions:
+                report += f"""
+                <div class="best-answer-container agreement">
+                    <h3>Best Answer</h3>
+                    <div class="best-answer-content">
+                        <div class="best-answer-icon">✨</div>
+                        <div class="best-answer-text">
+                            <p><strong>Agreed Solution:</strong> {last_solutions[0]['solution']}</p>
+                            <p><strong>Models:</strong> {last_solutions[0]['model_name']} & {last_solutions[1]['model_name']}</p>
+                        </div>
+                    </div>
+                </div>
+                """
             
+        report += """
             <div class="timeline-container">
         """
 
@@ -877,6 +902,47 @@ def create_polythink_interface():
         box-shadow: 0 6px 15px rgba(0, 0, 0, 0.4) !important;
         background: linear-gradient(45deg, #666666, #888888) !important;
     }
+    
+    /* Best Answer styling */
+    .best-answer-container {
+        background: #1a1a1a;
+        border-radius: 8px;
+        padding: 20px;
+        margin: 20px 0;
+        box-shadow: 0 4px 15px rgba(0, 0, 0, 0.5);
+        border: 1px solid #4CAF50;
+    }
+    
+    .best-answer-container h3 {
+        color: #4CAF50;
+        margin-top: 0;
+        margin-bottom: 15px;
+        font-size: 1.5em;
+    }
+    
+    .best-answer-content {
+        display: flex;
+        align-items: flex-start;
+        gap: 15px;
+    }
+    
+    .best-answer-icon {
+        font-size: 24px;
+        color: #4CAF50;
+    }
+    
+    .best-answer-text {
+        flex: 1;
+    }
+    
+    .best-answer-text p {
+        margin: 5px 0;
+        color: #ffffff;
+    }
+    
+    .best-answer-text strong {
+        color: #4CAF50;
+    }
     """
 
     # Hardcoded model configurations
@@ -899,7 +965,7 @@ def create_polythink_interface():
     }
 
     async def solve_problem(problem: str, max_rounds: int):
-        # Get API keys from environment variables
+        # Get API keys from environment variables or Hugging Face secrets
         api_clients = {}
         
         # Cohere client
@@ -917,6 +983,16 @@ def create_polythink_interface():
         if gemini_key:
             genai.configure(api_key=gemini_key)
             api_clients["gemini"] = genai
+            
+        # Anthropic client
+        anthropic_key = os.getenv("ANTHROPIC_API_KEY")
+        if anthropic_key:
+            api_clients["anthropic"] = Anthropic(api_key=anthropic_key)
+            
+        # OpenAI client
+        openai_key = os.getenv("OPENAI_API_KEY")
+        if openai_key:
+            api_clients["openai"] = openai.OpenAI(api_key=openai_key)
 
         # Check if all required API keys are present
         required_providers = {solver1_config["provider"], solver2_config["provider"], judge_config["provider"]}