Update app.py

app.py

@@ -82,7 +82,7 @@ class PokerSwarm:
         best_strategy = max(self.swarm, key=lambda x: x['win_prob'])
         return best_strategy['strategy'], best_strategy['win_prob']
 
-def predict_optimal_strategy(player_hand, pot_size, chips, n_opponents, flop_cards):
+def predict_optimal_strategy(player_hand, pot_size, chips, n_opponents, flop_cards, n_simulations):
     player_hand = player_hand.split(',')
     flop_cards = flop_cards.split(',')
     
@@ -90,7 +90,7 @@ def predict_optimal_strategy(player_hand, pot_size, chips, n_opponents, flop_car
     deck = [r + s for r in '23456789TJQKA' for s in 'SHDC']
     deck = [card for card in deck if card not in player_hand + flop_cards]
     
-    poker_swarm = PokerSwarm(n_particles=30, n_iterations=50, n_simulations=1000, player_hand=player_hand, pot_size=int(pot_size), chips=int(chips), n_opponents=int(n_opponents), flop_cards=flop_cards)
+    poker_swarm = PokerSwarm(n_particles=30, n_iterations=50, n_simulations=int(n_simulations), player_hand=player_hand, pot_size=int(pot_size), chips=int(chips), n_opponents=int(n_opponents), flop_cards=flop_cards)
     optimal_strategy, optimal_win_prob = poker_swarm.optimize()
     return optimal_strategy, f'{optimal_win_prob:.2%}'
 
@@ -100,7 +100,8 @@ inputs = [
     gr.components.Textbox(label="Pot Size"),
     gr.components.Textbox(label="Player's Amount of Chips"),
     gr.components.Textbox(label="Number of Opponents"),
-    gr.components.Textbox(label="Flop Cards (comma-separated, e.g., '2H,7D,9C')")
+    gr.components.Textbox(label="Flop Cards (comma-separated, e.g., '2H,7D,9C')"),
+    gr.components.Textbox(label="Number of Simulated Hands")
 ]
 
 outputs = [