refactoring to add support for US States as well

.gitignore

@@ -1,2 +1,6 @@
 .env
-*.DS_Store
+*.DS_Store
+__pycache__/
+*.py[cod]
+*$py.class
+.pytest_cache/

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import re
+
 import random
 import os
-import requests
-from bs4 import BeautifulSoup
+import re
 from dotenv import load_dotenv
 
+from rag import fetch_facts
+
 # Load environment variables from .env file if it exists
 load_dotenv()
 
@@ -17,209 +18,84 @@ LOCAL_MODE = bool(BASE_URL and LOCAL_TOKEN)
 MODEL_NAME = os.getenv('MODEL_NAME', 'openai/gpt-oss-20b')
 
 
-# List of countries for the game with URLs
-COUNTRIES = [
-    {"name": "Algeria", "url": "/algeria-facts-for-kids.html"},
-    {"name": "Angola", "url": "/facts-about-angola.html"},
-    {"name": "Argentina", "url": "/argentina-facts.html"},
-    {"name": "Australia", "url": "/australia-facts.html"},
-    {"name": "Austria", "url": "/austria-facts.html"},
-    {"name": "Bahamas", "url": "/facts-about-the-bahamas.html"},
-    {"name": "Barbados", "url": "/barbados-facts.html"},
-    {"name": "Belgium", "url": "/belgium-facts.html"},
-    {"name": "Belize", "url": "/facts-about-belize.html"},
-    {"name": "Bhutan", "url": "/bhutan-facts.html"},
-    {"name": "Bolivia", "url": "/bolivia-facts.html"},
-    {"name": "Botswana", "url": "/facts-about-botswana.html"},
-    {"name": "Brazil", "url": "/brazil-facts.html"},
-    {"name": "Bulgaria", "url": "/facts-about-bulgaria.html"},
-    {"name": "Canada", "url": "/canada-facts-for-kids.html"},
-    {"name": "Chile", "url": "/chile-facts.html"},
-    {"name": "China", "url": "/china-facts.html"},
-    {"name": "Colombia", "url": "/colombia-facts.html"},
-    {"name": "Comoros", "url": "/comoros-facts.html"},
-    {"name": "Costa Rica", "url": "/costa-rica-facts.html"},
-    {"name": "Croatia", "url": "/croatia-facts.html"},
-    {"name": "Cuba", "url": "/cuba-facts.html"},
-    {"name": "Cyprus", "url": "/cyprus-facts-for-kids.html"},
-    {"name": "Denmark", "url": "/denmark-facts.html"},
-    {"name": "Dominican Republic", "url": "/dominican-republic-facts.html"},
-    {"name": "Ecuador", "url": "/ecuador-facts.html"},
-    {"name": "Egypt", "url": "/egypt-facts.html"},
-    {"name": "Estonia", "url": "/estonia-facts-for-kids.html"},
-    {"name": "Eswatini", "url": "/eswatini-facts.html"},
-    {"name": "Ethiopia", "url": "/ethiopia-facts.html"},
-    {"name": "Fiji", "url": "/facts-about-fiji.html"},
-    {"name": "Finland", "url": "/finland-facts.html"},
-    {"name": "France", "url": "/france-facts.html"},
-    {"name": "Georgia", "url": "/georgia-facts.html"},
-    {"name": "Germany", "url": "/germany-facts.html"},
-    {"name": "Ghana", "url": "/ghana-facts.html"},
-    {"name": "Greece", "url": "/greece-facts.html"},
-    {"name": "Greenland", "url": "/facts-about-greenland.html"},
-    {"name": "Guatemala", "url": "/guatemala-facts.html"},
-    {"name": "Guyana", "url": "/guyana-facts.html"},
-    {"name": "Honduras", "url": "/honduras-facts-for-kids.html"},
-    {"name": "Hong Kong", "url": "/facts-about-hong-kong.html"},
-    {"name": "Hungary", "url": "/hungary-facts.html"},
-    {"name": "Iceland", "url": "/iceland-facts.html"},
-    {"name": "India", "url": "/india-for-kids.html"},
-    {"name": "Indonesia", "url": "/indonesia-facts.html"},
-    {"name": "Iran", "url": "/iran-facts-for-kids.html"},
-    {"name": "Ireland", "url": "/ireland-for-kids.html"},
-    {"name": "Israel", "url": "/israel-facts.html"},
-    {"name": "Italy", "url": "/italy-facts.html"},
-    {"name": "Jamaica", "url": "/jamaica-facts.html"},
-    {"name": "Japan", "url": "/japan-facts.html"},
-    {"name": "Kenya", "url": "/facts-about-kenya.html"},
-    {"name": "Kiribati", "url": "/facts-about-kiribati.html"},
-    {"name": "Latvia", "url": "/latvia-facts-for-kids.html"},
-    {"name": "Lesotho", "url": "/lesotho-facts.html"},
-    {"name": "Liberia", "url": "/facts-about-liberia.html"},
-    {"name": "Lithuania", "url": "/lithuania-facts-for-kids.html"},
-    {"name": "Luxembourg", "url": "/luxembourg-facts.html"},
-    {"name": "Macao", "url": "/facts-about-macao.html"},
-    {"name": "Madagascar", "url": "/facts-about-madagascar.html"},
-    {"name": "Malaysia", "url": "/malaysia-facts.html"},
-    {"name": "Maldives", "url": "/maldives-facts.html"},
-    {"name": "Malta", "url": "/malta-for-kids.html"},
-    {"name": "Mauritius", "url": "/mauritius-facts.html"},
-    {"name": "Mexico", "url": "/mexico-facts.html"},
-    {"name": "Micronesia", "url": "/facts-about-micronesia.html"},
-    {"name": "Moldova", "url": "/moldova-facts-for-kids.html"},
-    {"name": "Monaco", "url": "/facts-about-monaco.html"},
-    {"name": "Morocco", "url": "/morocco-facts.html"},
-    {"name": "Mozambique", "url": "/mozambique-facts.html"},
-    {"name": "Myanmar", "url": "/myanmar-facts.html"},
-    {"name": "Namibia", "url": "/namibia-facts.html"},
-    {"name": "Nauru", "url": "/facts-about-nauru.html"},
-    {"name": "Nepal", "url": "/nepal-facts.html"},
-    {"name": "Netherlands", "url": "/facts-about-the-netherlands.html"},
-    {"name": "New Zealand", "url": "/new-zealand-facts.html"},
-    {"name": "Nicaragua", "url": "/nicaragua-facts.html"},
-    {"name": "Nigeria", "url": "/nigeria-facts.html"},
-    {"name": "Norway", "url": "/norway-facts.html"},
-    {"name": "Pakistan", "url": "/pakistan-facts.html"},
-    {"name": "Panama", "url": "/panama-facts.html"},
-    {"name": "Papua New Guinea", "url": "/papua-new-guinea.html"},
-    {"name": "Peru", "url": "/peru-facts.html"},
-    {"name": "Philippines", "url": "/philippines-facts.html"},
-    {"name": "Poland", "url": "/poland-facts.html"},
-    {"name": "Portugal", "url": "/portugal-facts.html"},
-    {"name": "Puerto Rico", "url": "/facts-about-puerto-rico.html"},
-    {"name": "Qatar", "url": "/qatar-facts.html"},
-    {"name": "Romania", "url": "/romania-facts-for-kids.html"},
-    {"name": "Russia", "url": "/russia-facts.html"},
-    {"name": "Samoa", "url": "/facts-about-samoa.html"},
-    {"name": "San Marino", "url": "/facts-about-san-marino.html"},
-    {"name": "Serbia", "url": "/facts-about-serbia.html"},
-    {"name": "Seychelles", "url": "/seychelles-facts.html"},
-    {"name": "Singapore", "url": "/singapore-facts.html"},
-    {"name": "Solomon Islands", "url": "/solomon-islands-facts.html"},
-    {"name": "South Africa", "url": "/south-africa-for-kids.html"},
-    {"name": "South Korea", "url": "/south-korea-facts.html"},
-    {"name": "Spain", "url": "/spain-facts.html"},
-    {"name": "Sri Lanka", "url": "/sri-lanka-facts.html"},
-    {"name": "Suriname", "url": "/suriname-facts.html"},
-    {"name": "Sweden", "url": "/Sweden-facts.html"},
-    {"name": "Switzerland", "url": "/switzerland-facts.html"},
-    {"name": "Taiwan", "url": "/taiwan-facts.html"},
-    {"name": "Tanzania", "url": "/tanzania-facts.html"},
-    {"name": "Thailand", "url": "/thailand-facts.html"},
-    {"name": "Togo", "url": "/togo-facts-for-kids.html"},
-    {"name": "Tonga", "url": "/facts-about-tonga.html"},
-    {"name": "Tunisia", "url": "/tunisia-facts.html"},
-    {"name": "Türkiye", "url": "/turkey-facts.html"},
-    {"name": "Tuvalu", "url": "/facts-about-tuvalu.html"},
-    {"name": "Uganda", "url": "/facts-about-uganda.html"},
-    {"name": "Ukraine", "url": "/ukraine-for-kids.html"},
-    {"name": "United Arab Emirates", "url": "/uae-facts.html"},
-    {"name": "United Kingdom", "url": "/uk-facts.html"},
-    {"name": "United States of America", "url": "/usa-facts.html"},
-    {"name": "Uruguay", "url": "/uruguay-facts.html"},
-    {"name": "Vanuatu", "url": "/facts-about-vanuatu.html"},
-    {"name": "Venezuela", "url": "/venezuela-for-kids.html"},
-    {"name": "Vietnam", "url": "/vietnam-facts.html"},
-    {"name": "Zambia", "url": "/zambia-facts.html"}
-]
+import json
 
-def fetch_country_facts(country_url):
-    """Fetch facts about a country from the Kids World Travel Guide website"""
-    base_url = "https://www.kids-world-travel-guide.com"
-    full_url = base_url + country_url
-    
-    try:
-        response = requests.get(full_url, timeout=10)
-        response.raise_for_status()
-        
-        # Parse the HTML content
-        soup = BeautifulSoup(response.content, 'html.parser')
-        
-        # Extract relevant facts - looking for common patterns in the website
-        facts = []
-        
-        # Look for fact sections, lists, and key information
-        # This is a basic parser - you might need to adjust based on the actual HTML structure
-        
-        # Try to find paragraphs with factual content
-        paragraphs = soup.find_all('p')
-        for p in paragraphs[:10]:  # Limit to first 10 paragraphs to avoid too much content
-            text = p.get_text().strip()
-            if len(text) > 50 and not text.startswith('Related'):  # Filter out short texts and navigation
-                facts.append(text)
-        
-        # Look for list items that might contain facts
-        list_items = soup.find_all('li')
-        for li in list_items[:15]:  # Limit to avoid too much content
-            text = li.get_text().strip()
-            if len(text) > 20 and len(text) < 200:  # Filter for reasonable fact lengths
-                facts.append(text)
-        
-        # Join facts with newlines, limit total length
-        facts_text = '\n'.join(facts[:10])  # Limit to 10 facts
-        
-        # Truncate if too long to avoid token limits
-        if len(facts_text) > 2000:
-            facts_text = facts_text[:2000] + "..."
-            
-        return facts_text
-        
-    except Exception as e:
-        print(f"Error fetching facts for {country_url}: {str(e)}")
-        return "Unable to fetch additional facts about this country."
+# Load country list from countries.json
+COUNTRIES_JSON_PATH = os.path.join(os.path.dirname(__file__), "countries.json")
+with open(COUNTRIES_JSON_PATH, "r", encoding="utf-8") as f:
+    GEOGRAPHY_GUESS_LIST = json.load(f)
 
-def get_system_message_with_country():
-    """Generate a system message with a randomly selected country"""
-    global selected_country, selected_country_dict
-    selected_country_dict = random.choice(COUNTRIES)
-    selected_country = selected_country_dict["name"]
+STATES_JSON_PATH = os.path.join(os.path.dirname(__file__), "us_states.json")
+with open(STATES_JSON_PATH, "r", encoding="utf-8") as f:
+    US_STATES_GUESS_LIST = json.load(f)
+
+# Game mode constants
+GAME_MODE_COUNTRIES = "Countries of the World"
+GAME_MODE_STATES = "US States"
+
+# Global variables for current game state
+current_system = ""
+selected_country = ""
+selected_country_dict = {}
+game_mode = GAME_MODE_COUNTRIES
+
+
+
+def get_system_message(mode):
+    """Generate a system message with a randomly selected location based on game mode"""
+    global selected_country, selected_country_dict, game_mode
+    
+    if mode == GAME_MODE_STATES:
+        selected_country_dict = random.choice(US_STATES_GUESS_LIST)
+        location_type = "state"
+        location_type_upper = "US state"
+    else:
+        selected_country_dict = random.choice(GEOGRAPHY_GUESS_LIST)
+        location_type = "country"
+        location_type_upper = "country"
     
-    # Fetch facts about the selected country
-    print(f"Selected country for this session: {selected_country}")
+    selected_country = selected_country_dict["name"]
+    game_mode = mode
+
+    print(f"Selected {location_type} for this session: {selected_country}")
     print(f"Fetching facts from: {selected_country_dict['url']}")
     
-    country_facts = fetch_country_facts(selected_country_dict["url"])
-    print(country_facts)
+    facts = fetch_facts(game_mode, selected_country_dict["url"])
+    print(facts)
+
+    if mode == GAME_MODE_STATES:
+        location_info = f"""
+    State: {selected_country_dict['name']}
+    Capital: {selected_country_dict['capital']}
+    Nickname: {selected_country_dict['nickname']}
+    Facts: {facts}
+    """
+        info_label = "STATE INFORMATION"
+        reveal_text = "Do NOT reveal the state name"
+    else:
+        location_info = facts
+        info_label = "COUNTRY FACTS"
+        reveal_text = "Do NOT reveal the country name"
 
-    return f"""You are a friendly geography game host playing 20 questions with students. You are thinking of the country: {selected_country}
+    return f"""You are a friendly geography game host playing 20 questions with students. You are thinking of the {location_type_upper}: {selected_country}
 
-COUNTRY FACTS (use these to answer questions accurately - DO NOT reveal the country name):
-{country_facts}
+{info_label} (use these to answer questions accurately - {reveal_text}):
+{location_info}
 
 RULES:
-1. NEVER reveal the country name ({selected_country}) in your responses
+1. NEVER reveal the {location_type} name ({selected_country}) in your responses
 2. Answer only 'Yes' or 'No' to their questions
 3. Keep track of how many questions they've asked
-4. When they correctly guess or ask if it is {selected_country}, respond with: 'Congratulations! The country was <<{selected_country}>>'
-5. If they reach 20 questions without guessing correctly, respond with: 'Game over! The country was <<{selected_country}>>'
+4. When they correctly guess or ask if it is {selected_country}, respond with: 'Congratulations! The {location_type} was <<{selected_country}>>'
+5. If they reach 20 questions without guessing correctly, respond with: 'Game over! The {location_type} was <<{selected_country}>>'
 6. Be encouraging and give helpful hints through your yes/no answers
 7. If they want to play again tell them they need to reload the page.
-8. IMPORTANT: Only accept the country name "{selected_country}" as correct, but Spelling is not important and they can ask a question like it is? Do NOT accept neighboring countries, similar countries, or regions that contain this country.
-9. If they guess a neighboring country or similar country, respond with "No" and continue the game.
-10. Be very strict about the exact country match - only "{selected_country}" is the correct answer.
-11. Use the COUNTRY FACTS above to provide accurate yes/no answers - do not make up information."""
+8. IMPORTANT: Only accept the {location_type} name "{selected_country}" as correct, but Spelling is not important and they can ask a question like it is? Do NOT accept neighboring {location_type}s, similar {location_type}s, or regions that contain this {location_type}.
+9. If they guess a neighboring {location_type} or similar {location_type}, respond with "No" and continue the game.
+10. Be very strict about the exact {location_type} match - only "{selected_country}" is the correct answer.
+11. Use the {info_label} above to provide accurate yes/no answers - do not make up information."""
 
-current_system = get_system_message_with_country()
+current_system = ""
 
 def clean_response(response):
     """Clean up the response by removing unwanted metadata and formatting artifacts"""
@@ -259,15 +135,18 @@ def clean_response(response):
 
 def format_game_result(response):
     """Format the game result with proper styling"""
-    if "The country was" in response:
-        print(f"🔍 DEBUG - Game end detected! Country extracted: {selected_country}")
+    global game_mode
+    location_type = "state" if game_mode == GAME_MODE_STATES else "country"
+
+    if f"The {location_type} was" in response:
+        print(f"🔍 DEBUG - Game end detected! {location_type.capitalize()} extracted: {selected_country}")
     else:
         print("🔍 DEBUG - Regular response (no game end)")
 
     if "Congratulations" in response:
         return f"🎉 **Congratulations!** You correctly guessed **{selected_country}**! Well done! 🎉\n\nTo play another round, please start a new conversation or reload the page."
     elif "Game over" in response:
-        return f"😔 **Game Over!** You've used all 20 questions. The country I was thinking of was **{selected_country}**. 😔\n\nTo try again, please start a new conversation or reload the page."
+        return f"😔 **Game Over!** You've used all 20 questions. The {location_type} I was thinking of was **{selected_country}**. 😔\n\nTo try again, please start a new conversation or reload the page."
     
     return response
 
@@ -279,6 +158,7 @@ def respond(
     max_tokens,
     temperature,
     top_p,
+    game_mode_selection,
     hf_token: gr.OAuthToken | None = None,
 ):
     """
@@ -286,10 +166,14 @@ def respond(
     """
     global current_system
     
-    # If this is the start of a new conversation (empty history), generate a new country
+    # If this is the start of a new conversation (empty history), generate a new country/state
     if not history:
-        current_system = get_system_message_with_country()
-        print(f"🔍 DEBUG - New session started, selected country: {selected_country}")
+        if game_mode_selection == GAME_MODE_STATES:
+            current_system = get_system_message(GAME_MODE_STATES)
+            print(f"🔍 DEBUG - New session started, selected state: {selected_country}")
+        else:
+            current_system = get_system_message(GAME_MODE_COUNTRIES)
+            print(f"🔍 DEBUG - New session started, selected country: {selected_country}")
 
     messages = [{"role": "system", "content": current_system}]
     messages.extend(history)
@@ -362,8 +246,8 @@ def respond(
         print(f"🔍 DEBUG - Cleaned response: {response}")
 
         # Check if this is a game end response and format it nicely
-        if "The country was" in response:
-            print(f"🔍 DEBUG - Game end detected! Country extracted: {selected_country}")
+        if "The country was" in response or "The state was" in response:
+            print(f"🔍 DEBUG - Game end detected! Location extracted: {selected_country}")
             return format_game_result(response)
         else:
             print("🔍 DEBUG - Regular response (no game end)")
@@ -378,64 +262,67 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 
 # Create description based on mode
 if LOCAL_MODE:
-    description = f"🏠 Running locally with {MODEL_NAME}. I am thinking of a country, you have 20 yes or no questions to ask me to help you figure out what the country is"
+    description = f"🏠 Running locally with {MODEL_NAME}. Choose your game mode and I'll think of a location for you to guess with 20 yes or no questions!"
 else:
-    description = "I am thinking of a country, you have 20 yes or no questions to ask me to help you figure out what the country is"
+    description = "Choose your game mode and I'll think of a location for you to guess with 20 yes or no questions!"
 
-# Common examples and settings
+# Common examples and settings - updated to be more generic
 examples = [
-    ["Is the country located in Europe?"],
+    ["Is it located in North America?"],
     ["Is it in the Northern Hemisphere?"],
-    ["Is the official language Spanish?"],
-    ["Is the capital city Rome?"],
-    ["Is this country bordered by an ocean?"],
-    ["Does this country have more than 100 million people?"],
-    ["Is this country known for producing coffee?"],
-    ["Was this country ever a colony of the United Kingdom?"],
-    ["Is this country located on an island?"],
-    ["Is the currency the Euro?"],
+    ["Does it border an ocean?"],
+    ["Is the official language English?"],
+    ["Does it have more than 10 million people?"],
+    ["Is it known for producing oil?"],
+    ["Was it ever a British colony?"],
+    ["Is it located on an island?"],
+    ["Does it use the US Dollar as currency?"],
+    ["Is it a landlocked location?"],
 ]
 
 # Create wrapper function that handles both local and cloud modes
-if LOCAL_MODE:
-    # Local mode - no OAuth needed
-    def custom_respond(message, history):
-        system_message = ""
-        max_tokens = 4000
-        temperature = 0.3
-        top_p = 0.6
-        return respond(message, history, system_message, max_tokens, temperature, top_p, None)
-    
-    chatbot = gr.ChatInterface(
-        custom_respond,
-        type="messages",
-        description=description,
-        examples=examples,
-        cache_examples=False,
-    )
-else:
-    # Cloud mode - use OAuth
-    chatbot = gr.ChatInterface(
-        respond,
-        type="messages",
-        description=description,
-        examples=examples,
-        cache_examples=False,
-        additional_inputs=[
-            gr.Textbox(value="", visible=False),  # system_message (hidden)
-            gr.Slider(minimum=1, maximum=4096, value=4000, visible=False),  # max_tokens (hidden)
-            gr.Slider(minimum=0.1, maximum=2.0, value=0.3, visible=False),  # temperature (hidden)
-            gr.Slider(minimum=0.1, maximum=1.0, value=0.6, visible=False),  # top_p (hidden)
-        ],
-    )
+# Local mode wrapper function
+def custom_respond(message, history, game_mode_selection):
+    return respond(message, history, "", 4000, 0.3, 0.6, game_mode_selection, None)
 
 with gr.Blocks() as demo:
     if not LOCAL_MODE:
-        # Only show login button when running in cloud mode
         with gr.Sidebar():
             gr.LoginButton()
-    chatbot.render()
-
+    
+    gr.Markdown("# 20 Questions Geography Game")
+    gr.Markdown(description)
+    
+    # Game mode selection at the top
+    game_mode_dropdown = gr.Dropdown(
+        choices=[GAME_MODE_STATES, GAME_MODE_COUNTRIES],
+        value=GAME_MODE_STATES,
+        label="Game Mode",
+        info="Choose what type of location to guess"
+    )
+    
+    if LOCAL_MODE:
+        chatbot = gr.ChatInterface(
+            custom_respond,
+            type="messages",
+            examples=examples,
+            cache_examples=False,
+            additional_inputs=[game_mode_dropdown],
+        )
+    else:
+        chatbot = gr.ChatInterface(
+            respond,
+            type="messages",
+            examples=examples,
+            cache_examples=False,
+            additional_inputs=[
+                gr.Textbox(value="", visible=False),  # system_message (hidden)
+                gr.Slider(minimum=1, maximum=4096, value=4000, visible=False),  # max_tokens (hidden)
+                gr.Slider(minimum=0.1, maximum=2.0, value=0.3, visible=False),  # temperature (hidden)
+                gr.Slider(minimum=0.1, maximum=1.0, value=0.6, visible=False),  # top_p (hidden)
+                game_mode_dropdown,
+            ],
+        )
 
 if __name__ == "__main__":
     demo.launch()

countries.json

@@ -0,0 +1,124 @@
+[
+    {"name": "Algeria", "url": "/algeria-facts-for-kids.html"},
+    {"name": "Angola", "url": "/facts-about-angola.html"},
+    {"name": "Argentina", "url": "/argentina-facts.html"},
+    {"name": "Australia", "url": "/australia-facts.html"},
+    {"name": "Austria", "url": "/austria-facts.html"},
+    {"name": "Bahamas", "url": "/facts-about-the-bahamas.html"},
+    {"name": "Barbados", "url": "/barbados-facts.html"},
+    {"name": "Belgium", "url": "/belgium-facts.html"},
+    {"name": "Belize", "url": "/facts-about-belize.html"},
+    {"name": "Bhutan", "url": "/bhutan-facts.html"},
+    {"name": "Bolivia", "url": "/bolivia-facts.html"},
+    {"name": "Botswana", "url": "/facts-about-botswana.html"},
+    {"name": "Brazil", "url": "/brazil-facts.html"},
+    {"name": "Bulgaria", "url": "/facts-about-bulgaria.html"},
+    {"name": "Canada", "url": "/canada-facts-for-kids.html"},
+    {"name": "Chile", "url": "/chile-facts.html"},
+    {"name": "China", "url": "/china-facts.html"},
+    {"name": "Colombia", "url": "/colombia-facts.html"},
+    {"name": "Comoros", "url": "/comoros-facts.html"},
+    {"name": "Costa Rica", "url": "/costa-rica-facts.html"},
+    {"name": "Croatia", "url": "/croatia-facts.html"},
+    {"name": "Cuba", "url": "/cuba-facts.html"},
+    {"name": "Cyprus", "url": "/cyprus-facts-for-kids.html"},
+    {"name": "Denmark", "url": "/denmark-facts.html"},
+    {"name": "Dominican Republic", "url": "/dominican-republic-facts.html"},
+    {"name": "Ecuador", "url": "/ecuador-facts.html"},
+    {"name": "Egypt", "url": "/egypt-facts.html"},
+    {"name": "Estonia", "url": "/estonia-facts-for-kids.html"},
+    {"name": "Eswatini", "url": "/eswatini-facts.html"},
+    {"name": "Ethiopia", "url": "/ethiopia-facts.html"},
+    {"name": "Fiji", "url": "/facts-about-fiji.html"},
+    {"name": "Finland", "url": "/finland-facts.html"},
+    {"name": "France", "url": "/france-facts.html"},
+    {"name": "Georgia", "url": "/georgia-facts.html"},
+    {"name": "Germany", "url": "/germany-facts.html"},
+    {"name": "Ghana", "url": "/ghana-facts.html"},
+    {"name": "Greece", "url": "/greece-facts.html"},
+    {"name": "Greenland", "url": "/facts-about-greenland.html"},
+    {"name": "Guatemala", "url": "/guatemala-facts.html"},
+    {"name": "Guyana", "url": "/guyana-facts.html"},
+    {"name": "Honduras", "url": "/honduras-facts-for-kids.html"},
+    {"name": "Hong Kong", "url": "/facts-about-hong-kong.html"},
+    {"name": "Hungary", "url": "/hungary-facts.html"},
+    {"name": "Iceland", "url": "/iceland-facts.html"},
+    {"name": "India", "url": "/india-for-kids.html"},
+    {"name": "Indonesia", "url": "/indonesia-facts.html"},
+    {"name": "Iran", "url": "/iran-facts-for-kids.html"},
+    {"name": "Ireland", "url": "/ireland-for-kids.html"},
+    {"name": "Israel", "url": "/israel-facts.html"},
+    {"name": "Italy", "url": "/italy-facts.html"},
+    {"name": "Jamaica", "url": "/jamaica-facts.html"},
+    {"name": "Japan", "url": "/japan-facts.html"},
+    {"name": "Kenya", "url": "/facts-about-kenya.html"},
+    {"name": "Kiribati", "url": "/facts-about-kiribati.html"},
+    {"name": "Latvia", "url": "/latvia-facts-for-kids.html"},
+    {"name": "Lesotho", "url": "/lesotho-facts.html"},
+    {"name": "Liberia", "url": "/facts-about-liberia.html"},
+    {"name": "Lithuania", "url": "/lithuania-facts-for-kids.html"},
+    {"name": "Luxembourg", "url": "/luxembourg-facts.html"},
+    {"name": "Macao", "url": "/facts-about-macao.html"},
+    {"name": "Madagascar", "url": "/facts-about-madagascar.html"},
+    {"name": "Malaysia", "url": "/malaysia-facts.html"},
+    {"name": "Maldives", "url": "/maldives-facts.html"},
+    {"name": "Malta", "url": "/malta-for-kids.html"},
+    {"name": "Mauritius", "url": "/mauritius-facts.html"},
+    {"name": "Mexico", "url": "/mexico-facts.html"},
+    {"name": "Micronesia", "url": "/facts-about-micronesia.html"},
+    {"name": "Moldova", "url": "/moldova-facts-for-kids.html"},
+    {"name": "Monaco", "url": "/facts-about-monaco.html"},
+    {"name": "Morocco", "url": "/morocco-facts.html"},
+    {"name": "Mozambique", "url": "/mozambique-facts.html"},
+    {"name": "Myanmar", "url": "/myanmar-facts.html"},
+    {"name": "Namibia", "url": "/namibia-facts.html"},
+    {"name": "Nauru", "url": "/facts-about-nauru.html"},
+    {"name": "Nepal", "url": "/nepal-facts.html"},
+    {"name": "Netherlands", "url": "/facts-about-the-netherlands.html"},
+    {"name": "New Zealand", "url": "/new-zealand-facts.html"},
+    {"name": "Nicaragua", "url": "/nicaragua-facts.html"},
+    {"name": "Nigeria", "url": "/nigeria-facts.html"},
+    {"name": "Norway", "url": "/norway-facts.html"},
+    {"name": "Pakistan", "url": "/pakistan-facts.html"},
+    {"name": "Panama", "url": "/panama-facts.html"},
+    {"name": "Papua New Guinea", "url": "/papua-new-guinea.html"},
+    {"name": "Peru", "url": "/peru-facts.html"},
+    {"name": "Philippines", "url": "/philippines-facts.html"},
+    {"name": "Poland", "url": "/poland-facts.html"},
+    {"name": "Portugal", "url": "/portugal-facts.html"},
+    {"name": "Puerto Rico", "url": "/facts-about-puerto-rico.html"},
+    {"name": "Qatar", "url": "/qatar-facts.html"},
+    {"name": "Romania", "url": "/romania-facts-for-kids.html"},
+    {"name": "Russia", "url": "/russia-facts.html"},
+    {"name": "Samoa", "url": "/facts-about-samoa.html"},
+    {"name": "San Marino", "url": "/facts-about-san-marino.html"},
+    {"name": "Serbia", "url": "/facts-about-serbia.html"},
+    {"name": "Seychelles", "url": "/seychelles-facts.html"},
+    {"name": "Singapore", "url": "/singapore-facts.html"},
+    {"name": "Solomon Islands", "url": "/solomon-islands-facts.html"},
+    {"name": "South Africa", "url": "/south-africa-for-kids.html"},
+    {"name": "South Korea", "url": "/south-korea-facts.html"},
+    {"name": "Spain", "url": "/spain-facts.html"},
+    {"name": "Sri Lanka", "url": "/sri-lanka-facts.html"},
+    {"name": "Suriname", "url": "/suriname-facts.html"},
+    {"name": "Sweden", "url": "/Sweden-facts.html"},
+    {"name": "Switzerland", "url": "/switzerland-facts.html"},
+    {"name": "Taiwan", "url": "/taiwan-facts.html"},
+    {"name": "Tanzania", "url": "/tanzania-facts.html"},
+    {"name": "Thailand", "url": "/thailand-facts.html"},
+    {"name": "Togo", "url": "/togo-facts-for-kids.html"},
+    {"name": "Tonga", "url": "/facts-about-tonga.html"},
+    {"name": "Tunisia", "url": "/tunisia-facts.html"},
+    {"name": "Türkiye", "url": "/turkey-facts.html"},
+    {"name": "Tuvalu", "url": "/facts-about-tuvalu.html"},
+    {"name": "Uganda", "url": "/facts-about-uganda.html"},
+    {"name": "Ukraine", "url": "/ukraine-for-kids.html"},
+    {"name": "United Arab Emirates", "url": "/uae-facts.html"},
+    {"name": "United Kingdom", "url": "/uk-facts.html"},
+    {"name": "United States of America", "url": "/usa-facts.html"},
+    {"name": "Uruguay", "url": "/uruguay-facts.html"},
+    {"name": "Vanuatu", "url": "/facts-about-vanuatu.html"},
+    {"name": "Venezuela", "url": "/venezuela-for-kids.html"},
+    {"name": "Vietnam", "url": "/vietnam-facts.html"},
+    {"name": "Zambia", "url": "/zambia-facts.html"}
+]

rag.py

@@ -0,0 +1,48 @@
+
+import requests
+from bs4 import BeautifulSoup
+
+def fetch_facts(game_mode, path):
+    """Fetch facts about a country from the Kids World Travel Guide website"""
+    base_url = "https://www.kids-world-travel-guide.com" if game_mode == "Countries of the World" else "https://kids.nationalgeographic.com/geography/states/article/"
+    full_url = base_url + path
+    
+    try:
+        response = requests.get(full_url, timeout=10)
+        response.raise_for_status()
+        
+        # Parse the HTML content
+        soup = BeautifulSoup(response.content, 'html.parser')
+        
+        # Extract relevant facts - looking for common patterns in the website
+        facts = []
+        
+        # Look for fact sections, lists, and key information
+        # This is a basic parser - you might need to adjust based on the actual HTML structure
+        
+        # Try to find paragraphs with factual content
+        paragraphs = soup.find_all('p')
+        for p in paragraphs[:10]:  # Limit to first 10 paragraphs to avoid too much content
+            text = p.get_text().strip()
+            if len(text) > 50 and not text.startswith('Related'):  # Filter out short texts and navigation
+                facts.append(text)
+        
+        # Look for list items that might contain facts
+        list_items = soup.find_all('li')
+        for li in list_items[:20]:  # Limit to avoid too much content
+            text = li.get_text().strip()
+            if len(text) > 20 and len(text) < 450:  # Filter for reasonable fact lengths
+                facts.append(text)
+        
+        # Join facts with newlines, limit total length
+        facts_text = '\n'.join(facts[:20])  # Limit to 20 facts
+        
+        # Truncate if too long to avoid token limits
+        if len(facts_text) > 2000:
+            facts_text = facts_text[:2000] + "..."
+            
+        return facts_text
+        
+    except Exception as e:
+        print(f"Error fetching facts for {path}: {str(e)}")
+        return "Unable to fetch additional facts about this country."

us_states.json

@@ -0,0 +1,52 @@
+[
+    {"name": "Alabama", "capital": "Montgomery", "nickname": "The Heart of Dixie", "url": "alabama"},
+    {"name": "Alaska", "capital": "Juneau", "nickname": "The Last Frontier", "url": "alaska"},
+    {"name": "Arizona", "capital": "Phoenix", "nickname": "The Grand Canyon State", "url": "arizona"},
+    {"name": "Arkansas", "capital": "Little Rock", "nickname": "The Natural State", "url": "arkansas"},
+    {"name": "California", "capital": "Sacramento", "nickname": "The Golden State", "url": "california"},
+    {"name": "Colorado", "capital": "Denver", "nickname": "The Centennial State", "url": "colorado"},
+    {"name": "Connecticut", "capital": "Hartford", "nickname": "The Constitution State", "url": "connecticut"},
+    {"name": "Delaware", "capital": "Dover", "nickname": "The First State", "url": "delaware"},
+    {"name": "Florida", "capital": "Tallahassee", "nickname": "The Sunshine State", "url": "florida"},
+    {"name": "Georgia", "capital": "Atlanta", "nickname": "The Peach State", "url": "georgia"},
+    {"name": "Hawaii", "capital": "Honolulu", "nickname": "The Aloha State", "url": "hawaii"},
+    {"name": "Idaho", "capital": "Boise", "nickname": "The Gem State", "url": "idaho"},
+    {"name": "Illinois", "capital": "Springfield", "nickname": "The Prairie State", "url": "illinois"},
+    {"name": "Indiana", "capital": "Indianapolis", "nickname": "The Hoosier State", "url": "indiana"},
+    {"name": "Iowa", "capital": "Des Moines", "nickname": "The Hawkeye State", "url": "iowa"},
+    {"name": "Kansas", "capital": "Topeka", "nickname": "The Sunflower State", "url": "kansas"},
+    {"name": "Kentucky", "capital": "Frankfort", "nickname": "The Bluegrass State", "url": "kentucky"},
+    {"name": "Louisiana", "capital": "Baton Rouge", "nickname": "The Pelican State", "url": "louisiana"},
+    {"name": "Maine", "capital": "Augusta", "nickname": "The Pine Tree State", "url": "maine"},
+    {"name": "Maryland", "capital": "Annapolis", "nickname": "The Old Line State", "url": "maryland"},
+    {"name": "Massachusetts", "capital": "Boston", "nickname": "The Bay State", "url": "massachusetts"},
+    {"name": "Michigan", "capital": "Lansing", "nickname": "The Great Lakes State", "url": "michigan"},
+    {"name": "Minnesota", "capital": "Saint Paul", "nickname": "The North Star State", "url": "minnesota"},
+    {"name": "Mississippi", "capital": "Jackson", "nickname": "The Magnolia State", "url": "mississippi"},
+    {"name": "Missouri", "capital": "Jefferson City", "nickname": "The Show-Me State", "url": "missouri"},
+    {"name": "Montana", "capital": "Helena", "nickname": "The Treasure State", "url": "montana"},
+    {"name": "Nebraska", "capital": "Lincoln", "nickname": "The Cornhusker State", "url": "nebraska"},
+    {"name": "Nevada", "capital": "Carson City", "nickname": "The Silver State", "url": "nevada"},
+    {"name": "New Hampshire", "capital": "Concord", "nickname": "The Live Free or Die State", "url": "new-hampshire"},
+    {"name": "New Jersey", "capital": "Trenton", "nickname": "The Garden State", "url": "new-jersey"},
+    {"name": "New Mexico", "capital": "Santa Fe", "nickname": "The Land of Enchantment", "url": "new-mexico"},
+    {"name": "New York", "capital": "Albany", "nickname": "The Empire State", "url": "new-york"},
+    {"name": "North Carolina", "capital": "Raleigh", "nickname": "The Tar Heel State", "url": "north-carolina"},
+    {"name": "North Dakota", "capital": "Bismarck", "nickname": "The Peace Garden State", "url": "north-dakota"},
+    {"name": "Ohio", "capital": "Columbus", "nickname": "The Buckeye State", "url": "ohio"},
+    {"name": "Oklahoma", "capital": "Oklahoma City", "nickname": "The Sooner State", "url": "oklahoma"},
+    {"name": "Oregon", "capital": "Salem", "nickname": "The Beaver State", "url": "oregon"},
+    {"name": "Pennsylvania", "capital": "Harrisburg", "nickname": "The Keystone State", "url": "pennsylvania"},
+    {"name": "Rhode Island", "capital": "Providence", "nickname": "The Ocean State", "url": "rhode-island"},
+    {"name": "South Carolina", "capital": "Columbia", "nickname": "The Palmetto State", "url": "south-carolina"},
+    {"name": "South Dakota", "capital": "Pierre", "nickname": "The Mount Rushmore State", "url": "south-dakota"},
+    {"name": "Tennessee", "capital": "Nashville", "nickname": "The Volunteer State", "url": "tennessee"},
+    {"name": "Texas", "capital": "Austin", "nickname": "The Lone Star State", "url": "texas"},
+    {"name": "Utah", "capital": "Salt Lake City", "nickname": "The Beehive State", "url": "utah"},
+    {"name": "Vermont", "capital": "Montpelier", "nickname": "The Green Mountain State", "url": "vermont"},
+    {"name": "Virginia", "capital": "Richmond", "nickname": "The Old Dominion", "url": "virginia"},
+    {"name": "Washington", "capital": "Olympia", "nickname": "The Evergreen State", "url": "washington"},
+    {"name": "West Virginia", "capital": "Charleston", "nickname": "The Mountain State", "url": "west-virginia"},
+    {"name": "Wisconsin", "capital": "Madison", "nickname": "The Badger State", "url": "wisconsin"},
+    {"name": "Wyoming", "capital": "Cheyenne", "nickname": "The Equality State", "url": "wyoming"}
+]