Update app.py

app.py

@@ -84,18 +84,27 @@ def main():
         PAYOFF = PRISONERS_DILEMMA_PAYOFF
     elif game_option == 'Battle of the Sexes':
         PAYOFF = BATTLE_SEXES_PAYOFF
-    else:
+    elif game_option == 'Custom Matrix':
         jj = st.text_input("Enter payoff for (J, J) as 'x,y':")
         jf = st.text_input("Enter payoff for (J, F) as 'x,y':")
         fj = st.text_input("Enter payoff for (F, J) as 'x,y':")
         ff = st.text_input("Enter payoff for (F, F) as 'x,y':")
 
-        PAYOFF = {
-            ("J", "J"): tuple(map(int, jj.split(','))),
-            ("J", "F"): tuple(map(int, jf.split(','))),
-            ("F", "J"): tuple(map(int, fj.split(','))),
-            ("F", "F"): tuple(map(int, ff.split(',')))
-        }
+        # Check if all inputs are filled
+        if not all([jj, jf, fj, ff]):
+            st.warning("Please fill in all the custom matrix values before starting the simulation.")
+            return
+
+        try:
+            PAYOFF = {
+                ("J", "J"): tuple(map(int, jj.split(','))),
+                ("J", "F"): tuple(map(int, jf.split(','))),
+                ("F", "J"): tuple(map(int, fj.split(','))),
+                ("F", "F"): tuple(map(int, ff.split(',')))
+            }
+        except ValueError:
+            st.error("Invalid custom matrix values. Please provide values in the format 'x,y'.")
+            return
 
     model_option = st.selectbox(
         'Which model version would you like to use?',
@@ -103,20 +112,23 @@ def main():
     )
     rounds = st.slider('Number of rounds to play', min_value=1, max_value=100, value=10, step=1)
 
-    st.write(f"Simulating {game_option} for {rounds} rounds using {model_option}.")
+    if st.button('Start Simulation'):
+        st.write(f"Simulating {game_option} for {rounds} rounds using {model_option}.")
+        last_round = None
+        for i in range(rounds):
+            player1_move = gpt_play(1, i + 1, last_round, model=model_option)
+            player2_move = gpt_play(2, i + 1, last_round, model=model_option)
 
-    last_round = None
-    for i in range(rounds):
-        player1_move = gpt_play(1, i + 1, last_round, model=model_option)
-        player2_move = gpt_play(2, i + 1, last_round, model=model_option)
+            last_round = (player1_move, player2_move)
+            score_p1, score_p2 = PAYOFF[last_round]
 
-        last_round = (player1_move, player2_move)
-        score_p1, score_p2 = PAYOFF[last_round]
+            st.write(
+                f"Round {i + 1}: Player 1 chose {player1_move} and won {score_p1} points. Player 2 chose {player2_move} and won {score_p2} points.")
 
-        st.write(
-            f"Round {i + 1}: Player 1 chose {player1_move} and won {score_p1} points. Player 2 chose {player2_move} and won {score_p2} points.")
+        st.write("Simulation complete!")
 
-    st.write("Simulation complete!")
+    if st.button('Reset'):
+        st.experimental_rerun()
 
 
 if __name__ == "__main__":