Update app.py

app.py

@@ -23,14 +23,17 @@ def randomly_select_from_json(world_data):
 
     # Town level
     towns = kingdom_data["towns"]
-    town_name, town_data = random.choice(list(towns.items()))
+
+    # Filter towns with NPCs only
+    valid_towns = {name: data for name, data in towns.items() if data.get("npcs")}
+    if not valid_towns:
+        raise ValueError(f"No towns with NPCs found in kingdom: {kingdom_name}")
+
+    town_name, town_data = random.choice(list(valid_towns.items()))
     town_description = town_data["description"]
 
     # Character (NPC) level
-    npcs = town_data.get("npcs")  # Use .get() to handle missing 'npcs' key
-    if not npcs:
-        raise ValueError(f"No NPCs found in town: {town_name}")
-
+    npcs = town_data["npcs"]
     character_name, character_data = random.choice(list(npcs.items()))
     character_description = character_data["description"]