Update app.py

app.py

@@ -56,27 +56,18 @@ def shuffle_monsters():
     MONSTERS = shuffled
     
 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
+    # 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")