Update queries_demo.py

queries_demo.py

@@ -42,8 +42,9 @@ class DemoDataGenerator:
         # US state abbreviations for realistic distribution
         states = ['CA', 'TX', 'NY', 'FL', 'IL', 'PA', 'OH', 'GA', 'NC', 'MI',
                   'NJ', 'VA', 'WA', 'AZ', 'MA', 'TN', 'IN', 'MO', 'MD', 'WI']
+        # Probabilities must sum to exactly 1.0
         state_weights = [0.15, 0.12, 0.10, 0.09, 0.06, 0.05, 0.05, 0.04, 0.04, 0.04,
-                         0.04, 0.04, 0.04, 0.03, 0.03, 0.02, 0.02, 0.02, 0.02, 0.01]
+                         0.04, 0.04, 0.03, 0.03, 0.03, 0.02, 0.02, 0.02, 0.02, 0.01]
         
         # Cities by state
         cities_by_state = {