Update app.py

app.py

@@ -33,7 +33,7 @@ MONSTERS = {
     "Gorgon": {"Picture": "🐍👁️‍🗨️", "Description": "A creature with the body of a lion and the head of a snake, whose gaze can turn people to stone.", "Attack": 6, "Defense": 4},
 }
 
-def get_random_monster():
+def get_random_monster_old():
     monster_name = st.session_state.selected_monster
     if monster_name:
         monster = MONSTERS[monster_name]
@@ -45,6 +45,29 @@ def get_random_monster():
     else:
         return None
 
+def get_random_monster():
+    monster_name = st.session_state.selected_monster
+    if monster_name:
+        monster = MONSTERS[monster_name]
+        st.write("Picture: ", monster["Picture"])
+        st.write("Description: ", monster["Description"])
+        st.write("Attack: ", monster["Attack"])
+        st.write("Defense: ", monster["Defense"])
+        return monster
+    else:
+        # Select a random monster based on its size
+        monster_sizes = [monster_data["Attack"] + monster_data["Defense"] for monster_data in MONSTERS.values()]
+        selected_size = random.choice(monster_sizes)
+        monster_names = [monster_name for monster_name, monster_data in MONSTERS.items() if monster_data["Attack"] + monster_data["Defense"] == selected_size]
+        selected_monster = random.choice(monster_names)
+        st.session_state.selected_monster = selected_monster
+        monster = MONSTERS[selected_monster]
+        st.write("Picture: ", monster["Picture"])
+        st.write("Description: ", monster["Description"])
+        st.write("Attack: ", monster["Attack"])
+        st.write("Defense: ", monster["Defense"])
+        return monster
+
 def save_data():
     data = pd.DataFrame.from_dict(st.session_state.players, orient="index")
     filename = "player_data.csv"