added Blackjack strategy tool

app.py

@@ -9,14 +9,48 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def blackjack_strategy(player_cards: list[str], dealer_card:str)-> str:
+    """A tool that determines optimal Blackjack strategy based on player's cards and dealer's face-up card. 
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        player_cards: A list of strings representing the player's cards.
+                      Each card is represented as two character string (ex: 'AH' for Ace of hearts, '10S' for 10 of Spades).     
+        dealer_card_rank: A string representing delaler's face-up card
+
+    Returns: Recommended action ("Hit", "Stand", or "Double Down"). 
     """
-    return "What magic will you build ?"
+    def card_value(card: str) -> int:
+        if card[0] == 'A': return 11
+        elif card[0] in 'KQJ': return 10
+        else: return int(card[:-1])
+
+    player_values = [card_value(card) for card in player_cards]
+    player_total = sum(player_values)
+    dealer_total = card_value(dealer_card)
+
+    if 11 in player_values and player_total > 21:
+        player_total -= 10
+    
+    if player_total >= 17:
+        return 'Stand'
+    elif player_total >= 13 and player_total <= 16:
+        if dealer_total >= 7:
+            return 'Hit'
+        else:
+            return 'Stand'
+    elif player_total == 12:
+        if dealer_total >= 4 and dealer_total <= 6:
+            return 'Stand'
+        else:
+            return 'Hit'
+    elif player_total <= 11:
+        # Double down condition (9, 10, or 11) - Only allowed if it's the first two cards
+        if len(player_cards) == 2 and (player_total in [9,10,11]):
+            if dealer_total >= 2 and dealer_total <= 9:  # Typical condition for doubling down
+                return 'Double Down'
+            else:
+                return 'Hit'
+        else:
+            return 'Hit'
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +89,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, blackjack_strategy], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,