Upload app.py

app.py

@@ -242,6 +242,13 @@ async def extract_tokens(pages_data, progress=gr.Progress()):
         state.log(f"   Radius:     {len(state.desktop_raw.radius)} values")
         state.log(f"   Shadows:    {len(state.desktop_raw.shadows)} values")
         
+        # Store foreground-background pairs for real AA checking in Stage 2
+        if hasattr(desktop_extractor, 'fg_bg_pairs') and desktop_extractor.fg_bg_pairs:
+            state.fg_bg_pairs = desktop_extractor.fg_bg_pairs
+            state.log(f"   FG/BG Pairs: {len(state.fg_bg_pairs)} unique pairs for AA checking")
+        else:
+            state.fg_bg_pairs = []
+
         # Log CSS variables if found
         if hasattr(desktop_extractor, 'css_variables') and desktop_extractor.css_variables:
             state.log("")
@@ -920,6 +927,7 @@ async def run_stage2_analysis_v2(
                 radius_tokens=desktop_dict.get("radius", {}),
                 shadow_tokens=desktop_dict.get("shadows", {}),
                 log_callback=state.log,
+                fg_bg_pairs=getattr(state, 'fg_bg_pairs', None),
             )
             
             state.rule_engine_results = rule_results
@@ -996,7 +1004,7 @@ async def run_stage2_analysis_v2(
             if benchmarks and rule_results:
                 benchmark_comparisons = researcher.compare_to_benchmarks(
                     your_ratio=rule_results.typography.detected_ratio,
-                    your_base_size=int(rule_results.typography.sizes_px[0]) if rule_results.typography.sizes_px else 16,
+                    your_base_size=int(rule_results.typography.base_size) if rule_results.typography.sizes_px else 16,
                     your_spacing_grid=rule_results.spacing.detected_base,
                     benchmarks=benchmarks,
                     log_callback=state.log,
@@ -1068,7 +1076,7 @@ async def run_stage2_analysis_v2(
                     try:
                         benchmark_advice = await benchmark_agent.analyze(
                             user_ratio=rule_results.typography.detected_ratio,
-                            user_base=int(rule_results.typography.sizes_px[0]) if rule_results.typography.sizes_px else 16,
+                            user_base=int(rule_results.typography.base_size) if rule_results.typography.sizes_px else 16,
                             user_spacing=rule_results.spacing.detected_base,
                             benchmark_comparisons=benchmark_comparisons,
                             log_callback=state.log,