Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py - FINAL ULTRA-ONLY VERSION (Single AI for Simplicity & Stability)
 import streamlit as st
 import os
 from datetime import datetime
@@ -6,92 +7,28 @@ from minimax_ai import MinimaxAI
 from llm_explanation import MoveExplainer
 from report_generator import generate_match_report
 
-# Page configuration
-st.set_page_config(
-    page_title="Connect Four AI Pro",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+st.set_page_config(page_title="Connect Four AI Pro", layout="wide", initial_sidebar_state="expanded")
 
-# Enhanced CSS
 st.markdown('''
 <style>
-    .board-display {
-        font-family: 'Courier New', monospace;
-        white-space: pre;
-        line-height: 1.4;
-        font-size: 16px;
-        padding: 20px;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        border-radius: 12px;
-        box-shadow: 0 8px 16px rgba(0,0,0,0.2);
-        margin: 0 auto;
-        display: inline-block;
-    }
-    .board-container {
-        text-align: center;
-    }
-    .stats-card {
-        background: white;
-        padding: 15px;
-        border-radius: 10px;
-        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-        margin: 10px 0;
-        border-left: 4px solid #667eea;
-    }
-    .stat-number {
-        font-size: 24px;
-        font-weight: bold;
-        color: #667eea;
-    }
-    .stat-label {
-        font-size: 12px;
-        color: #666;
-        text-transform: uppercase;
-    }
-    .move-card {
-        background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%);
-        color: white;
-        padding: 20px;
-        border-radius: 12px;
-        margin: 10px 0;
-        box-shadow: 0 6px 12px rgba(0,0,0,0.15);
-    }
-    .player-section {
-        background: white;
-        padding: 20px;
-        border-radius: 12px;
-        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
-        margin: 15px 0;
-    }
-    .player1-section {
-        border-left: 5px solid #ef4444;
-    }
-    .player2-section {
-        border-left: 5px solid #f59e0b;
-    }
-    .mode-badge {
-        display: inline-block;
-        padding: 4px 12px;
-        border-radius: 20px;
-        font-size: 12px;
-        font-weight: bold;
-        margin-right: 8px;
-    }
-    .ultra-fast { background: #10b981; color: white; }
-    .balanced { background: #3b82f6; color: white; }
-    .two-player { background: #f59e0b; color: white; }
+    .board-display { font-family: 'Courier New', monospace; white-space: pre; line-height: 1.4; font-size: 16px; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; border-radius: 12px; box-shadow: 0 8px 16px rgba(0,0,0,0.2); margin: 0 auto; display: inline-block; }
+    .board-container { text-align: center; }
+    .stats-card { background: white; padding: 15px; border-radius: 10px; box-shadow: 0 4px 6px rgba(0,0,0,0.1); margin: 10px 0; border-left: 4px solid #667eea; }
+    .stat-number { font-size: 24px; font-weight: bold; color: #667eea; }
+    .stat-label { font-size: 12px; color: #666; text-transform: uppercase; }
+    .move-card { background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%); color: white; padding: 20px; border-radius: 12px; margin: 10px 0; box-shadow: 0 6px 12px rgba(0,0,0,0.15); }
+    .player-section { background: white; padding: 20px; border-radius: 12px; box-shadow: 0 4px 8px rgba(0,0,0,0.1); margin: 15px 0; }
+    .player1-section { border-left: 5px solid #ef4444; }
+    .player2-section { border-left: 5px solid #f59e0b; }
+    .mode-badge { display: inline-block; padding: 4px 12px; border-radius: 20px; font-size: 12px; font-weight: bold; margin-right: 8px; background: #10b981; color: white; }
+    .two-player { background: #f59e0b; }
 </style>
 ''', unsafe_allow_html=True)
 
-# Initialize session state
 if 'game' not in st.session_state:
     st.session_state.game = ConnectFour()
-    st.session_state.ai_mode = "Ultra-Fast"
     st.session_state.game_mode = "vs AI"
-    st.session_state.ai_ultra = MinimaxAI(st.session_state.game, depth=5)
-    st.session_state.ai_balanced = MinimaxAI(st.session_state.game, depth=6)
+    st.session_state.ai = MinimaxAI(st.session_state.game, depth=5)
     
     hf_token = os.getenv('HF_TOKEN')
     st.session_state.explainer = MoveExplainer(hf_token=hf_token) if hf_token else None
@@ -107,23 +44,17 @@ if 'game' not in st.session_state:
     st.session_state.move_count = 0
     st.session_state.current_depth = 5
 
-# Title
 col_title1, col_title2 = st.columns([3, 1])
 with col_title1:
     st.title("🎮 Connect Four AI Pro")
 with col_title2:
     if st.session_state.game_mode == "vs AI":
-        mode_class = "ultra-fast" if st.session_state.ai_mode == "Ultra-Fast" else "balanced"
-        st.markdown(f'<span class="mode-badge {mode_class}">{st.session_state.ai_mode} AI</span>', 
-                   unsafe_allow_html=True)
+        st.markdown(f'<span class="mode-badge">Ultra AI</span>', unsafe_allow_html=True)
     else:
-        st.markdown('<span class="mode-badge two-player">Two Player</span>', 
-                   unsafe_allow_html=True)
+        st.markdown('<span class="mode-badge two-player">Two Player</span>', unsafe_allow_html=True)
 
-# Sidebar
 st.sidebar.header("⚙️ Game Settings")
 
-# Game Mode
 game_mode = st.sidebar.radio("🎮 Game Mode", ["vs AI", "Two Player"], 
                              index=0 if st.session_state.game_mode == "vs AI" else 1)
 
@@ -139,36 +70,15 @@ if game_mode != st.session_state.game_mode:
     st.session_state.move_count = 0
     st.rerun()
 
-# AI Mode
 if st.session_state.game_mode == "vs AI":
-    st.sidebar.markdown("---")
-    st.sidebar.subheader("🤖 AI Difficulty")
-    ai_mode = st.sidebar.radio("Select AI Mode", ["Ultra-Fast ⚡", "Balanced 🎯"],
-                               index=0 if st.session_state.ai_mode == "Ultra-Fast" else 1,
-                               help="Ultra-Fast: ~200ms | Balanced: ~400ms",
-                               key="ai_mode_radio")
-    
-    new_mode = "Ultra-Fast" if "Ultra-Fast" in ai_mode else "Balanced"
-    if new_mode != st.session_state.ai_mode:
-        st.session_state.ai_mode = new_mode
-    
-    st.session_state.current_ai = st.session_state.ai_ultra if st.session_state.ai_mode == "Ultra-Fast" else st.session_state.ai_balanced
-    
-    # Depth slider
     st.sidebar.markdown("---")
     st.sidebar.subheader("🎚️ Search Depth")
-    if st.session_state.ai_mode == "Ultra-Fast":
-        depth = st.sidebar.slider("Depth (Ultra-Fast)", 3, 8, 5)
-        st.session_state.ai_ultra.depth = depth
-        st.session_state.current_depth = depth
-    else:
-        depth = st.sidebar.slider("Depth (Balanced)", 4, 8, 6)
-        st.session_state.ai_balanced.depth = depth
-        st.session_state.current_depth = depth
+    depth = st.sidebar.slider("AI Depth (3-8)", 3, 8, 5, help="Lower = faster, Higher = stronger")
+    st.session_state.ai.depth = depth
+    st.session_state.current_depth = depth
 
 st.sidebar.markdown("---")
 
-# Controls
 col1, col2 = st.sidebar.columns(2)
 with col1:
     if st.button("🔄 New Game", use_container_width=True):
@@ -192,34 +102,35 @@ with col2:
                 st.session_state.game.undo_move(st.session_state.move_history[-1])
                 st.session_state.move_history.pop()
                 st.session_state.move_history_detailed.pop()
-                if st.session_state.game_mode == "Two Player":
-                    if st.session_state.player1_analyses and st.session_state.player2_analyses:
-                        if st.session_state.move_count % 2 == 0:
-                            st.session_state.player2_analyses.pop()
-                        else:
-                            st.session_state.player1_analyses.pop()
+                if st.session_state.game_mode == "Two Player" and st.session_state.player1_analyses and st.session_state.player2_analyses:
+                    if st.session_state.move_count % 2 == 0:
+                        st.session_state.player2_analyses.pop()
+                    else:
+                        st.session_state.player1_analyses.pop()
                 st.session_state.move_count -= 1
         st.session_state.game_over = False
         st.session_state.show_winner_popup = False
         st.rerun()
 
 st.sidebar.markdown("---")
-
-# Game Stats
 st.sidebar.subheader("📊 Game Stats")
 st.sidebar.write(f"**Moves:** {st.session_state.move_count}")
 if st.session_state.game_mode == "Two Player":
     st.sidebar.write(f"**P1:** {len(st.session_state.player1_analyses)}")
     st.sidebar.write(f"**P2:** {len(st.session_state.player2_analyses)}")
 
-# Main area
+if st.session_state.move_history:
+    with st.sidebar.expander("📜 Move History"):
+        for i, col in enumerate(st.session_state.move_history, 1):
+            player = "🔴" if i % 2 == 1 else "🟡"
+            st.write(f"Move {i}: {player} → Col {col}")
+
 col1, col2 = st.columns([2, 3])
 
 with col1:
     st.subheader("🎯 Game Board")
     board_str = st.session_state.game.board_to_string()
-    st.markdown(f'<div class="board-container"><div class="board-display">{board_str}</div></div>', 
-                unsafe_allow_html=True)
+    st.markdown(f'<div class="board-container"><div class="board-display">{board_str}</div></div>', unsafe_allow_html=True)
     
     if not st.session_state.game_over:
         player_name = f"Player {st.session_state.current_player}" if st.session_state.game_mode == "Two Player" else "Your"
@@ -227,20 +138,16 @@ with col1:
         st.markdown(f"### {player_name} Move {player_emoji}")
         
         valid_cols = st.session_state.game.get_valid_moves()
-        
         if valid_cols:
             human_move = st.selectbox("Choose column:", valid_cols, key="move_select")
             
             if st.button("📍 Play Move", use_container_width=True, type="primary"):
                 current_player = st.session_state.current_player if st.session_state.game_mode == "Two Player" else 1
-                
                 st.session_state.game.make_move(human_move, player=current_player)
                 st.session_state.move_history.append(human_move)
                 st.session_state.move_count += 1
                 
-                # Get AI analysis - ALWAYS use ai_balanced for analysis (consistent)
-                ai_analysis = st.session_state.ai_balanced.get_best_move()
-                
+                ai_analysis = st.session_state.ai.get_best_move()
                 move_detail = {
                     'player': current_player,
                     'player_name': f"Player {current_player}" if st.session_state.game_mode == "Two Player" else "Human",
@@ -254,7 +161,6 @@ with col1:
                         'suggested_column': ai_analysis['move'],
                         'score_diff': abs(ai_analysis['score'])
                     }
-                    
                     if current_player == 1:
                         st.session_state.player1_analyses.append(move_detail)
                     else:
@@ -262,7 +168,6 @@ with col1:
                 
                 st.session_state.move_history_detailed.append(move_detail)
                 
-                # Check winner
                 if st.session_state.game.check_winner(current_player):
                     st.session_state.game_over = True
                     st.session_state.winner = f"Player {current_player}" if st.session_state.game_mode == "Two Player" else "Human"
@@ -273,9 +178,8 @@ with col1:
                     st.session_state.show_winner_popup = True
                 else:
                     if st.session_state.game_mode == "vs AI":
-                        with st.spinner(f"🤖 {st.session_state.ai_mode} AI thinking..."):
-                            # Use current_ai for BOTH move AND analysis in vs AI mode
-                            ai_result = st.session_state.current_ai.get_best_move()
+                        with st.spinner("🤖 AI thinking..."):
+                            ai_result = st.session_state.ai.get_best_move()
                         
                         st.session_state.game.make_move(ai_result['move'], player=2)
                         st.session_state.move_history.append(ai_result['move'])
@@ -323,18 +227,12 @@ with col2:
             explanation = {
                 'explanation': f"Column {ai_move['move']} scores {ai_move['score']} points.",
                 'threat_analysis': "; ".join(ai_move['threats']) if ai_move['threats'] else "Strategic move",
-                'key_insight': "Good positioning",
+                'key_insight': "Strong position",
                 'success': False
             }
             has_llm = False
         
-        st.markdown(f'''
-        <div class="move-card">
-            <h3>🎯 AI Played: Column {ai_move['move']}</h3>
-            <p><strong>Score:</strong> {ai_move['score']} | <strong>Mode:</strong> {st.session_state.ai_mode}</p>
-        </div>
-        ''', unsafe_allow_html=True)
-        
+        st.markdown(f'''<div class="move-card"><h3>🎯 AI Played: Column {ai_move['move']}</h3><p><strong>Score:</strong> {ai_move['score']} points</p></div>''', unsafe_allow_html=True)
         st.success(f"💡 **Why This Move:**\n{explanation['explanation']}")
         
         if explanation.get('threat_analysis'):
@@ -347,42 +245,17 @@ with col2:
         
         stat_col1, stat_col2, stat_col3 = st.columns(3)
         with stat_col1:
-            st.markdown(f'''
-            <div class="stats-card">
-                <div class="stat-number">{ai_move["nodes_explored"]:,}</div>
-                <div class="stat-label">Nodes Explored</div>
-            </div>
-            ''', unsafe_allow_html=True)
+            st.markdown(f'<div class="stats-card"><div class="stat-number">{ai_move["nodes_explored"]:,}</div><div class="stat-label">Explored</div></div>', unsafe_allow_html=True)
         with stat_col2:
-            st.markdown(f'''
-            <div class="stats-card">
-                <div class="stat-number">{ai_move["nodes_pruned"]:,}</div>
-                <div class="stat-label">Nodes Pruned</div>
-            </div>
-            ''', unsafe_allow_html=True)
+            st.markdown(f'<div class="stats-card"><div class="stat-number">{ai_move["nodes_pruned"]:,}</div><div class="stat-label">Pruned</div></div>', unsafe_allow_html=True)
         with stat_col3:
-            st.markdown(f'''
-            <div class="stats-card">
-                <div class="stat-number">{ai_move["pruning_efficiency"]:.1f}%</div>
-                <div class="stat-label">Efficiency</div>
-            </div>
-            ''', unsafe_allow_html=True)
+            st.markdown(f'<div class="stats-card"><div class="stat-number">{ai_move["pruning_efficiency"]:.1f}%</div><div class="stat-label">Efficiency</div></div>', unsafe_allow_html=True)
         
         stat_col4, stat_col5 = st.columns(2)
         with stat_col4:
-            st.markdown(f'''
-            <div class="stats-card">
-                <div class="stat-number">{ai_move["depth"]}</div>
-                <div class="stat-label">Search Depth</div>
-            </div>
-            ''', unsafe_allow_html=True)
+            st.markdown(f'<div class="stats-card"><div class="stat-number">{ai_move["depth"]}</div><div class="stat-label">Search Depth</div></div>', unsafe_allow_html=True)
         with stat_col5:
-            st.markdown(f'''
-            <div class="stats-card">
-                <div class="stat-number">{ai_move["time_ms"]:.0f}ms</div>
-                <div class="stat-label">Response Time</div>
-            </div>
-            ''', unsafe_allow_html=True)
+            st.markdown(f'<div class="stats-card"><div class="stat-number">{ai_move["time_ms"]:.0f}ms</div><div class="stat-label">Response</div></div>', unsafe_allow_html=True)
         
         if not has_llm:
             st.info("ℹ️ Using fallback explanation (HF API unavailable)")
@@ -401,29 +274,13 @@ with col2:
             if latest_p1['analysis']['threats']:
                 st.markdown(f"⚔️ **Threats:** {', '.join(latest_p1['analysis']['threats'])}")
             
-            # Show AI analysis text (explanation doesn't exist in Two Player but we can add general analysis)
             st.markdown(f"**Score:** {latest_p1['analysis']['score']} points")
-            
-            # Algorithm Performance
             st.markdown("**Algorithm Analysis:**")
             perf_col1, perf_col2 = st.columns(2)
-
             with perf_col1:
-                st.markdown(f'''
-                <div class="stats-card">
-                    <div class="stat-number">{latest_p1['analysis']['nodes_explored']:,}</div>
-                    <div class="stat-label">Explored</div>
-                </div>
-                ''', unsafe_allow_html=True)
-
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{latest_p1["analysis"]["nodes_explored"]:,}</div><div class="stat-label">Explored</div></div>', unsafe_allow_html=True)
             with perf_col2:
-                st.markdown(f'''
-                <div class="stats-card">
-                    <div class="stat-number">{latest_p1['analysis']['nodes_pruned']:,}</div>
-                    <div class="stat-label">Pruned</div>
-                </div>
-                ''', unsafe_allow_html=True)
-            
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{latest_p1["analysis"]["nodes_pruned"]:,}</div><div class="stat-label">Pruned</div></div>', unsafe_allow_html=True)
             st.markdown('</div>', unsafe_allow_html=True)
         
         if st.session_state.player2_analyses:
@@ -439,36 +296,21 @@ with col2:
             if latest_p2['analysis']['threats']:
                 st.markdown(f"⚔️ **Threats:** {', '.join(latest_p2['analysis']['threats'])}")
             
-            # Show AI analysis text
             st.markdown(f"**Score:** {latest_p2['analysis']['score']} points")
-            
-            # Algorithm Performance
             st.markdown("**Algorithm Analysis:**")
             perf_col1, perf_col2 = st.columns(2)
             with perf_col1:
-                st.markdown(f'''
-                <div class="stats-card">
-                    <div class="stat-number">{latest_p2['analysis']['nodes_explored']:,}</div>
-                    <div class="stat-label">Explored</div>
-                </div>
-                ''', unsafe_allow_html=True)
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{latest_p2["analysis"]["nodes_explored"]:,}</div><div class="stat-label">Explored</div></div>', unsafe_allow_html=True)
             with perf_col2:
-                st.markdown(f'''
-                <div class="stats-card">
-                    <div class="stat-number">{latest_p2['analysis']['nodes_pruned']:,}</div>
-                    <div class="stat-label">Pruned</div>
-                </div>
-                ''', unsafe_allow_html=True)
-            
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{latest_p2["analysis"]["nodes_pruned"]:,}</div><div class="stat-label">Pruned</div></div>', unsafe_allow_html=True)
             st.markdown('</div>', unsafe_allow_html=True)
 
-# Winner popup
 if st.session_state.show_winner_popup and st.session_state.game_over:
     if st.session_state.winner == "Human":
         st.balloons()
         st.success("# 🎉 YOU WON! 🎉")
     elif st.session_state.winner == "AI":
-        st.info(f"# 🤖 {st.session_state.ai_mode} AI WON!")
+        st.info("# 🤖 AI WON!")
     elif "Player" in st.session_state.winner:
         st.balloons()
         st.success(f"# 🎉 {st.session_state.winner.upper()} WON! 🎉")
@@ -477,7 +319,6 @@ if st.session_state.show_winner_popup and st.session_state.game_over:
     
     st.session_state.show_winner_popup = False
 
-# Game over - report download
 if st.session_state.game_over:
     st.markdown("---")
     st.markdown("### 📄 Download Match Report")
@@ -485,7 +326,7 @@ if st.session_state.game_over:
     
     report = generate_match_report(
         st.session_state.game_mode,
-        st.session_state.ai_mode,
+        "Ultra AI",
         st.session_state.winner,
         st.session_state.move_history_detailed,
         st.session_state.game.board_to_string(),