Upload 5 files

app.py

@@ -6,334 +6,359 @@ from minimax_ai import MinimaxAI
 from llm_explanation import MoveExplainer
 from report_generator import generate_match_report
 
-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"
+)
 
+# Modern Gaming UI Theme with Connect Four Colors
 st.markdown('''
 <style>
-    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
+    /* Import Google Fonts */
+    @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;700;900&family=Rajdhani:wght@300;400;500;600;700&display=swap');
     
+    /* Global Styles */
     * {
-        font-family: 'Inter', sans-serif;
+        font-family: 'Rajdhani', sans-serif !important;
     }
     
+    /* Main Background with Gaming Gradient */
     .stApp {
-        background: #0f172a;
+        background: linear-gradient(135deg, #0f0c29 0%, #302b63 50%, #24243e 100%);
+        background-attachment: fixed;
     }
     
     /* Sidebar Styling */
     [data-testid="stSidebar"] {
-        background: #1e293b;
-        border-right: 1px solid #334155;
+        background: linear-gradient(180deg, #1a1a2e 0%, #16213e 100%);
+        border-right: 2px solid rgba(255, 215, 0, 0.3);
     }
     
-    [data-testid="stSidebar"] * {
-        color: #e2e8f0 !important;
+    [data-testid="stSidebar"] h1, 
+    [data-testid="stSidebar"] h2, 
+    [data-testid="stSidebar"] h3 {
+        color: #FFD700 !important;
+        font-family: 'Orbitron', sans-serif !important;
+        text-shadow: 0 0 10px rgba(255, 215, 0, 0.5);
     }
     
-    [data-testid="stSidebar"] .stRadio > label {
-        color: #f1f5f9 !important;
-        font-weight: 600;
+    [data-testid="stSidebar"] p,
+    [data-testid="stSidebar"] label {
+        color: #E0E0E0 !important;
+        font-weight: 500;
     }
     
-    [data-testid="stSidebar"] h3 {
-        color: #f8fafc !important;
-        font-size: 16px !important;
-        font-weight: 700 !important;
-        margin-bottom: 16px !important;
-    }
-    
-    /* Main Content */
+    /* Main Title */
     h1 {
-        color: #f8fafc !important;
-        font-weight: 700 !important;
-        font-size: 36px !important;
-        margin-bottom: 8px !important;
+        font-family: 'Orbitron', sans-serif !important;
+        font-weight: 900 !important;
+        font-size: 3rem !important;
+        background: linear-gradient(90deg, #FFD700, #FFA500, #FF6347);
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+        text-align: center;
+        text-shadow: 0 0 30px rgba(255, 215, 0, 0.5);
+        margin-bottom: 1rem !important;
+        animation: glow 2s ease-in-out infinite alternate;
     }
     
-    h2 {
-        color: #f1f5f9 !important;
-        font-weight: 600 !important;
-        font-size: 24px !important;
-        margin-top: 24px !important;
-        margin-bottom: 16px !important;
+    @keyframes glow {
+        from {
+            filter: drop-shadow(0 0 5px #FFD700);
+        }
+        to {
+            filter: drop-shadow(0 0 20px #FFD700);
+        }
     }
     
-    h3 {
-        color: #e2e8f0 !important;
-        font-weight: 600 !important;
-        font-size: 18px !important;
-        margin-bottom: 12px !important;
+    /* Subheaders */
+    h2, h3 {
+        font-family: 'Orbitron', sans-serif !important;
+        color: #FFD700 !important;
+        font-weight: 700 !important;
+        text-shadow: 0 0 10px rgba(255, 215, 0, 0.3);
+        border-bottom: 2px solid rgba(255, 215, 0, 0.3);
+        padding-bottom: 0.5rem;
+        margin-top: 1.5rem !important;
     }
     
-    /* Board Container */
+    /* Game Board Container */
     .board-container {
-        background: #1e293b;
-        border-radius: 16px;
-        padding: 24px;
-        margin-bottom: 24px;
-        border: 1px solid #334155;
-    }
-    
-    .board-display {
-        font-family: 'Courier New', monospace;
-        white-space: pre;
-        line-height: 1.5;
-        font-size: 20px;
-        padding: 24px;
-        background: #0f172a;
-        color: #f8fafc;
-        border-radius: 12px;
-        border: 2px solid #475569;
-        display: inline-block;
-        margin: 0 auto;
-    }
-    
-    /* Stats Cards */
-    .stats-card {
-        background: #1e293b;
-        padding: 20px;
-        border-radius: 12px;
-        border: 1px solid #334155;
+        background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
+        border-radius: 20px;
+        padding: 2rem;
+        box-shadow: 0 20px 60px rgba(0, 0, 0, 0.5),
+                    inset 0 0 30px rgba(255, 215, 0, 0.1);
+        border: 3px solid rgba(255, 215, 0, 0.3);
+        margin: 1rem 0;
+    }
+    
+    /* Board Display */
+    pre {
+        background: linear-gradient(135deg, #0a0e27 0%, #1a1f3a 100%) !important;
+        border-radius: 15px !important;
+        padding: 2rem !important;
+        font-size: 2rem !important;
+        line-height: 2.5rem !important;
+        border: 3px solid rgba(255, 215, 0, 0.4) !important;
+        box-shadow: 0 10px 40px rgba(0, 0, 0, 0.6),
+                    inset 0 0 20px rgba(255, 215, 0, 0.05) !important;
+        font-family: 'Courier New', monospace !important;
         text-align: center;
-        margin: 8px 0;
+        overflow-x: auto;
     }
     
-    .stat-number {
-        font-size: 28px;
-        font-weight: 700;
-        color: #60a5fa;
-        display: block;
-        margin-bottom: 4px;
-    }
-    
-    .stat-label {
-        font-size: 11px;
-        font-weight: 600;
-        color: #94a3b8;
+    /* Buttons - Gaming Style */
+    .stButton button {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        color: white !important;
+        font-family: 'Orbitron', sans-serif !important;
+        font-weight: 700 !important;
+        font-size: 1.1rem !important;
+        border: 2px solid rgba(255, 215, 0, 0.5) !important;
+        border-radius: 12px !important;
+        padding: 0.75rem 1.5rem !important;
+        box-shadow: 0 8px 20px rgba(102, 126, 234, 0.4) !important;
+        transition: all 0.3s ease !important;
         text-transform: uppercase;
-        letter-spacing: 0.5px;
-    }
-    
-    /* Player Sections */
-    .player-section {
-        background: #1e293b;
-        padding: 24px;
-        border-radius: 12px;
-        border: 1px solid #334155;
-        margin-bottom: 20px;
+        letter-spacing: 1px;
     }
     
-    .human-section {
-        border-left: 4px solid #3b82f6;
+    .stButton button:hover {
+        background: linear-gradient(135deg, #764ba2 0%, #667eea 100%) !important;
+        transform: translateY(-3px);
+        box-shadow: 0 12px 30px rgba(102, 126, 234, 0.6) !important;
+        border-color: #FFD700 !important;
     }
     
-    .ai-section {
-        border-left: 4px solid #f59e0b;
+    .stButton button:disabled {
+        background: linear-gradient(135deg, #4a4a4a 0%, #2d2d2d 100%) !important;
+        opacity: 0.5 !important;
+        cursor: not-allowed !important;
+        box-shadow: none !important;
     }
     
-    .player1-section {
-        border-left: 4px solid #ef4444;
+    /* Column Buttons - Special Styling */
+    div[data-testid="column"] .stButton button {
+        background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%) !important;
+        width: 100%;
+        height: 60px;
+        font-size: 1.3rem !important;
+        margin: 0.2rem 0;
     }
     
-    .player2-section {
-        border-left: 4px solid #f59e0b;
+    div[data-testid="column"] .stButton button:hover {
+        background: linear-gradient(135deg, #fa709a 0%, #fee140 100%) !important;
+        transform: scale(1.05);
     }
     
-    .section-header {
-        font-size: 16px;
-        font-weight: 700;
-        color: #f1f5f9;
-        margin-bottom: 16px;
-        display: flex;
-        align-items: center;
-        gap: 8px;
+    /* Info Cards */
+    .info-card {
+        background: linear-gradient(135deg, #232526 0%, #414345 100%);
+        border-radius: 15px;
+        padding: 1.5rem;
+        margin: 1rem 0;
+        border: 2px solid rgba(255, 215, 0, 0.3);
+        box-shadow: 0 10px 30px rgba(0, 0, 0, 0.5);
     }
     
-    .info-row {
-        color: #cbd5e1;
-        font-size: 14px;
-        margin: 8px 0;
-        line-height: 1.6;
-    }
-    
-    .info-row strong {
-        color: #f1f5f9;
-        font-weight: 600;
+    /* Stats Display */
+    .metric-container {
+        background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
+        border-radius: 12px;
+        padding: 1rem;
+        margin: 0.5rem 0;
+        border-left: 4px solid #FFD700;
+        box-shadow: 0 5px 15px rgba(0, 0, 0, 0.3);
     }
     
-    /* Badges */
-    .badge {
+    /* AI Badge */
+    .ai-badge {
         display: inline-block;
-        padding: 4px 12px;
-        border-radius: 20px;
-        font-size: 12px;
-        font-weight: 600;
-        margin: 4px 4px 4px 0;
-    }
-    
-    .badge-success {
-        background: #10b981;
+        background: linear-gradient(135deg, #FF6347, #FF4500);
         color: white;
+        padding: 0.5rem 1.5rem;
+        border-radius: 25px;
+        font-family: 'Orbitron', sans-serif;
+        font-weight: 700;
+        font-size: 1.2rem;
+        border: 2px solid rgba(255, 215, 0, 0.5);
+        box-shadow: 0 5px 15px rgba(255, 99, 71, 0.4);
+        text-transform: uppercase;
+        letter-spacing: 2px;
+        animation: pulse 2s infinite;
     }
     
-    .badge-warning {
-        background: #f59e0b;
-        color: white;
+    @keyframes pulse {
+        0%, 100% {
+            transform: scale(1);
+        }
+        50% {
+            transform: scale(1.05);
+        }
     }
     
-    .badge-info {
-        background: #3b82f6;
+    /* Two Player Badge */
+    .two-player-badge {
+        display: inline-block;
+        background: linear-gradient(135deg, #4facfe, #00f2fe);
         color: white;
+        padding: 0.5rem 1.5rem;
+        border-radius: 25px;
+        font-family: 'Orbitron', sans-serif;
+        font-weight: 700;
+        font-size: 1.2rem;
+        border: 2px solid rgba(255, 215, 0, 0.5);
+        box-shadow: 0 5px 15px rgba(79, 172, 254, 0.4);
+        text-transform: uppercase;
+        letter-spacing: 2px;
     }
     
-    .badge-threat {
-        background: #1e293b;
-        color: #94a3b8;
-        border: 1px solid #334155;
-    }
-    
-    /* Move Card */
-    .move-card {
-        background: linear-gradient(135deg, #f59e0b 0%, #d97706 100%);
-        color: white;
-        padding: 20px;
-        border-radius: 12px;
-        margin-bottom: 16px;
+    /* Winner Popup */
+    .winner-popup {
+        position: fixed;
+        top: 50%;
+        left: 50%;
+        transform: translate(-50%, -50%);
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 3rem;
+        border-radius: 20px;
+        border: 4px solid #FFD700;
+        box-shadow: 0 20px 60px rgba(0, 0, 0, 0.8);
+        z-index: 9999;
         text-align: center;
+        animation: popIn 0.5s ease-out;
     }
     
-    .move-card-title {
-        font-size: 18px;
-        font-weight: 700;
-        margin-bottom: 8px;
+    @keyframes popIn {
+        0% {
+            transform: translate(-50%, -50%) scale(0.5);
+            opacity: 0;
+        }
+        100% {
+            transform: translate(-50%, -50%) scale(1);
+            opacity: 1;
+        }
     }
     
-    .move-card-score {
-        font-size: 14px;
-        opacity: 0.9;
-    }
-    
-    /* Winner Banner */
-    .winner-banner {
-        background: linear-gradient(135deg, #10b981 0%, #059669 100%);
-        color: white;
-        padding: 32px;
-        border-radius: 16px;
-        text-align: center;
-        font-size: 32px;
-        font-weight: 700;
-        margin: 24px 0;
-        box-shadow: 0 10px 30px rgba(16, 185, 129, 0.3);
+    /* Expander Styling */
+    .streamlit-expanderHeader {
+        background: linear-gradient(135deg, #232526 0%, #414345 100%) !important;
+        border-radius: 10px !important;
+        border: 1px solid rgba(255, 215, 0, 0.3) !important;
+        color: #FFD700 !important;
+        font-weight: 600 !important;
     }
     
-    .winner-banner.ai {
-        background: linear-gradient(135deg, #f59e0b 0%, #d97706 100%);
-        box-shadow: 0 10px 30px rgba(245, 158, 11, 0.3);
+    /* Radio Buttons */
+    .stRadio label {
+        color: #E0E0E0 !important;
+        font-weight: 600 !important;
+        font-size: 1.1rem !important;
     }
     
-    .winner-banner.draw {
-        background: linear-gradient(135deg, #6366f1 0%, #4f46e5 100%);
-        box-shadow: 0 10px 30px rgba(99, 102, 241, 0.3);
+    /* Slider */
+    .stSlider {
+        padding: 1rem 0;
     }
     
-    /* Buttons */
-    .stButton > button {
-        background: #3b82f6;
-        color: white;
-        border: none;
-        border-radius: 8px;
-        font-weight: 600;
-        padding: 8px 16px;
-        transition: all 0.2s;
+    /* Text & Paragraphs */
+    p, li, span {
+        color: #E0E0E0 !important;
+        font-size: 1.05rem !important;
+        line-height: 1.6 !important;
     }
     
-    .stButton > button:hover {
-        background: #2563eb;
-        box-shadow: 0 4px 12px rgba(59, 130, 246, 0.3);
+    /* Strong/Bold Text */
+    strong {
+        color: #FFD700 !important;
+        font-weight: 700 !important;
     }
     
-    .stButton > button[kind="primary"] {
-        background: #10b981;
+    /* Code blocks */
+    code {
+        background: rgba(255, 215, 0, 0.1) !important;
+        color: #FFD700 !important;
+        padding: 0.2rem 0.5rem !important;
+        border-radius: 5px !important;
     }
     
-    .stButton > button[kind="primary"]:hover {
-        background: #059669;
-        box-shadow: 0 4px 12px rgba(16, 185, 129, 0.3);
+    /* Horizontal Rules */
+    hr {
+        border-color: rgba(255, 215, 0, 0.3) !important;
+        margin: 2rem 0 !important;
     }
     
-    /* Select Box */
-    .stSelectbox > div > div {
-        background: #1e293b;
-        border: 1px solid #334155;
-        color: #f1f5f9;
-        border-radius: 8px;
+    /* Success/Info/Warning/Error Messages */
+    .stSuccess, .stInfo, .stWarning, .stError {
+        border-radius: 10px !important;
+        border-left: 5px solid #FFD700 !important;
     }
     
-    /* Metric boxes */
-    [data-testid="stMetricValue"] {
-        color: #60a5fa !important;
-        font-size: 24px !important;
-        font-weight: 700 !important;
+    /* Scrollbar */
+    ::-webkit-scrollbar {
+        width: 12px;
+        height: 12px;
     }
     
-    [data-testid="stMetricLabel"] {
-        color: #94a3b8 !important;
-        font-size: 13px !important;
+    ::-webkit-scrollbar-track {
+        background: #1a1a2e;
     }
     
-    /* Streamlit native elements */
-    .stAlert {
-        background: #1e293b;
-        border: 1px solid #334155;
-        border-radius: 8px;
-        color: #e2e8f0;
+    ::-webkit-scrollbar-thumb {
+        background: linear-gradient(135deg, #667eea, #764ba2);
+        border-radius: 6px;
     }
     
-    div[data-baseweb="notification"] {
-        background: #1e293b;
-        border: 1px solid #334155;
+    ::-webkit-scrollbar-thumb:hover {
+        background: linear-gradient(135deg, #764ba2, #667eea);
     }
     
-    /* Success/Warning/Info boxes */
-    .stSuccess {
-        background: rgba(16, 185, 129, 0.1) !important;
-        border-left: 4px solid #10b981 !important;
-        color: #a7f3d0 !important;
+    /* Download Button */
+    .stDownloadButton button {
+        background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%) !important;
+        color: white !important;
+        font-family: 'Orbitron', sans-serif !important;
+        font-weight: 700 !important;
+        border: 2px solid rgba(255, 215, 0, 0.5) !important;
+        border-radius: 12px !important;
+        padding: 0.75rem 1.5rem !important;
+        box-shadow: 0 8px 20px rgba(17, 153, 142, 0.4) !important;
+        transition: all 0.3s ease !important;
     }
     
-    .stWarning {
-        background: rgba(245, 158, 11, 0.1) !important;
-        border-left: 4px solid #f59e0b !important;
-        color: #fde68a !important;
+    .stDownloadButton button:hover {
+        background: linear-gradient(135deg, #38ef7d 0%, #11998e 100%) !important;
+        transform: translateY(-3px);
+        box-shadow: 0 12px 30px rgba(17, 153, 142, 0.6) !important;
     }
     
-    .stInfo {
-        background: rgba(59, 130, 246, 0.1) !important;
-        border-left: 4px solid #3b82f6 !important;
-        color: #bfdbfe !important;
+    /* Analysis Cards */
+    .analysis-card {
+        background: linear-gradient(135deg, #2c3e50 0%, #34495e 100%);
+        border-radius: 15px;
+        padding: 1.5rem;
+        margin: 1rem 0;
+        border-left: 5px solid #FFD700;
+        box-shadow: 0 10px 30px rgba(0, 0, 0, 0.4);
     }
     
-    /* Hide Streamlit elements */
-    #MainMenu {visibility: hidden;}
-    footer {visibility: hidden;}
-    
-    /* Expander */
-    .streamlit-expanderHeader {
-        background: #1e293b;
-        color: #f1f5f9 !important;
-        border-radius: 8px;
+    .analysis-card h4 {
+        color: #FFD700 !important;
+        font-family: 'Orbitron', sans-serif !important;
+        margin-bottom: 1rem !important;
     }
 </style>
 ''', unsafe_allow_html=True)
 
+# Initialize Session State
 if 'game' not in st.session_state:
     st.session_state.game = ConnectFour()
     st.session_state.game_mode = "vs AI"
     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
-    
     st.session_state.move_history = []
     st.session_state.move_history_detailed = []
     st.session_state.player1_analyses = []
@@ -348,20 +373,24 @@ if 'game' not in st.session_state:
     st.session_state.last_ai_move_analysis = None
 
 # Header
-col_title1, col_title2 = st.columns([4, 1])
+col_title1, col_title2 = st.columns([3, 1])
 with col_title1:
-    st.title("🎮 Connect Four AI Pro")
+    st.title("🎮 CONNECT FOUR AI PRO")
+    
 with col_title2:
     if st.session_state.game_mode == "vs AI":
-        st.markdown('<span class="badge badge-info">🤖 VS AI</span>', unsafe_allow_html=True)
+        st.markdown(f'<div class="ai-badge">⚡ ULTRA AI</div>', unsafe_allow_html=True)
     else:
-        st.markdown('<span class="badge badge-warning">👥 Two Player</span>', unsafe_allow_html=True)
+        st.markdown('<div class="two-player-badge">👥 TWO PLAYER</div>', unsafe_allow_html=True)
 
-# Sidebar
-st.sidebar.markdown("### ⚙️ Game Settings")
+# Sidebar Configuration
+st.sidebar.header("⚙️ GAME SETTINGS")
 
-game_mode = st.sidebar.radio("🎮 Game Mode", ["vs AI", "Two Player"], 
-                             index=0 if st.session_state.game_mode == "vs AI" else 1)
+game_mode = st.sidebar.radio(
+    "🎮 Game Mode",
+    ["vs AI", "Two Player"],
+    index=0 if st.session_state.game_mode == "vs AI" else 1
+)
 
 if game_mode != st.session_state.game_mode:
     st.session_state.game_mode = game_mode
@@ -377,13 +406,14 @@ if game_mode != st.session_state.game_mode:
 
 if st.session_state.game_mode == "vs AI":
     st.sidebar.markdown("---")
-    st.sidebar.markdown("### 🎚️ AI Difficulty")
-    depth = st.sidebar.slider("Search Depth", 3, 8, 5)
+    st.sidebar.subheader("🎚️ AI Search Depth")
+    depth = st.sidebar.slider("Depth Level (3-8)", 3, 8, 5)
     st.session_state.ai.depth = depth
     st.session_state.current_depth = depth
 
 st.sidebar.markdown("---")
 
+# Control Buttons
 col1, col2 = st.sidebar.columns(2)
 with col1:
     if st.button("🔄 New Game", use_container_width=True):
@@ -419,325 +449,293 @@ with col2:
         st.session_state.show_winner_popup = False
         st.rerun()
 
+# Game Statistics
 st.sidebar.markdown("---")
-st.sidebar.markdown("### 📊 Game Stats")
-st.sidebar.metric("Total Moves", st.session_state.move_count)
+st.sidebar.subheader("📊 GAME STATISTICS")
+st.sidebar.markdown(f'<div class="metric-container"><strong>Total Moves:</strong> {st.session_state.move_count}</div>', unsafe_allow_html=True)
+
 if st.session_state.game_mode == "Two Player":
-    col1, col2 = st.sidebar.columns(2)
-    with col1:
-        st.metric("🔴 P1", len(st.session_state.player1_analyses))
-    with col2:
-        st.metric("🟡 P2", len(st.session_state.player2_analyses))
+    st.sidebar.markdown(f'<div class="metric-container"><strong>Player 1 (🔴):</strong> {len(st.session_state.player1_analyses)} moves</div>', unsafe_allow_html=True)
+    st.sidebar.markdown(f'<div class="metric-container"><strong>Player 2 (🟡):</strong> {len(st.session_state.player2_analyses)} moves</div>', unsafe_allow_html=True)
 
+# Move History
 if st.session_state.move_history:
-    with st.sidebar.expander("📜 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}")
+            st.write(f"**Move {i}:** {player} → Column {col}")
 
-# Main layout
+# Main Game Layout
 if st.session_state.game_mode == "vs AI":
     col1, col2 = st.columns([2, 3])
     
     with col1:
-        st.markdown('<div class="board-container" style="text-align: center;">', unsafe_allow_html=True)
-        st.markdown("### 🎯 Game Board")
+        st.subheader("🎯 GAME BOARD")
         board_str = st.session_state.game.board_to_string()
-        st.markdown(f'<div class="board-display">{board_str}</div>', unsafe_allow_html=True)
-        st.markdown('</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="board-container"><pre>{board_str}</pre></div>', unsafe_allow_html=True)
         
-        if not st.session_state.game_over:
-            st.markdown("### 🔴 Your Turn")
-            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 = 1
-                    st.session_state.game.make_move(human_move, player=current_player)
-                    st.session_state.move_history.append(human_move)
+        # Column Selection
+        st.markdown("### 🎲 SELECT YOUR MOVE")
+        cols = st.columns(7)
+        for i, col in enumerate(cols):
+            with col:
+                if st.button(f"↓\n{i}", key=f"col_{i}", 
+                           disabled=st.session_state.game_over or not st.session_state.game.is_valid_move(i)):
+                    # Human Move
+                    st.session_state.game.make_move(i, 1)
+                    st.session_state.move_history.append(i)
                     st.session_state.move_count += 1
                     
-                    ai_analysis = st.session_state.ai.get_best_move()
-                    st.session_state.last_human_move_analysis = ai_analysis
+                    # Get AI recommendation
+                    ai_move = st.session_state.ai.get_best_move()
+                    is_optimal = (i == ai_move['move'])
+                    recommendation = {
+                        'is_optimal': is_optimal,
+                        'suggested_column': ai_move['move'],
+                        'score_diff': abs(ai_move['score'] - st.session_state.game.evaluate(2))
+                    }
+                    
+                    # Generate explanation
+                    explanation_data = None
+                    if st.session_state.explainer:
+                        move_data = {
+                            'move': i,
+                            'score': st.session_state.game.evaluate(2),
+                            'threats': ai_move.get('threats', []),
+                            'nodes_explored': ai_move.get('nodes_explored', 0)
+                        }
+                        explanation_result = st.session_state.explainer.explain_move(move_data)
+                        if explanation_result['success']:
+                            explanation_data = explanation_result['explanation']
+                            st.session_state.last_human_move_analysis = explanation_result
                     
                     move_detail = {
-                        'player': current_player,
-                        'player_name': "Human",
-                        'column': human_move,
-                        'analysis': ai_analysis
+                        'player': 1,
+                        'player_name': 'Human',
+                        'column': i,
+                        'analysis': ai_move,
+                        'explanation': explanation_data,
+                        'recommendation': recommendation
                     }
                     st.session_state.move_history_detailed.append(move_detail)
                     
-                    if st.session_state.game.check_winner(current_player):
+                    # Check win
+                    if st.session_state.game.check_winner(1):
                         st.session_state.game_over = True
                         st.session_state.winner = "Human"
                         st.session_state.show_winner_popup = True
-                    elif st.session_state.game.is_board_full():
+                        st.rerun()
+                    
+                    if st.session_state.game.is_board_full():
                         st.session_state.game_over = True
                         st.session_state.winner = "Draw"
                         st.session_state.show_winner_popup = True
-                    else:
-                        with st.spinner("🤖 AI is 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'])
+                        st.rerun()
+                    
+                    # AI Move
+                    if not st.session_state.game_over:
+                        ai_move = st.session_state.ai.get_best_move()
+                        st.session_state.game.make_move(ai_move['move'], 2)
+                        st.session_state.move_history.append(ai_move['move'])
                         st.session_state.move_count += 1
                         
+                        # AI explanation
+                        ai_explanation_data = None
                         if st.session_state.explainer:
-                            explanation = st.session_state.explainer.explain_move(ai_result)
-                            explanation_text = explanation.get('explanation', '')
-                        else:
-                            explanation = {
-                                'explanation': f"Column {ai_result['move']} scores {ai_result['score']} points.",
-                                'threat_analysis': "; ".join(ai_result['threats']) if ai_result['threats'] else "Strategic move",
-                                'key_insight': "Strong position",
-                                'success': False
+                            ai_move_data = {
+                                'move': ai_move['move'],
+                                'score': ai_move['score'],
+                                'threats': ai_move.get('threats', []),
+                                'nodes_explored': ai_move.get('nodes_explored', 0)
                             }
-                            explanation_text = explanation['explanation']
+                            ai_explanation_result = st.session_state.explainer.explain_move(ai_move_data)
+                            if ai_explanation_result['success']:
+                                ai_explanation_data = ai_explanation_result['explanation']
+                                st.session_state.last_ai_move_analysis = ai_explanation_result
                         
                         ai_move_detail = {
                             'player': 2,
-                            'player_name': 'AI',
-                            'column': ai_result['move'],
-                            'analysis': ai_result,
-                            'explanation': explanation_text
+                            'player_name': 'AI (Ultra)',
+                            'column': ai_move['move'],
+                            'analysis': ai_move,
+                            'explanation': ai_explanation_data
                         }
                         st.session_state.move_history_detailed.append(ai_move_detail)
-                        st.session_state.last_ai_move_analysis = ai_result
-                        st.session_state.last_explanation = explanation
                         
+                        # Check AI win
                         if st.session_state.game.check_winner(2):
                             st.session_state.game_over = True
                             st.session_state.winner = "AI"
                             st.session_state.show_winner_popup = True
-                        elif st.session_state.game.is_board_full():
+                            st.rerun()
+                        
+                        if st.session_state.game.is_board_full():
                             st.session_state.game_over = True
                             st.session_state.winner = "Draw"
                             st.session_state.show_winner_popup = True
+                            st.rerun()
                     
                     st.rerun()
     
     with col2:
-        st.markdown("## 📊 Move Analysis")
+        st.subheader("🤖 AI ANALYSIS & INSIGHTS")
         
-        # Human Move Analysis
         if st.session_state.last_human_move_analysis:
-            human_analysis = st.session_state.last_human_move_analysis
-            st.markdown('<div class="player-section human-section">', unsafe_allow_html=True)
-            st.markdown('<div class="section-header">🔴 Your Last Move</div>', unsafe_allow_html=True)
-            
-            st.markdown(f'<div class="info-row"><strong>Column:</strong> {st.session_state.move_history[-2] if len(st.session_state.move_history) >= 2 else "?"}</div>', unsafe_allow_html=True)
-            st.markdown(f'<div class="info-row"><strong>Score:</strong> {human_analysis["score"]} points</div>', unsafe_allow_html=True)
-            
-            if human_analysis['threats']:
-                st.markdown('<div class="info-row"><strong>Threats:</strong></div>', unsafe_allow_html=True)
-                for threat in human_analysis['threats']:
-                    st.markdown(f'<span class="badge badge-threat">{threat}</span>', unsafe_allow_html=True)
-            
-            perf_col1, perf_col2, perf_col3 = st.columns(3)
-            with perf_col1:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{human_analysis["nodes_explored"]:,}</span><span class="stat-label">Explored</span></div>', unsafe_allow_html=True)
-            with perf_col2:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{human_analysis["nodes_pruned"]:,}</span><span class="stat-label">Pruned</span></div>', unsafe_allow_html=True)
-            with perf_col3:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{human_analysis["pruning_efficiency"]:.1f}%</span><span class="stat-label">Efficiency</span></div>', unsafe_allow_html=True)
-            
+            st.markdown('<div class="analysis-card">', unsafe_allow_html=True)
+            st.markdown("#### 👤 YOUR LAST MOVE ANALYSIS")
+            analysis = st.session_state.last_human_move_analysis
+            st.info(f"**Explanation:** {analysis['explanation']}")
+            st.success(f"**Threat Analysis:** {analysis['threat_analysis']}")
+            st.warning(f"**Key Insight:** {analysis['key_insight']}")
             st.markdown('</div>', unsafe_allow_html=True)
         
-        # AI Move Analysis
-        if hasattr(st.session_state, 'last_ai_move_analysis') and st.session_state.last_ai_move_analysis:
-            ai_move = st.session_state.last_ai_move_analysis
-            explanation = getattr(st.session_state, 'last_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': "Strong position",
-                'success': False
-            })
-            
-            st.markdown('<div class="player-section ai-section">', unsafe_allow_html=True)
-            st.markdown('<div class="section-header">🟡 AI Move Analysis</div>', unsafe_allow_html=True)
-            
-            st.markdown(f'<div class="move-card"><div class="move-card-title">Column {ai_move["move"]}</div><div class="move-card-score">Score: {ai_move["score"]} points</div></div>', unsafe_allow_html=True)
-            
-            st.success(f"💡 **Strategy:** {explanation['explanation']}")
-            
-            if explanation.get('threat_analysis'):
-                st.warning(f"⚔️ **Threats:** {explanation['threat_analysis']}")
-            
-            if explanation.get('key_insight'):
-                st.info(f"🔑 **Insight:** {explanation['key_insight']}")
-            
-            perf_col1, perf_col2, perf_col3 = st.columns(3)
-            with perf_col1:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{ai_move["nodes_explored"]:,}</span><span class="stat-label">Explored</span></div>', unsafe_allow_html=True)
-            with perf_col2:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{ai_move["nodes_pruned"]:,}</span><span class="stat-label">Pruned</span></div>', unsafe_allow_html=True)
-            with perf_col3:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{ai_move["pruning_efficiency"]:.1f}%</span><span class="stat-label">Efficiency</span></div>', unsafe_allow_html=True)
-            
-            stat_col4, stat_col5 = st.columns(2)
-            with stat_col4:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{ai_move["depth"]}</span><span class="stat-label">Depth</span></div>', unsafe_allow_html=True)
-            with stat_col5:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{ai_move["time_ms"]:.0f}ms</span><span class="stat-label">Time</span></div>', unsafe_allow_html=True)
-            
+        if st.session_state.last_ai_move_analysis:
+            st.markdown('<div class="analysis-card">', unsafe_allow_html=True)
+            st.markdown("#### 🤖 AI'S LAST MOVE ANALYSIS")
+            analysis = st.session_state.last_ai_move_analysis
+            st.info(f"**Explanation:** {analysis['explanation']}")
+            st.success(f"**Threat Analysis:** {analysis['threat_analysis']}")
+            st.warning(f"**Key Insight:** {analysis['key_insight']}")
             st.markdown('</div>', unsafe_allow_html=True)
+        
+        if not st.session_state.last_human_move_analysis and not st.session_state.last_ai_move_analysis:
+            st.info("🎲 Make your first move to see AI analysis!")
 
 else:  # Two Player Mode
-    st.markdown('<div class="board-container" style="text-align: center;">', unsafe_allow_html=True)
-    st.markdown("### 🎯 Game Board")
-    board_str = st.session_state.game.board_to_string()
-    st.markdown(f'<div class="board-display">{board_str}</div>', unsafe_allow_html=True)
-    st.markdown('</div>', unsafe_allow_html=True)
-    
-    if not st.session_state.game_over:
-        player_name = f"Player {st.session_state.current_player}"
-        player_emoji = "🔴" if st.session_state.current_player == 1 else "🟡"
-        st.markdown(f"### {player_emoji} {player_name}'s Turn")
-        
-        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
-                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
-                
-                ai_analysis = st.session_state.ai.get_best_move()
-                move_detail = {
-                    'player': current_player,
-                    'player_name': f"Player {current_player}",
-                    'column': human_move,
-                    'analysis': ai_analysis
-                }
-                
-                move_detail['recommendation'] = {
-                    'is_optimal': ai_analysis['move'] == human_move,
-                    'suggested_column': ai_analysis['move'],
-                    'score_diff': abs(ai_analysis['score'])
-                }
-                
-                if current_player == 1:
-                    st.session_state.player1_analyses.append(move_detail)
-                else:
-                    st.session_state.player2_analyses.append(move_detail)
-                
-                st.session_state.move_history_detailed.append(move_detail)
-                
-                if st.session_state.game.check_winner(current_player):
-                    st.session_state.game_over = True
-                    st.session_state.winner = f"Player {current_player}"
-                    st.session_state.show_winner_popup = True
-                elif st.session_state.game.is_board_full():
-                    st.session_state.game_over = True
-                    st.session_state.winner = "Draw"
-                    st.session_state.show_winner_popup = True
-                else:
-                    st.session_state.current_player = 2 if current_player == 1 else 1
-                
-                st.rerun()
-    
-    # Player Analysis
-    st.markdown("---")
-    st.markdown("## 📊 Player Analysis")
-    
-    col1, col2 = st.columns(2)
+    col1, col2 = st.columns([2, 3])
     
     with col1:
-        if st.session_state.player1_analyses:
-            st.markdown('<div class="player-section player1-section">', unsafe_allow_html=True)
-            st.markdown('<div class="section-header">🔴 Player 1</div>', unsafe_allow_html=True)
-            latest_p1 = st.session_state.player1_analyses[-1]
-            
-            st.markdown(f'<div class="info-row"><strong>Last Move:</strong> Column {latest_p1["column"]}</div>', unsafe_allow_html=True)
-            
-            if latest_p1['recommendation']['is_optimal']:
-                st.markdown('<span class="badge badge-success">✅ Optimal Move!</span>', unsafe_allow_html=True)
-            else:
-                st.markdown(f'<span class="badge badge-warning">💡 AI suggests Column {latest_p1["recommendation"]["suggested_column"]}</span>', unsafe_allow_html=True)
-            
-            if latest_p1['analysis']['threats']:
-                st.markdown('<div class="info-row" style="margin-top: 12px;"><strong>Threats:</strong></div>', unsafe_allow_html=True)
-                for threat in latest_p1['analysis']['threats']:
-                    st.markdown(f'<span class="badge badge-threat">{threat}</span>', unsafe_allow_html=True)
-            
-            st.markdown(f'<div class="info-row" style="margin-top: 12px;"><strong>Score:</strong> {latest_p1["analysis"]["score"]} points</div>', unsafe_allow_html=True)
-            
-            perf_col1, perf_col2 = st.columns(2)
-            with perf_col1:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{latest_p1["analysis"]["nodes_explored"]:,}</span><span class="stat-label">Nodes Explored</span></div>', unsafe_allow_html=True)
-            with perf_col2:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{latest_p1["analysis"]["pruning_efficiency"]:.1f}%</span><span class="stat-label">Efficiency</span></div>', unsafe_allow_html=True)
-            st.markdown('</div>', unsafe_allow_html=True)
+        st.subheader("🎯 GAME BOARD")
+        board_str = st.session_state.game.board_to_string()
+        st.markdown(f'<div class="board-container"><pre>{board_str}</pre></div>', unsafe_allow_html=True)
+        
+        # Current Player Indicator
+        current_player_symbol = "🔴 RED" if st.session_state.current_player == 1 else "🟡 YELLOW"
+        st.markdown(f'<div class="info-card"><h3 style="text-align: center;">Current Turn: {current_player_symbol}</h3></div>', unsafe_allow_html=True)
+        
+        # Column Selection
+        st.markdown("### 🎲 SELECT COLUMN")
+        cols = st.columns(7)
+        for i, col in enumerate(cols):
+            with col:
+                if st.button(f"↓\n{i}", key=f"col_{i}", 
+                           disabled=st.session_state.game_over or not st.session_state.game.is_valid_move(i)):
+                    # Make move
+                    st.session_state.game.make_move(i, st.session_state.current_player)
+                    st.session_state.move_history.append(i)
+                    st.session_state.move_count += 1
+                    
+                    # Get AI analysis
+                    ai_move = st.session_state.ai.get_best_move()
+                    is_optimal = (i == ai_move['move'])
+                    recommendation = {
+                        'is_optimal': is_optimal,
+                        'suggested_column': ai_move['move'],
+                        'score_diff': abs(ai_move['score'] - st.session_state.game.evaluate(2))
+                    }
+                    
+                    # Generate explanation
+                    explanation_data = None
+                    if st.session_state.explainer:
+                        move_data = {
+                            'move': i,
+                            'score': st.session_state.game.evaluate(2),
+                            'threats': ai_move.get('threats', []),
+                            'nodes_explored': ai_move.get('nodes_explored', 0)
+                        }
+                        explanation_result = st.session_state.explainer.explain_move(move_data)
+                        if explanation_result['success']:
+                            explanation_data = explanation_result['explanation']
+                            if st.session_state.current_player == 1:
+                                st.session_state.player1_analyses.append(explanation_result)
+                            else:
+                                st.session_state.player2_analyses.append(explanation_result)
+                    
+                    player_name = f"Player {st.session_state.current_player}"
+                    move_detail = {
+                        'player': st.session_state.current_player,
+                        'player_name': player_name,
+                        'column': i,
+                        'analysis': ai_move,
+                        'explanation': explanation_data,
+                        'recommendation': recommendation
+                    }
+                    st.session_state.move_history_detailed.append(move_detail)
+                    
+                    # Check win
+                    if st.session_state.game.check_winner(st.session_state.current_player):
+                        st.session_state.game_over = True
+                        st.session_state.winner = f"Player {st.session_state.current_player}"
+                        st.session_state.show_winner_popup = True
+                        st.rerun()
+                    
+                    if st.session_state.game.is_board_full():
+                        st.session_state.game_over = True
+                        st.session_state.winner = "Draw"
+                        st.session_state.show_winner_popup = True
+                        st.rerun()
+                    
+                    # Switch player
+                    st.session_state.current_player = 2 if st.session_state.current_player == 1 else 1
+                    st.rerun()
     
     with col2:
+        st.subheader("📊 PLAYER ANALYSIS")
+        
+        if st.session_state.player1_analyses:
+            st.markdown('<div class="analysis-card">', unsafe_allow_html=True)
+            st.markdown("#### 🔴 PLAYER 1 - LAST MOVE")
+            analysis = st.session_state.player1_analyses[-1]
+            st.info(f"**Explanation:** {analysis['explanation']}")
+            st.success(f"**Threat Analysis:** {analysis['threat_analysis']}")
+            st.warning(f"**Key Insight:** {analysis['key_insight']}")
+            st.markdown('</div>', unsafe_allow_html=True)
+        
         if st.session_state.player2_analyses:
-            st.markdown('<div class="player-section player2-section">', unsafe_allow_html=True)
-            st.markdown('<div class="section-header">🟡 Player 2</div>', unsafe_allow_html=True)
-            latest_p2 = st.session_state.player2_analyses[-1]
-            
-            st.markdown(f'<div class="info-row"><strong>Last Move:</strong> Column {latest_p2["column"]}</div>', unsafe_allow_html=True)
-            
-            if latest_p2['recommendation']['is_optimal']:
-                st.markdown('<span class="badge badge-success">✅ Optimal Move!</span>', unsafe_allow_html=True)
-            else:
-                st.markdown(f'<span class="badge badge-warning">💡 AI suggests Column {latest_p2["recommendation"]["suggested_column"]}</span>', unsafe_allow_html=True)
-            
-            if latest_p2['analysis']['threats']:
-                st.markdown('<div class="info-row" style="margin-top: 12px;"><strong>Threats:</strong></div>', unsafe_allow_html=True)
-                for threat in latest_p2['analysis']['threats']:
-                    st.markdown(f'<span class="badge badge-threat">{threat}</span>', unsafe_allow_html=True)
-            
-            st.markdown(f'<div class="info-row" style="margin-top: 12px;"><strong>Score:</strong> {latest_p2["analysis"]["score"]} points</div>', unsafe_allow_html=True)
-            
-            perf_col1, perf_col2 = st.columns(2)
-            with perf_col1:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{latest_p2["analysis"]["nodes_explored"]:,}</span><span class="stat-label">Nodes Explored</span></div>', unsafe_allow_html=True)
-            with perf_col2:
-                st.markdown(f'<div class="stats-card"><span class="stat-number">{latest_p2["analysis"]["pruning_efficiency"]:.1f}%</span><span class="stat-label">Efficiency</span></div>', unsafe_allow_html=True)
+            st.markdown('<div class="analysis-card">', unsafe_allow_html=True)
+            st.markdown("#### 🟡 PLAYER 2 - LAST MOVE")
+            analysis = st.session_state.player2_analyses[-1]
+            st.info(f"**Explanation:** {analysis['explanation']}")
+            st.success(f"**Threat Analysis:** {analysis['threat_analysis']}")
+            st.warning(f"**Key Insight:** {analysis['key_insight']}")
             st.markdown('</div>', unsafe_allow_html=True)
+        
+        if not st.session_state.player1_analyses and not st.session_state.player2_analyses:
+            st.info("🎲 Start playing to see move analysis!")
 
 # Winner Display
-if st.session_state.show_winner_popup and st.session_state.game_over:
-    if st.session_state.winner == "Human":
+if st.session_state.show_winner_popup:
+    if st.session_state.winner == "Draw":
         st.balloons()
-        st.markdown('<div class="winner-banner">🎉 YOU WON! 🎉</div>', unsafe_allow_html=True)
-    elif st.session_state.winner == "AI":
-        st.markdown('<div class="winner-banner ai">🤖 AI WINS!</div>', unsafe_allow_html=True)
-    elif "Player" in st.session_state.winner:
+        st.success("🤝 **GAME OVER - IT'S A DRAW!**")
+    elif st.session_state.winner == "Human":
         st.balloons()
-        st.markdown(f'<div class="winner-banner">🎉 {st.session_state.winner.upper()} WINS! 🎉</div>', unsafe_allow_html=True)
+        st.success("🏆 **CONGRATULATIONS! YOU WON!**")
+    elif st.session_state.winner == "AI":
+        st.error("🤖 **AI WINS! Better luck next time!**")
     else:
-        st.markdown('<div class="winner-banner draw">🤝 IT\'S A DRAW!</div>', unsafe_allow_html=True)
-    
-    st.session_state.show_winner_popup = False
+        st.balloons()
+        st.success(f"🏆 **{st.session_state.winner.upper()} WINS!**")
 
-# Download Report
-if st.session_state.game_over:
+# Match Report Download
+if st.session_state.game_over and st.session_state.move_history_detailed:
     st.markdown("---")
-    st.markdown("## 📄 Match Report")
+    st.subheader("📄 MATCH REPORT")
     
     report = generate_match_report(
-        st.session_state.game_mode,
-        "Ultra AI",
-        st.session_state.winner,
-        st.session_state.move_history_detailed,
-        st.session_state.game.board_to_string(),
+        game_mode=st.session_state.game_mode,
+        ai_mode="Ultra AI" if st.session_state.game_mode == "vs AI" else "N/A",
+        winner=st.session_state.winner,
+        move_history_data=st.session_state.move_history_detailed,
+        game_board=st.session_state.game.board_to_string(),
         search_depth=st.session_state.current_depth if st.session_state.game_mode == "vs AI" else None
     )
     
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    filename = f"connect_four_report_{timestamp}.txt"
-    
     st.download_button(
-        label="📥 Download Full Match Report",
+        label="📥 DOWNLOAD MATCH REPORT",
         data=report,
-        file_name=filename,
+        file_name=f"connect_four_match_report_{timestamp}.txt",
         mime="text/plain",
-        use_container_width=True,
-        type="primary"
-    )
+        use_container_width=True
+    )

connect_four_game.py

@@ -3,21 +3,21 @@ from typing import Tuple, List, Dict
 
 class ConnectFour:
     '''Connect Four game engine with clean board display'''
-
+    
     def __init__(self, rows=6, cols=7):
         self.rows = rows
         self.cols = cols
         self.board = np.zeros((rows, cols), dtype=int)
         self.move_count = 0
-
+    
     def is_valid_move(self, col: int) -> bool:
         if col < 0 or col >= self.cols:
             return False
         return self.board[0][col] == 0
-
+    
     def get_valid_moves(self) -> List[int]:
         return [col for col in range(self.cols) if self.is_valid_move(col)]
-
+    
     def make_move(self, col: int, player: int) -> bool:
         for row in range(self.rows - 1, -1, -1):
             if self.board[row][col] == 0:
@@ -25,49 +25,49 @@ class ConnectFour:
                 self.move_count += 1
                 return True
         return False
-
+    
     def undo_move(self, col: int):
         for row in range(self.rows):
             if self.board[row][col] != 0:
                 self.board[row][col] = 0
                 self.move_count -= 1
                 return
-
+    
     def check_winner(self, player: int) -> bool:
         # Horizontal
         for row in range(self.rows):
             for col in range(self.cols - 3):
                 if all(self.board[row][col + i] == player for i in range(4)):
                     return True
-
+        
         # Vertical
         for row in range(self.rows - 3):
             for col in range(self.cols):
                 if all(self.board[row + i][col] == player for i in range(4)):
                     return True
-
+        
         # Diagonal
         for row in range(self.rows - 3):
             for col in range(self.cols - 3):
                 if all(self.board[row + i][col + i] == player for i in range(4)):
                     return True
-
+        
         for row in range(3, self.rows):
             for col in range(self.cols - 3):
                 if all(self.board[row - i][col + i] == player for i in range(4)):
                     return True
-
+        
         return False
-
+    
     def is_board_full(self) -> bool:
         return len(self.get_valid_moves()) == 0
-
+    
     def get_board_copy(self):
         return self.board.copy()
-
+    
     def count_threats(self, player: int) -> Dict[str, int]:
         threats = {'threes': 0, 'twos': 0, 'center': 0}
-
+        
         for row in range(self.rows):
             for col in range(self.cols):
                 if self.board[row][col] == player:
@@ -79,53 +79,53 @@ class ConnectFour:
                         threats['threes'] += 1
                     if row - 3 >= 0 and col + 3 < self.cols and all(self.board[row - i][col + i] == player for i in range(1, 4)):
                         threats['threes'] += 1
-
+        
         for row in range(self.rows):
             for col in [2, 3, 4]:
                 if self.board[row][col] == player:
                     threats['center'] += 1
-
+        
         return threats
-
+    
     def evaluate(self, ai_player=2) -> int:
         human = 1
-
+        
         if self.check_winner(ai_player):
             return 10000
         if self.check_winner(human):
             return -10000
         if self.is_board_full():
             return 0
-
+        
         score = 0
         ai_threats = self.count_threats(ai_player)
         human_threats = self.count_threats(human)
-
+        
         score += ai_threats['threes'] * 1200
         score -= human_threats['threes'] * 1100
         score += ai_threats['twos'] * 60
         score -= human_threats['twos'] * 50
         score += ai_threats['center'] * 4
         score -= human_threats['center'] * 3
-
+        
         return score
-
+    
     def board_to_string(self) -> str:
         '''Clean board display without dotted lines'''
         mapping = {0: '⚪', 1: '🔴', 2: '🟡'}
-
+        
         # Column header with proper spacing
-        result = "       0   1   2   3   4   5   6\n"
+        result = "  0  1  2  3  4  5  6\n"
         result += "\n"
-
+        
         for row in range(self.rows):
-            result += f"   {row}  │ "
+            result += f" {row} │ "
             for col in range(self.cols):
-                result += mapping[self.board[row][col]] + "  "
+                result += mapping[self.board[row][col]] + " "
             result += "│\n"
-
+        
         return result
-
+    
     def reset(self):
         self.board = np.zeros((self.rows, self.cols), dtype=int)
         self.move_count = 0

minimax_ai.py

@@ -33,10 +33,9 @@ class MinimaxAI:
         except:
             return None
     
-    def minimax(self, depth: int, alpha: int, beta: int, 
+    def minimax(self, depth: int, alpha: int, beta: int,
                 is_maximizing: bool, current_depth=0) -> int:
         '''Minimax with Alpha-Beta Pruning (Fixed)'''
-        
         self.max_depth_reached = max(self.max_depth_reached, current_depth)
         self.nodes_explored += 1
         
@@ -64,7 +63,6 @@ class MinimaxAI:
             return self.game.evaluate(self.ai_player)
         
         valid_moves = self.game.get_valid_moves()
-        
         if not valid_moves:
             return self.game.evaluate(self.ai_player)
         
@@ -90,7 +88,6 @@ class MinimaxAI:
             if board_key:
                 self.transposition_table[board_key] = (max_eval, depth)
             return max_eval
-        
         else:
             min_eval = float('inf')
             for i, col in enumerate(sorted_moves):
@@ -113,7 +110,6 @@ class MinimaxAI:
     
     def get_best_move(self) -> Dict:
         '''Find best move (Fixed Pruning Counter)'''
-        
         self.reset_stats()
         start_time = time.time()
         
@@ -144,7 +140,6 @@ class MinimaxAI:
             
             # Minimax search
             score = self.minimax(self.depth - 1, float('-inf'), float('inf'), False)
-            
             self.game.undo_move(col)
             
             if score > best_score:
@@ -152,9 +147,8 @@ class MinimaxAI:
                 best_move = col
         
         elapsed_time = (time.time() - start_time) * 1000
-        
         pruning_efficiency = (self.nodes_pruned / (self.nodes_explored + self.nodes_pruned) * 100) \
-                            if (self.nodes_explored + self.nodes_pruned) > 0 else 0
+            if (self.nodes_explored + self.nodes_pruned) > 0 else 0
         
         return {
             'move': best_move,

report_generator.py

@@ -2,25 +2,22 @@ from datetime import datetime
 
 def generate_match_report(game_mode, ai_mode, winner, move_history_data, game_board, search_depth=None):
     """Generate comprehensive match report with algorithm details"""
-    
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     
     # Build report
     lines = []
     lines.append("=" * 79)
     lines.append("")
-    lines.append("                CONNECT FOUR AI - MATCH REPORT")
+    lines.append("           CONNECT FOUR AI - MATCH REPORT")
     lines.append("")
     lines.append("=" * 79)
     lines.append("")
     lines.append(f"Date: {timestamp}")
     lines.append(f"Game Mode: {game_mode}")
-    
     if game_mode == "vs AI":
         lines.append(f"AI Difficulty: {ai_mode}")
         if search_depth:
             lines.append(f"Selected Search Depth: {search_depth}")
-    
     lines.append(f"Winner: {winner}")
     lines.append(f"Total Moves: {len(move_history_data)}")
     lines.append("")
@@ -55,34 +52,34 @@ def generate_match_report(game_mode, ai_mode, winner, move_history_data, game_bo
                 for threat in analysis['threats']:
                     lines.append(f"    * {threat}")
                 lines.append("")
-            
-            if 'explanation' in move_data and move_data['explanation']:
-                lines.append("  Move Explanation:")
-                explanation = move_data['explanation']
-                words = explanation.split()
-                current_line = "  "
-                for word in words:
-                    if len(current_line + word) > 75:
-                        lines.append(current_line)
-                        current_line = "  " + word + " "
-                    else:
-                        current_line += word + " "
-                if current_line.strip():
+        
+        if 'explanation' in move_data and move_data['explanation']:
+            lines.append("  Move Explanation:")
+            explanation = move_data['explanation']
+            words = explanation.split()
+            current_line = "    "
+            for word in words:
+                if len(current_line + word) > 75:
                     lines.append(current_line)
-                lines.append("")
+                    current_line = "    " + word + " "
+                else:
+                    current_line += word + " "
+            if current_line.strip():
+                lines.append(current_line)
+            lines.append("")
         
         if 'recommendation' in move_data:
             lines.append("")
             if move_data['recommendation']['is_optimal']:
-                lines.append("  EXCELLENT MOVE! Matches AI recommendation.")
+                lines.append("  ✓ EXCELLENT MOVE! Matches AI recommendation.")
             else:
-                lines.append(f"  AI Alternative: Column {move_data['recommendation']['suggested_column']}")
-                lines.append(f"  (Score difference: {move_data['recommendation']['score_diff']} points)")
+                lines.append(f"  ⓘ AI Alternative: Column {move_data['recommendation']['suggested_column']}")
+                lines.append(f"    (Score difference: {move_data['recommendation']['score_diff']} points)")
+            lines.append("")
         
+        lines.append("=" * 79)
         lines.append("")
     
-    lines.append("=" * 79)
-    lines.append("")
     lines.append("MATCH STATISTICS")
     lines.append("-" * 79)
     lines.append("")
@@ -141,23 +138,21 @@ def generate_match_report(game_mode, ai_mode, winner, move_history_data, game_bo
     lines.append("")
     
     if winner == "Human":
-        lines.append("Congratulations! You defeated the AI!")
+        lines.append("🏆 Congratulations! You defeated the AI!")
     elif winner == "AI":
-        lines.append("The AI won this match. Better luck next time!")
+        lines.append("🤖 The AI won this match. Better luck next time!")
     elif "Player" in winner:
-        lines.append(f"{winner} won the match!")
+        lines.append(f"🏆 {winner} won the match!")
     else:
-        lines.append("The match ended in a draw.")
+        lines.append("🤝 The match ended in a draw.")
     
     lines.append("")
     lines.append(f"Total Moves Played: {len(move_history_data)}")
     lines.append(f"Game Mode: {game_mode}")
-    
     if game_mode == "vs AI":
         lines.append(f"AI Difficulty: {ai_mode}")
         if search_depth:
             lines.append(f"Selected Search Depth: {search_depth}")
-    
     lines.append("")
     lines.append("=" * 79)
     lines.append("")