Update app.py

app.py

@@ -21,6 +21,8 @@ st.markdown('''
     .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; }
+    .human-section { background: linear-gradient(135deg, #e0e7ff 0%, #f3e8ff 100%); border-left: 5px solid #8b5cf6; }
+    .ai-section { background: linear-gradient(135deg, #fef3c7 0%, #fecaca 100%); border-left: 5px solid #f97316; }
 </style>
 ''', unsafe_allow_html=True)
 
@@ -42,6 +44,8 @@ if 'game' not in st.session_state:
     st.session_state.show_winner_popup = False
     st.session_state.move_count = 0
     st.session_state.current_depth = 5
+    st.session_state.last_human_move_analysis = None
+    st.session_state.last_ai_move_analysis = None
 
 col_title1, col_title2 = st.columns([3, 1])
 with col_title1:
@@ -91,6 +95,8 @@ with col1:
         st.session_state.current_player = 1
         st.session_state.show_winner_popup = False
         st.session_state.move_count = 0
+        st.session_state.last_human_move_analysis = None
+        st.session_state.last_ai_move_analysis = None
         st.rerun()
 
 with col2:
@@ -146,6 +152,8 @@ if st.session_state.game_mode == "vs AI":
                     st.session_state.move_count += 1
                     
                     ai_analysis = st.session_state.ai.get_best_move()
+                    st.session_state.last_human_move_analysis = ai_analysis
+                    
                     move_detail = {
                         'player': current_player,
                         'player_name': "Human",
@@ -174,7 +182,13 @@ if st.session_state.game_mode == "vs AI":
                             explanation = st.session_state.explainer.explain_move(ai_result)
                             explanation_text = explanation.get('explanation', '')
                         else:
-                            explanation_text = f"AI played column {ai_result['move']}."
+                            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
+                            }
+                            explanation_text = explanation['explanation']
                         
                         ai_move_detail = {
                             'player': 2,
@@ -184,7 +198,8 @@ if st.session_state.game_mode == "vs AI":
                             'explanation': explanation_text
                         }
                         st.session_state.move_history_detailed.append(ai_move_detail)
-                        st.session_state.latest_ai_move = ai_result
+                        st.session_state.last_ai_move_analysis = ai_result
+                        st.session_state.last_explanation = explanation
                         
                         if st.session_state.game.check_winner(2):
                             st.session_state.game_over = True
@@ -198,24 +213,45 @@ if st.session_state.game_mode == "vs AI":
                     st.rerun()
     
     with col2:
-        st.subheader("🤖 AI Analysis & Insights")
+        st.subheader("📊 Analysis Performance")
         
-        if hasattr(st.session_state, 'latest_ai_move'):
-            ai_move = st.session_state.latest_ai_move
+        # 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("### 🔴 Your Move Analysis")
             
-            if st.session_state.explainer:
-                explanation = st.session_state.explainer.explain_move(ai_move)
-                has_llm = explanation.get('success', False)
-            else:
-                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
-                }
-                has_llm = False
+            st.write(f"**Your Last Move:** Column {st.session_state.move_history[-2] if len(st.session_state.move_history) >= 2 else '?'}")
+            st.write(f"**Score Evaluation:** {human_analysis['score']} points")
+            
+            if human_analysis['threats']:
+                st.markdown(f"⚔️ **Strategic Threats:** {', '.join(human_analysis['threats'])}")
+            
+            st.markdown("**Algorithm Metrics:**")
+            perf_col1, perf_col2, perf_col3 = st.columns(3)
+            with perf_col1:
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{human_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">{human_analysis["nodes_pruned"]:,}</div><div class="stat-label">Pruned</div></div>', unsafe_allow_html=True)
+            with perf_col3:
+                st.markdown(f'<div class="stats-card"><div class="stat-number">{human_analysis["pruning_efficiency"]:.1f}%</div><div class="stat-label">Efficiency</div></div>', unsafe_allow_html=True)
+            
+            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("### 🟡 AI Move Analysis")
             
-            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.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'):
@@ -224,14 +260,13 @@ if st.session_state.game_mode == "vs AI":
             if explanation.get('key_insight'):
                 st.info(f"🔑 **Strategic Insight:**\n{explanation['key_insight']}")
             
-            st.markdown("### 📊 Algorithm Performance")
-            
-            stat_col1, stat_col2, stat_col3 = st.columns(3)
-            with stat_col1:
+            st.markdown("**Algorithm Metrics:**")
+            perf_col1, perf_col2, perf_col3 = st.columns(3)
+            with perf_col1:
                 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:
+            with perf_col2:
                 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:
+            with perf_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)
             
             stat_col4, stat_col5 = st.columns(2)
@@ -240,8 +275,10 @@ if st.session_state.game_mode == "vs AI":
             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">Time</div></div>', unsafe_allow_html=True)
             
-            if not has_llm:
-                st.info("ℹ️ Using fallback explanation (HF API unavailable)")
+            if not explanation.get('success', False):
+                st.info("ℹ️ Using smart rule-based explanations")
+            
+            st.markdown('</div>', unsafe_allow_html=True)
 
 else:  # Two Player Mode
     st.subheader("🎯 Game Board")
@@ -319,6 +356,11 @@ else:  # Two Player Mode
             
             st.markdown(f"**Score:** {latest_p1['analysis']['score']} points")
             
+            st.markdown("**AI Insights:**")
+            st.write(f"💭 This move evaluates to {latest_p1['analysis']['score']} points. " + 
+                    ("AI recommends this column for the best position." if latest_p1['recommendation']['is_optimal'] else 
+                     f"AI would prefer Column {latest_p1['recommendation']['suggested_column']} for better positioning."))
+            
             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)
@@ -342,6 +384,11 @@ else:  # Two Player Mode
             
             st.markdown(f"**Score:** {latest_p2['analysis']['score']} points")
             
+            st.markdown("**AI Insights:**")
+            st.write(f"💭 This move evaluates to {latest_p2['analysis']['score']} points. " + 
+                    ("AI recommends this column for the best position." if latest_p2['recommendation']['is_optimal'] else 
+                     f"AI would prefer Column {latest_p2['recommendation']['suggested_column']} for better positioning."))
+            
             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)