Going Back to Older Version

App.py

@@ -8,7 +8,6 @@ import google.generativeai as genai
 from anthropic import Anthropic
 import openai
 from typing import List, Dict, Any, Optional
-from datetime import datetime
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -488,20 +487,10 @@ class PolyThinkOrchestrator:
                 """
                 
                 for sol in step["solutions"]:
-                    # Fix the backslash issue by using a different approach to escape single quotes
-                    escaped_solution = sol['solution'].replace("'", "'")
                     report += f"""
                     <div class="solution-item">
-                        <div class="solution-header">
-                            <span>{sol['model_name']}</span>
-                            <div class="solution-controls">
-                                <button class="copy-btn" onclick="copyToClipboard('{escaped_solution}')" title="Copy solution">📋</button>
-                                <button class="toggle-btn" onclick="toggleSolution(this)" title="Toggle solution">▼</button>
-                            </div>
-                        </div>
-                        <div class="solution-body" style="display: none;">
-                            {sol['solution']}
-                        </div>
+                        <div class="solution-header">{sol['model_name']}</div>
+                        <div class="solution-body">{sol['solution']}</div>
                     </div>
                     """
                 
@@ -540,20 +529,10 @@ class PolyThinkOrchestrator:
                 """
                 
                 for sol in step["solutions"]:
-                    # Fix the backslash issue by using a different approach to escape single quotes
-                    escaped_solution = sol['solution'].replace("'", "&apos;")
                     report += f"""
                     <div class="solution-item">
-                        <div class="solution-header">
-                            <span>{sol['model_name']}</span>
-                            <div class="solution-controls">
-                                <button class="copy-btn" onclick="copyToClipboard('{escaped_solution}')" title="Copy solution">📋</button>
-                                <button class="toggle-btn" onclick="toggleSolution(this)" title="Toggle solution">▼</button>
-                            </div>
-                        </div>
-                        <div class="solution-body" style="display: none;">
-                            {sol['solution']}
-                        </div>
+                        <div class="solution-header">{sol['model_name']}</div>
+                        <div class="solution-body">{sol['solution']}</div>
                     </div>
                     """
                 
@@ -566,40 +545,18 @@ class PolyThinkOrchestrator:
         last_judgment = next((step.get("judgment", "") for step in reversed(history) if "judgment" in step), "")
         if "AGREEMENT: YES" in last_judgment.upper():
             confidence = "100%" if len(history) == 2 else "80%"
-            # Get the final agreed-upon solution
-            final_solutions = next((step["solutions"] for step in reversed(history) if "solutions" in step), None)
-            if final_solutions:
-                # Extract the numerical answer and reasoning from the first solution
-                solution_parts = final_solutions[0]["solution"].split("\n")
-                answer = solution_parts[0] if solution_parts else final_solutions[0]["solution"]
-                reasoning = solution_parts[1] if len(solution_parts) > 1 else "No additional reasoning provided."
-                
-                report += f"""
-                <div class="conclusion-container agreement">
-                    <h3>Final Agreed Solution</h3>
-                    <div class="conclusion-content">
-                        <div class="conclusion-icon">✅</div>
-                        <div class="conclusion-text">
-                            <p><strong>Answer:</strong> {answer}</p>
-                            <p><strong>Reasoning:</strong> {reasoning}</p>
-                            <p>Confidence level: <strong>{confidence}</strong></p>
-                        </div>
-                    </div>
-                </div>
-                """
-            else:
-                report += f"""
-                <div class="conclusion-container agreement">
-                    <h3>Conclusion</h3>
-                    <div class="conclusion-content">
-                        <div class="conclusion-icon">✅</div>
-                        <div class="conclusion-text">
-                            <p>Models reached <strong>AGREEMENT</strong></p>
-                            <p>Confidence level: <strong>{confidence}</strong></p>
-                        </div>
+            report += f"""
+            <div class="conclusion-container agreement">
+                <h3>Conclusion</h3>
+                <div class="conclusion-content">
+                    <div class="conclusion-icon">✅</div>
+                    <div class="conclusion-text">
+                        <p>Models reached <strong>AGREEMENT</strong></p>
+                        <p>Confidence level: <strong>{confidence}</strong></p>
                     </div>
                 </div>
-                """
+            </div>
+            """
         else:
             report += f"""
             <div class="conclusion-container disagreement">
@@ -617,33 +574,13 @@ class PolyThinkOrchestrator:
         report += """
             </div>
         </div>
-        <script>
-        function copyToClipboard(text) {
-            navigator.clipboard.writeText(text).then(() => {
-                // Show a temporary success message
-                const btn = event.target;
-                const originalText = btn.textContent;
-                btn.textContent = '✓';
-                setTimeout(() => {
-                    btn.textContent = originalText;
-                }, 1000);
-            });
-        }
-        
-        function toggleSolution(btn) {
-            const solutionBody = btn.closest('.solution-item').querySelector('.solution-body');
-            const isHidden = solutionBody.style.display === 'none';
-            solutionBody.style.display = isHidden ? 'block' : 'none';
-            btn.textContent = isHidden ? '▲' : '▼';
-        }
-        </script>
         """
         
         return report
 
 # --- Gradio Interface ---
 def create_polythink_interface():
-    css = """
+    custom_css = """
     /* Reverted to Original Black Theme */
     body {
         background: #000000;
@@ -707,56 +644,13 @@ def create_polythink_interface():
     .step-section strong {
         color: #ffffff;
     }
-    
-    /* Enhanced Status Bar */
     .status-bar {
         background: #1a1a1a;
-        padding: 15px;
+        padding: 10px;
         border-radius: 5px;
         font-size: 1.1em;
         margin-bottom: 20px;
         border-left: 4px solid #666666;
-        position: relative;
-        overflow: hidden;
-    }
-    
-    .status-progress {
-        position: absolute;
-        bottom: 0;
-        left: 0;
-        height: 3px;
-        background: linear-gradient(90deg, #4CAF50, #2196F3);
-        transition: width 0.3s ease;
-    }
-    
-    .status-icon {
-        display: inline-block;
-        margin-right: 10px;
-        font-size: 1.2em;
-    }
-    
-    .status-message {
-        display: inline-block;
-        vertical-align: middle;
-    }
-    
-    .status-stage {
-        display: inline-block;
-        margin-left: 15px;
-        padding: 4px 8px;
-        border-radius: 12px;
-        font-size: 0.9em;
-        background: rgba(255, 255, 255, 0.1);
-    }
-    
-    .status-stage.active {
-        background: rgba(76, 175, 80, 0.2);
-        color: #4CAF50;
-    }
-    
-    .status-stage.completed {
-        background: rgba(33, 150, 243, 0.2);
-        color: #2196F3;
     }
     
     /* Agreement/Disagreement styling */
@@ -891,12 +785,6 @@ def create_polythink_interface():
         border-radius: 5px;
         overflow: hidden;
         border: 1px solid #383838;
-        transition: all 0.3s ease;
-    }
-    
-    .solution-item:hover {
-        transform: translateY(-2px);
-        box-shadow: 0 4px 15px rgba(0, 0, 0, 0.3);
     }
     
     .solution-header {
@@ -905,9 +793,6 @@ def create_polythink_interface():
         font-weight: bold;
         color: #dddddd;
         border-bottom: 1px solid #444444;
-        display: flex;
-        justify-content: space-between;
-        align-items: center;
     }
     
     .solution-body {
@@ -951,41 +836,14 @@ def create_polythink_interface():
         margin: 5px 0;
     }
     
-    /* Enhanced Header styling with animation */
+    /* Header styling */
     .app-header {
-        background: linear-gradient(-45deg, #222222, #333333, #444444, #333333);
-        background-size: 400% 400%;
-        animation: gradient 15s ease infinite;
+        background: linear-gradient(45deg, #222222, #333333);
         padding: 20px;
         border-radius: 10px;
         margin-bottom: 20px;
         box-shadow: 0 4px 10px rgba(0, 0, 0, 0.3);
         border: 1px solid #444444;
-        position: relative;
-        overflow: hidden;
-    }
-    
-    @keyframes gradient {
-        0% {
-            background-position: 0% 50%;
-        }
-        50% {
-            background-position: 100% 50%;
-        }
-        100% {
-            background-position: 0% 50%;
-        }
-    }
-    
-    .app-header::before {
-        content: '';
-        position: absolute;
-        top: 0;
-        left: 0;
-        right: 0;
-        bottom: 0;
-        background: linear-gradient(45deg, rgba(255,255,255,0.1) 0%, rgba(255,255,255,0) 100%);
-        pointer-events: none;
     }
     
     .app-title {
@@ -995,14 +853,12 @@ def create_polythink_interface():
         -webkit-background-clip: text;
         -webkit-text-fill-color: transparent;
         display: inline-block;
-        position: relative;
     }
     
     .app-subtitle {
         font-size: 16px;
         color: #aaaaaa;
         margin: 0;
-        position: relative;
     }
     
     /* Button style */
@@ -1021,194 +877,6 @@ def create_polythink_interface():
         box-shadow: 0 6px 15px rgba(0, 0, 0, 0.4) !important;
         background: linear-gradient(45deg, #666666, #888888) !important;
     }
-    
-    /* Mobile Responsiveness */
-    @media (max-width: 768px) {
-        .solutions-container {
-            flex-direction: column;
-        }
-        
-        .solution-item {
-            min-width: 100%;
-            margin-bottom: 10px;
-        }
-        
-        .app-title {
-            font-size: 24px;
-        }
-        
-        .app-subtitle {
-            font-size: 14px;
-        }
-        
-        .timeline-item {
-            flex-direction: column;
-        }
-        
-        .timeline-marker {
-            margin-bottom: 10px;
-        }
-    }
-    
-    /* Solution Controls */
-    .solution-controls {
-        display: flex;
-        gap: 8px;
-    }
-    
-    .copy-btn, .toggle-btn {
-        background: none;
-        border: none;
-        color: #bbbbbb;
-        cursor: pointer;
-        padding: 4px 8px;
-        border-radius: 4px;
-        transition: all 0.2s ease;
-        font-size: 1.1em;
-    }
-    
-    .copy-btn:hover, .toggle-btn:hover {
-        background: rgba(255, 255, 255, 0.1);
-        color: #ffffff;
-    }
-    
-    .copy-btn:active, .toggle-btn:active {
-        transform: scale(0.95);
-    }
-    
-    /* Solution Item Enhancements */
-    .solution-item {
-        position: relative;
-        overflow: hidden;
-    }
-    
-    .solution-item::before {
-        content: '';
-        position: absolute;
-        top: 0;
-        left: 0;
-        width: 4px;
-        height: 100%;
-        background: linear-gradient(to bottom, #4CAF50, #2196F3);
-        opacity: 0.5;
-    }
-    
-    .solution-body {
-        padding: 15px;
-        background: rgba(0, 0, 0, 0.2);
-        border-top: 1px solid rgba(255, 255, 255, 0.1);
-        transition: all 0.3s ease;
-    }
-    
-    /* Loading States */
-    .loading-skeleton {
-        background: linear-gradient(90deg, var(--card-bg) 25%, var(--border-color) 50%, var(--card-bg) 75%);
-        background-size: 200% 100%;
-        animation: loading 1.5s infinite;
-        border-radius: 8px;
-        margin: 10px 0;
-    }
-    
-    @keyframes loading {
-        0% { background-position: 200% 0; }
-        100% { background-position: -200% 0; }
-    }
-    
-    /* Error States */
-    .error-message {
-        background: rgba(255, 59, 48, 0.1);
-        border: 1px solid rgba(255, 59, 48, 0.2);
-        border-radius: 8px;
-        padding: 15px;
-        margin: 10px 0;
-        color: #ff3b30;
-        font-size: 14px;
-        line-height: 1.5;
-    }
-    
-    .error-message::before {
-        content: "⚠️";
-        margin-right: 8px;
-    }
-    
-    /* Light Theme Error States */
-    body.light-theme .error-message {
-        background: rgba(255, 59, 48, 0.05);
-        border-color: rgba(255, 59, 48, 0.1);
-    }
-    
-    /* Status Bar States */
-    .status-bar {
-        display: flex;
-        align-items: center;
-        gap: 10px;
-        padding: 10px;
-        background: var(--card-bg);
-        border-radius: 8px;
-        margin: 10px 0;
-        font-size: 14px;
-    }
-    
-    .status-icon {
-        font-size: 16px;
-    }
-    
-    .status-message {
-        flex: 1;
-        color: var(--text-color);
-    }
-    
-    .status-stage {
-        padding: 4px 8px;
-        border-radius: 4px;
-        font-size: 12px;
-        background: var(--border-color);
-        color: var(--text-color);
-        opacity: 0.7;
-    }
-    
-    .status-stage.active {
-        background: var(--primary-color);
-        color: white;
-        opacity: 1;
-    }
-    
-    .status-stage.completed {
-        background: var(--success-color);
-        color: white;
-        opacity: 1;
-    }
-    
-    .status-progress {
-        height: 4px;
-        background: var(--primary-color);
-        border-radius: 2px;
-        transition: width 0.3s ease;
-    }
-    
-    /* Light Theme Status Bar */
-    body.light-theme .status-bar {
-        background: white;
-        border: 1px solid var(--border-color);
-    }
-    
-    body.light-theme .status-stage {
-        background: #f0f0f0;
-    }
-    
-    /* Mobile Responsive Status Bar */
-    @media (max-width: 768px) {
-        .status-bar {
-            flex-direction: column;
-            align-items: flex-start;
-            gap: 5px;
-        }
-        
-        .status-stage {
-            font-size: 10px;
-            padding: 2px 4px;
-        }
-    }
     """
 
     # Hardcoded model configurations
@@ -1231,563 +899,164 @@ def create_polythink_interface():
     }
 
     async def solve_problem(problem: str, max_rounds: int):
-        try:
-            # Get API keys from environment variables
-            api_clients = {}
+        # Get API keys from environment variables
+        api_clients = {}
+        
+        # Cohere client
+        cohere_key = os.getenv("COHERE_API_KEY")
+        if cohere_key:
+            api_clients["cohere"] = cohere.Client(cohere_key)
             
-            # Cohere client
-            cohere_key = os.getenv("COHERE_API_KEY")
-            if cohere_key:
-                api_clients["cohere"] = cohere.Client(cohere_key)
-                
-            # Hugging Face client
-            hf_key = os.getenv("HF_API_KEY")
-            if hf_key:
-                api_clients["huggingface"] = hf_key
-                
-            # Gemini client
-            gemini_key = os.getenv("GEMINI_API_KEY")
-            if gemini_key:
-                genai.configure(api_key=gemini_key)
-                api_clients["gemini"] = genai
-
-            # Check if all required API keys are present
-            required_providers = {solver1_config["provider"], solver2_config["provider"], judge_config["provider"]}
-            missing_keys = [p for p in required_providers if p not in api_clients]
-            if missing_keys:
-                yield [
-                    gr.update(value=f"""
-                    <div class="error-message">
-                        Missing API keys for {', '.join(missing_keys)}. Please check your environment variables.
-                    </div>
-                    """, visible=True),
-                    *[gr.update(visible=False) for _ in range(6)],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">❌</span>
-                        <span class="status-message">Missing API keys for {', '.join(missing_keys)}</span>
-                        <div class="status-progress" style="width: 0%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=False)
-                ]
-                return
+        # Hugging Face client
+        hf_key = os.getenv("HF_API_KEY")
+        if hf_key:
+            api_clients["huggingface"] = hf_key
+            
+        # Gemini client
+        gemini_key = os.getenv("GEMINI_API_KEY")
+        if gemini_key:
+            genai.configure(api_key=gemini_key)
+            api_clients["gemini"] = genai
 
-            # Show loading state
+        # Check if all required API keys are present
+        required_providers = {solver1_config["provider"], solver2_config["provider"], judge_config["provider"]}
+        missing_keys = [p for p in required_providers if p not in api_clients]
+        if missing_keys:
             yield [
-                gr.update(value="""
-                <div class="loading-skeleton" style="height: 100px;"></div>
-                """, visible=True),
-                *[gr.update(visible=False) for _ in range(6)],
+                gr.update(value=f"Error: Missing API keys for {', '.join(missing_keys)}", visible=True),
                 gr.update(visible=False),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">⏳</span>
-                    <span class="status-message">Initializing AI team...</span>
-                    <div class="status-progress" style="width: 10%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=False)
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(value=f"### Status: ❌ Missing API keys for {', '.join(missing_keys)}", visible=True)
             ]
+            return
 
-            orchestrator = PolyThinkOrchestrator(solver1_config, solver2_config, judge_config, api_clients)
+        orchestrator = PolyThinkOrchestrator(solver1_config, solver2_config, judge_config, api_clients)
 
-            # Show loading state for initial solutions
-            yield [
-                gr.update(value="""
-                <div class="loading-skeleton" style="height: 150px;"></div>
-                """, visible=True),
-                *[gr.update(visible=False) for _ in range(6)],
-                gr.update(visible=False),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">🤖</span>
-                    <span class="status-message">Generating initial solutions...</span>
-                    <div class="status-progress" style="width: 15%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=False)
-            ]
+        initial_solutions = await orchestrator.get_initial_solutions(problem)
+        initial_content = f"## Initial Solutions\n**Problem:** `{problem}`\n\n**Solutions:**\n- **{initial_solutions[0]['model_name']}**: {initial_solutions[0]['solution']}\n- **{initial_solutions[1]['model_name']}**: {initial_solutions[1]['solution']}"
+        yield [
+            gr.update(value=initial_content, visible=True),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="", visible=False),
+            gr.update(value="### Status: 📋 Initial solutions generated", visible=True)
+        ]
+        await asyncio.sleep(1)
 
-            try:
-                initial_solutions = await orchestrator.get_initial_solutions(problem)
-            except Exception as e:
-                yield [
-                    gr.update(value=f"""
-                    <div class="error-message">
-                        Error generating initial solutions: {str(e)}
-                        <br><br>
-                        This appears to be a server-side issue with the AI APIs. Please try again later.
-                    </div>
-                    """, visible=True),
-                    *[gr.update(visible=False) for _ in range(6)],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">❌</span>
-                        <span class="status-message">Error: Failed to generate initial solutions</span>
-                        <div class="status-progress" style="width: 0%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=False)
-                ]
-                return
+        solutions = initial_solutions
+        history = [{"solutions": initial_solutions}]
+        max_outputs = max(int(max_rounds) * 2, 6)
+        round_outputs = [""] * max_outputs
 
-            initial_content = f"## Initial Solutions\n**Problem:** `{problem}`\n\n**Solutions:**\n- **{initial_solutions[0]['model_name']}**: {initial_solutions[0]['solution']}\n- **{initial_solutions[1]['model_name']}**: {initial_solutions[1]['solution']}"
+        for round_num in range(int(max_rounds)):
+            judgment = await orchestrator.get_judgment(problem, solutions)
+            history.append({"judgment": judgment["judgment"]})
+            
+            is_agreement = "AGREEMENT: YES" in judgment["judgment"].upper()
+            agreement_class = "agreement" if is_agreement else "disagreement"
+            agreement_icon = "✅" if is_agreement else "❌"
+            
+            judgment_content = f"## Round {round_num + 1} Judgment\n**Evaluation:** <div class='{agreement_class}'>{agreement_icon} {judgment['judgment']}</div>"
+            round_outputs[round_num * 2] = judgment_content
             
-            # Show initial solutions
             yield [
                 gr.update(value=initial_content, visible=True),
-                *[gr.update(visible=False) for _ in range(6)],
-                gr.update(visible=False),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">📋</span>
-                    <span class="status-message">Initial solutions generated</span>
-                    <span class="status-stage completed">Initial Solutions</span>
-                    <div class="status-progress" style="width: 20%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=True)
+                gr.update(value=round_outputs[0], visible=bool(round_outputs[0])),
+                gr.update(value=round_outputs[1], visible=bool(round_outputs[1])),
+                gr.update(value=round_outputs[2], visible=bool(round_outputs[2])),
+                gr.update(value=round_outputs[3], visible=bool(round_outputs[3])),
+                gr.update(value=round_outputs[4], visible=bool(round_outputs[4])),
+                gr.update(value=round_outputs[5], visible=bool(round_outputs[5])),
+                gr.update(value="", visible=False),
+                gr.update(value=f"### Status: 🔍 Round {round_num + 1} judgment complete", visible=True)
             ]
             await asyncio.sleep(1)
 
-            solutions = initial_solutions
-            history = [{"solutions": initial_solutions}]
-            round_outputs = [""] * 6
-
-            for round_num in range(int(max_rounds)):
-                # Show loading state for judgment
-                yield [
-                    gr.update(value=initial_content, visible=True),
-                    *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">⚖️</span>
-                        <span class="status-message">Evaluating solutions (Round {round_num + 1})...</span>
-                        <span class="status-stage completed">Initial Solutions</span>
-                        <span class="status-stage active">Judgment {round_num + 1}</span>
-                        <div class="status-progress" style="width: {(round_num * 2 + 1) / (int(max_rounds) * 2) * 100}%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=True)
-                ]
-
-                try:
-                    judgment = await orchestrator.get_judgment(problem, solutions)
-                except Exception as e:
-                    yield [
-                        gr.update(value=initial_content, visible=True),
-                        *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                        gr.update(visible=False),
-                        gr.update(value=f"""
-                        <div class="status-bar">
-                            <span class="status-icon">❌</span>
-                            <span class="status-message">Error: Failed to evaluate solutions</span>
-                            <div class="status-progress" style="width: {(round_num * 2 + 1) / (int(max_rounds) * 2) * 100}%"></div>
-                        </div>
-                        """, visible=True),
-                        gr.update(visible=True)
-                    ]
-                    return
-
-                history.append({"judgment": judgment["judgment"]})
-                
-                is_agreement = "AGREEMENT: YES" in judgment["judgment"].upper()
-                agreement_class = "agreement" if is_agreement else "disagreement"
-                agreement_icon = "✅" if is_agreement else "❌"
-                
-                judgment_content = f"## Round {round_num + 1} Judgment\n**Evaluation:** <div class='{agreement_class}'>{agreement_icon} {judgment['judgment']}</div>"
-                round_outputs[round_num * 2] = judgment_content
-                
-                progress = (round_num * 2 + 1) / (int(max_rounds) * 2) * 100
-                
-                # Show judgment
-                yield [
-                    gr.update(value=initial_content, visible=True),
-                    *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">🔍</span>
-                        <span class="status-message">Round {round_num + 1} judgment complete</span>
-                        <span class="status-stage completed">Initial Solutions</span>
-                        <span class="status-stage completed">Judgment {round_num + 1}</span>
-                        <div class="status-progress" style="width: {progress}%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=True)
-                ]
-                await asyncio.sleep(1)
-
-                if not judgment["reprompt_needed"]:
-                    break
-
-                # Show loading state for revised solutions
-                yield [
-                    gr.update(value=initial_content, visible=True),
-                    *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">🔄</span>
-                        <span class="status-message">Generating revised solutions (Round {round_num + 1})...</span>
-                        <span class="status-stage completed">Initial Solutions</span>
-                        <span class="status-stage completed">Judgment {round_num + 1}</span>
-                        <span class="status-stage active">Revision {round_num + 1}</span>
-                        <div class="status-progress" style="width: {progress}%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=True)
-                ]
-
-                try:
-                    revised_solutions = await orchestrator.get_revised_solutions(problem, solutions, judgment["judgment"])
-                except Exception as e:
-                    yield [
-                        gr.update(value=initial_content, visible=True),
-                        *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                        gr.update(visible=False),
-                        gr.update(value=f"""
-                        <div class="status-bar">
-                            <span class="status-icon">❌</span>
-                            <span class="status-message">Error: Failed to generate revised solutions</span>
-                            <div class="status-progress" style="width: {progress}%"></div>
-                        </div>
-                        """, visible=True),
-                        gr.update(visible=True)
-                    ]
-                    return
-
-                history.append({"solutions": revised_solutions})
-                revision_content = f"## Round {round_num + 1} Revised Solutions\n**Revised Solutions:**\n- **{revised_solutions[0]['model_name']}**: {revised_solutions[0]['solution']}\n- **{revised_solutions[1]['model_name']}**: {revised_solutions[1]['solution']}"
-                round_outputs[round_num * 2 + 1] = revision_content
-                
-                progress = (round_num * 2 + 2) / (int(max_rounds) * 2) * 100
-                
-                # Show revised solutions
-                yield [
-                    gr.update(value=initial_content, visible=True),
-                    *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                    gr.update(visible=False),
-                    gr.update(value=f"""
-                    <div class="status-bar">
-                        <span class="status-icon">🔄</span>
-                        <span class="status-message">Round {round_num + 1} revised solutions generated</span>
-                        <span class="status-stage completed">Initial Solutions</span>
-                        <span class="status-stage completed">Judgment {round_num + 1}</span>
-                        <span class="status-stage completed">Revision {round_num + 1}</span>
-                        <div class="status-progress" style="width: {progress}%"></div>
-                    </div>
-                    """, visible=True),
-                    gr.update(visible=True)
-                ]
-                await asyncio.sleep(1)
-                solutions = revised_solutions
-
-            # Show loading state for final report
-            yield [
-                gr.update(value=initial_content, visible=True),
-                *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                gr.update(value="""
-                <div class="loading-skeleton" style="height: 200px;"></div>
-                """, visible=True),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">📊</span>
-                    <span class="status-message">Generating final report...</span>
-                    <span class="status-stage completed">Initial Solutions</span>
-                    <span class="status-stage completed">All Judgments</span>
-                    <span class="status-stage completed">All Revisions</span>
-                    <div class="status-progress" style="width: 95%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=True)
-            ]
+            if not judgment["reprompt_needed"]:
+                break
 
-            final_report_content = orchestrator.generate_final_report(problem, history)
-            
-            # Show final report
+            revised_solutions = await orchestrator.get_revised_solutions(problem, solutions, judgment["judgment"])
+            history.append({"solutions": revised_solutions})
+            revision_content = f"## Round {round_num + 1} Revised Solutions\n**Revised Solutions:**\n- **{revised_solutions[0]['model_name']}**: {revised_solutions[0]['solution']}\n- **{revised_solutions[1]['model_name']}**: {revised_solutions[1]['solution']}"
+            round_outputs[round_num * 2 + 1] = revision_content
             yield [
                 gr.update(value=initial_content, visible=True),
-                *[gr.update(value=output, visible=bool(output)) for output in round_outputs],
-                gr.update(value=final_report_content, visible=True),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">✨</span>
-                    <span class="status-message">Process complete! Completed {round_num + 1} round(s)</span>
-                    <span class="status-stage completed">Initial Solutions</span>
-                    <span class="status-stage completed">All Judgments</span>
-                    <span class="status-stage completed">All Revisions</span>
-                    <div class="status-progress" style="width: 100%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=True)
-            ]
-
-        except Exception as e:
-            yield [
-                gr.update(value=f"""
-                <div class="error-message">
-                    An unexpected error occurred: {str(e)}
-                    <br><br>
-                    This appears to be a server-side issue with the AI APIs. Please try again later.
-                </div>
-                """, visible=True),
-                *[gr.update(visible=False) for _ in range(6)],
-                gr.update(visible=False),
-                gr.update(value=f"""
-                <div class="status-bar">
-                    <span class="status-icon">❌</span>
-                    <span class="status-message">Error: An unexpected error occurred</span>
-                    <div class="status-progress" style="width: 0%"></div>
-                </div>
-                """, visible=True),
-                gr.update(visible=False)
+                gr.update(value=round_outputs[0], visible=bool(round_outputs[0])),
+                gr.update(value=round_outputs[1], visible=bool(round_outputs[1])),
+                gr.update(value=round_outputs[2], visible=bool(round_outputs[2])),
+                gr.update(value=round_outputs[3], visible=bool(round_outputs[3])),
+                gr.update(value=round_outputs[4], visible=bool(round_outputs[4])),
+                gr.update(value=round_outputs[5], visible=bool(round_outputs[5])),
+                gr.update(value="", visible=False),
+                gr.update(value=f"### Status: 🔄 Round {round_num + 1} revised solutions generated", visible=True)
             ]
+            await asyncio.sleep(1)
+            solutions = revised_solutions
 
-    def export_report(problem: str, report: str) -> None:
-        """Export the problem-solving report to a text file."""
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"polythink_report_{timestamp}.txt"
-        
-        content = f"""PolyThink Problem Solving Report
-Generated: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
-
-Problem:
-{problem}
-
-Report:
-{report}
-"""
-        
-        # Create a download link
-        html = f"""
-        <a href="data:text/plain;charset=utf-8,{content}" 
-           download="{filename}" 
-           style="display: none;" 
-           id="download-link">
-        </a>
-        <script>
-            document.getElementById('download-link').click();
-        </script>
-        """
-        
-        return gr.update(value=html, visible=True)
-    
-    def share_results(problem: str, report: str) -> None:
-        """Share the problem-solving results."""
-        # Create a shareable text
-        share_text = f"""Check out this problem-solving result from PolyThink!
-
-Problem: {problem}
-
-{report}
-
-Generated with PolyThink Alpha
-"""
-        
-        # Try to use the Web Share API if available
-        html = f"""
-        <script>
-            if (navigator.share) {{
-                navigator.share({{
-                    title: 'PolyThink Problem Solving Result',
-                    text: `{share_text}`,
-                }}).catch(console.error);
-            }} else {{
-                // Fallback to copying to clipboard
-                navigator.clipboard.writeText(`{share_text}`).then(() => {{
-                    alert('Results copied to clipboard!');
-                }}).catch(console.error);
-            }}
-        </script>
-        """
-        
-        return gr.update(value=html, visible=True)
+        final_report_content = orchestrator.generate_final_report(problem, history)
+        yield [
+            gr.update(value=initial_content, visible=True),
+            gr.update(value=round_outputs[0], visible=True),
+            gr.update(value=round_outputs[1], visible=bool(round_outputs[1])),
+            gr.update(value=round_outputs[2], visible=bool(round_outputs[2])),
+            gr.update(value=round_outputs[3], visible=bool(round_outputs[3])),
+            gr.update(value=round_outputs[4], visible=bool(round_outputs[4])),
+            gr.update(value=round_outputs[5], visible=bool(round_outputs[5])),
+            gr.update(value=final_report_content, visible=True),
+            gr.update(value=f"### Status: ✨ Process complete! Completed {round_num + 1} round(s)", visible=True)
+        ]
 
-    with gr.Blocks(title="PolyThink Alpha", css=css) as demo:
+    with gr.Blocks(title="PolyThink Alpha", css=custom_css) as demo:
         with gr.Column(elem_classes=["app-header"]):
             gr.Markdown("<h1 class='app-title'>PolyThink Alpha</h1>", show_label=False)
             gr.Markdown("<p class='app-subtitle'>Multi-Agent Problem Solving System</p>", show_label=False)
-            
-            with gr.Row(elem_classes=["header-controls"]):
-                gr.Button("🌙 Theme", elem_classes=["theme-toggle"], show_label=False)
-                gr.Button("❓ Help", elem_classes=["help-button"], show_label=False)
 
         with gr.Row():
             with gr.Column(scale=2):
                 gr.Markdown("### Problem Input")
-                problem_input = gr.Textbox(
-                    label="Problem",
-                    placeholder="Enter your problem here...",
-                    lines=5,
-                    elem_classes=["problem-input"]
-                )
-                
-                with gr.Row():
-                    max_rounds = gr.Slider(
-                        minimum=1,
-                        maximum=5,
-                        value=3,
-                        step=1,
-                        label="Maximum Rounds",
-                        elem_classes=["rounds-slider"]
-                    )
-                    solve_btn = gr.Button("Solve", variant="primary", elem_classes=["solve-button"])
-                
-                status_bar = gr.HTML(
-                    value="",
-                    elem_classes=["status-bar"],
-                    visible=False
-                )
-                
-            with gr.Column(scale=3):
-                with gr.Tabs() as output_tabs:
-                    with gr.TabItem("Process", elem_classes=["process-tab"]):
-                        initial_solutions = gr.Markdown(visible=False)
-                        round_outputs = [gr.Markdown(visible=False) for _ in range(6)]
-                        
-                    with gr.TabItem("Final Report", elem_classes=["report-tab"]):
-                        final_report = gr.Markdown(visible=False)
-                        
-                with gr.Row(elem_classes=["action-buttons"], visible=False) as action_buttons:
-                    export_btn = gr.Button("Export Report", elem_classes=["action-button"])
-                    share_btn = gr.Button("Share Results", elem_classes=["action-button"])
-        
-        def toggle_theme():
-            return """
-            <script>
-            function toggleTheme() {
-                document.body.classList.toggle('light-theme');
-                const themeBtn = document.querySelector('.theme-toggle');
-                themeBtn.textContent = document.body.classList.contains('light-theme') ? '🌞 Theme' : '🌙 Theme';
-                
-                // Save theme preference
-                localStorage.setItem('theme', document.body.classList.contains('light-theme') ? 'light' : 'dark');
-            }
-            
-            // Load saved theme preference
-            const savedTheme = localStorage.getItem('theme');
-            if (savedTheme === 'light') {
-                document.body.classList.add('light-theme');
-                document.querySelector('.theme-toggle').textContent = '🌞 Theme';
-            }
-            
-            document.querySelector('.theme-toggle').onclick = toggleTheme;
-            </script>
-            """
+                problem_input = gr.Textbox(label="Problem", placeholder="e.g., What is 32 + 63?", lines=3)
+                rounds_slider = gr.Slider(2, 6, value=2, step=1, label="Maximum Rounds")
+                solve_button = gr.Button("Solve Problem", elem_classes=["primary-button"])
 
-        def show_help():
-            return """
-            <script>
-            function showHelp() {
-                const modal = document.createElement('div');
-                modal.className = 'help-modal';
-                modal.innerHTML = `
-                    <h3>How to Use PolyThink Alpha</h3>
-                    <ol>
-                        <li>Enter your problem in the input box</li>
-                        <li>Adjust the maximum number of rounds if needed</li>
-                        <li>Click "Solve Problem" to start the process</li>
-                        <li>Watch as the AI team works together to solve your problem</li>
-                        <li>Review the final report and export/share if desired</li>
-                    </ol>
-                    <p>Note: The AI team consists of:</p>
-                    <ul>
-                        <li>Solver 1: Cohere Command R</li>
-                        <li>Solver 2: Llama 3.2 3B Instruct</li>
-                        <li>Judge: Gemini 2.0 Flash Thinking Experimental 01-21</li>
-                    </ul>
-                    <button onclick="closeHelp()" style="
-                        background: linear-gradient(45deg, #2196F3, #4CAF50);
-                        color: white;
-                        border: none;
-                        padding: 10px 20px;
-                        border-radius: 5px;
-                        cursor: pointer;
-                        margin-top: 20px;
-                    ">Close</button>
-                `;
-                
-                const overlay = document.createElement('div');
-                overlay.className = 'modal-overlay';
-                
-                document.body.appendChild(overlay);
-                document.body.appendChild(modal);
-                
-                // Show with animation
-                setTimeout(() => {
-                    overlay.classList.add('active');
-                    modal.classList.add('active');
-                }, 10);
-            }
-            
-            function closeHelp() {
-                const modal = document.querySelector('.help-modal');
-                const overlay = document.querySelector('.modal-overlay');
-                
-                modal.classList.remove('active');
-                overlay.classList.remove('active');
-                
-                setTimeout(() => {
-                    modal.remove();
-                    overlay.remove();
-                }, 300);
-            }
-            
-            document.querySelector('.help-button').onclick = showHelp;
-            
-            // Close modal when clicking overlay
-            document.addEventListener('click', (e) => {
-                if (e.target.classList.contains('modal-overlay')) {
-                    closeHelp();
-                }
-            });
-            
-            // Close modal with Escape key
-            document.addEventListener('keydown', (e) => {
-                if (e.key === 'Escape') {
-                    closeHelp();
-                }
-            });
-            </script>
-            """
+        status_text = gr.Markdown("### Status: Ready", elem_classes=["status-bar"], visible=True)
 
-        solve_btn.click(
-            fn=show_loading_state,
-            outputs=[
-                initial_solutions,
-                *round_outputs,
-                final_report,
-                status_bar,
-                action_buttons
-            ]
-        ).then(
+        with gr.Column():
+            initial_solutions = gr.Markdown(elem_classes=["step-section"], visible=False)
+            round_judgment_1 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            revised_solutions_1 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            round_judgment_2 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            revised_solutions_2 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            round_judgment_3 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            revised_solutions_3 = gr.Markdown(elem_classes=["step-section"], visible=False)
+            final_report = gr.HTML(elem_classes=["final-report"], visible=False)
+
+        solve_button.click(
             fn=solve_problem,
-            inputs=[problem_input, max_rounds],
+            inputs=[
+                problem_input,
+                rounds_slider
+            ],
             outputs=[
                 initial_solutions,
-                *round_outputs,
+                round_judgment_1,
+                revised_solutions_1,
+                round_judgment_2,
+                revised_solutions_2,
+                round_judgment_3,
+                revised_solutions_3,
                 final_report,
-                status_bar,
-                action_buttons
+                status_text
             ]
         )
-        
-        export_btn.click(
-            fn=export_report,
-            inputs=[problem_input, final_report],
-            outputs=None
-        )
-        
-        share_btn.click(
-            fn=share_results,
-            inputs=[problem_input, final_report],
-            outputs=None
-        )
 
     return demo.queue()