Update app.py

app.py

@@ -105,6 +105,7 @@ if 'game' not in st.session_state:
     st.session_state.current_player = 1
     st.session_state.show_winner_popup = False
     st.session_state.move_count = 0
+    st.session_state.current_depth = 5
 
 # Title
 col_title1, col_title2 = st.columns([3, 1])
@@ -138,34 +139,32 @@ if game_mode != st.session_state.game_mode:
     st.session_state.move_count = 0
     st.rerun()
 
-# AI Mode - DON'T reset game when switching modes
+# AI Mode
 if st.session_state.game_mode == "vs AI":
     st.sidebar.markdown("---")
-    st.sidebar.subheader("🤖 AI Efficiency")
+    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, deeper analysis",
                                key="ai_mode_radio")
     
-    # Update mode without resetting game
     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
-        # Don't reset - just update the AI reference
     
     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 - MAX 8 FOR BOTH
+    # 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, 
-                                   help="Lower = faster, Higher = stronger")
+        depth = st.sidebar.slider("Depth (Ultra-Fast)", 3, 8, 5, help="Lower = faster, Higher = stronger")
         st.session_state.ai_ultra.depth = depth
+        st.session_state.current_depth = depth
     else:
-        depth = st.sidebar.slider("Depth (Balanced)", 4, 8, 6,
-                                   help="Lower = faster, Higher = stronger")
+        depth = st.sidebar.slider("Depth (Balanced)", 4, 8, 6, help="Lower = faster, Higher = stronger")
         st.session_state.ai_balanced.depth = depth
+        st.session_state.current_depth = depth
 
 st.sidebar.markdown("---")
 
@@ -270,12 +269,10 @@ with col1:
                 st.session_state.move_history.append(human_move)
                 st.session_state.move_count += 1
                 
-                # Get AI analysis - USE CURRENT MODE'S AI (not always balanced)
+                # Get AI analysis
                 if st.session_state.game_mode == "Two Player":
-                    # For two player, use balanced for analysis
                     analysis_ai = st.session_state.ai_balanced
                 else:
-                    # For vs AI, use whichever mode is selected
                     analysis_ai = st.session_state.current_ai
                 
                 ai_analysis = analysis_ai.get_best_move()
@@ -354,7 +351,6 @@ with col2:
     if st.session_state.game_mode == "vs AI" and hasattr(st.session_state, 'latest_ai_move'):
         ai_move = st.session_state.latest_ai_move
         
-        # Get FULL explanation
         if st.session_state.explainer:
             explanation = st.session_state.explainer.explain_move(ai_move)
             has_llm = explanation.get('success', False)
@@ -375,7 +371,6 @@ with col2:
         </div>
         ''', unsafe_allow_html=True)
         
-        # FULL analysis display
         st.success(f"💡 **Why This Move:**\n{explanation['explanation']}")
         
         if explanation.get('threat_analysis'):
@@ -384,7 +379,6 @@ with col2:
         if explanation.get('key_insight'):
             st.info(f"🔑 **Strategic Insight:**\n{explanation['key_insight']}")
         
-        # Complete 5 metrics
         st.markdown("### 📊 Algorithm Performance")
         
         stat_col1, stat_col2, stat_col3 = st.columns(3)
@@ -426,12 +420,10 @@ with col2:
             </div>
             ''', unsafe_allow_html=True)
         
-        # Only show fallback message if LLM actually failed
         if not has_llm:
             st.info("ℹ️ Using fallback explanation (HF API unavailable)")
     
     elif st.session_state.game_mode == "Two Player":
-        # Separate player sections
         if st.session_state.player1_analyses:
             st.markdown("### 🔴 Player 1 Analysis")
             latest_p1 = st.session_state.player1_analyses[-1]
@@ -444,6 +436,27 @@ with col2:
             
             if latest_p1['analysis']['threats']:
                 st.info(f"⚔️ **Threats:** {', '.join(latest_p1['analysis']['threats'])}")
+            
+            # Algorithm Performance for Two Player
+            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">Nodes 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">Nodes Pruned</div>
+                </div>
+                ''', unsafe_allow_html=True)
+            
             st.markdown('</div>', unsafe_allow_html=True)
         
         if st.session_state.player2_analyses:
@@ -458,6 +471,27 @@ with col2:
             
             if latest_p2['analysis']['threats']:
                 st.info(f"⚔️ **Threats:** {', '.join(latest_p2['analysis']['threats'])}")
+            
+            # Algorithm Performance for Two Player
+            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">Nodes 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">Nodes Pruned</div>
+                </div>
+                ''', unsafe_allow_html=True)
+            
             st.markdown('</div>', unsafe_allow_html=True)
 
 # Winner popup
@@ -479,14 +513,15 @@ if st.session_state.show_winner_popup and st.session_state.game_over:
 if st.session_state.game_over:
     st.markdown("---")
     st.markdown("### 📄 Download Match Report")
-    st.info("📥 Get a detailed analysis of this match including move-by-move breakdown")
+    st.info("📥 Get a detailed analysis with algorithm metrics and search depth")
     
     report = generate_match_report(
         st.session_state.game_mode,
         st.session_state.ai_mode,
         st.session_state.winner,
         st.session_state.move_history_detailed,
-        st.session_state.game.board_to_string()
+        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")