Update app.py

app.py

@@ -9,99 +9,35 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def animal_battle(animal1: str, animal2: str) -> str:
+def animal_battle(arg1: str, arg2: str) -> str:
     """A tool that takes two animals and decides who would win in a fight and why
     Args:
-        animal1: the first animal
-        animal2: the second animal
-    Returns:
-        str: A description of the battle outcome and reasoning
+        arg1: the first animal
+        arg2: the second animal
     """
-    # Dictionary of animals and their characteristics
     animal_stats = {
-        "lion": {"strength": 9, "speed": 8, "size": 7, "weapons": "claws and teeth", "habitat": "savanna", "intelligence": 7},
-        "elephant": {"strength": 10, "speed": 5, "size": 10, "weapons": "tusks and trunk", "habitat": "savanna/forest", "intelligence": 9},
-        "gorilla": {"strength": 8, "speed": 6, "size": 6, "weapons": "strength and arms", "habitat": "forest", "intelligence": 8},
-        "tiger": {"strength": 9, "speed": 9, "size": 7, "weapons": "claws and teeth", "habitat": "jungle", "intelligence": 7},
-        "bear": {"strength": 9, "speed": 6, "size": 8, "weapons": "claws and strength", "habitat": "forest", "intelligence": 7},
-        "wolf": {"strength": 6, "speed": 8, "size": 5, "weapons": "teeth and pack tactics", "habitat": "forest", "intelligence": 8},
-        "rhinoceros": {"strength": 9, "speed": 6, "size": 9, "weapons": "horn and bulk", "habitat": "savanna", "intelligence": 6},
-        "hippopotamus": {"strength": 9, "speed": 5, "size": 9, "weapons": "jaws and bulk", "habitat": "water/land", "intelligence": 6},
-        "crocodile": {"strength": 8, "speed": 7, "size": 7, "weapons": "jaws and ambush", "habitat": "water", "intelligence": 5},
-        "anaconda": {"strength": 7, "speed": 6, "size": 6, "weapons": "constriction", "habitat": "water/jungle", "intelligence": 4},
-        "cheetah": {"strength": 6, "speed": 10, "size": 5, "weapons": "speed and agility", "habitat": "savanna", "intelligence": 7},
-        "kangaroo": {"strength": 7, "speed": 7, "size": 6, "weapons": "kicks and balance", "habitat": "grassland", "intelligence": 6},
-        "komodo dragon": {"strength": 6, "speed": 6, "size": 5, "weapons": "bacteria and venom", "habitat": "islands", "intelligence": 4},
-        "eagle": {"strength": 5, "speed": 9, "size": 4, "weapons": "talons and beak", "habitat": "air", "intelligence": 7},
+        "lion": {"power": 9, "special": "claws and teeth"},
+        "elephant": {"power": 10, "special": "size and strength"},
+        "gorilla": {"power": 8, "special": "strength and arms"},
+        "tiger": {"power": 9, "special": "speed and power"},
+        "bear": {"power": 9, "special": "claws and bulk"},
+        "wolf": {"power": 7, "special": "pack tactics"},
+        "eagle": {"power": 6, "special": "flight and talons"},
+        "komodo dragon": {"power": 7, "special": "venom and bite"}
     }
     
-    animal1 = animal1.lower()
-    animal2 = animal2.lower()
+    arg1 = arg1.lower()
+    arg2 = arg2.lower()
     
-    # Check if both animals are in our database
-    if animal1 not in animal_stats or animal2 not in animal_stats:
-        return f"Sorry, I don't have enough information about {'both' if animal1 not in animal_stats and animal2 not in animal_stats else animal1 if animal1 not in animal_stats else animal2} to determine the outcome."
+    if arg1 not in animal_stats or arg2 not in animal_stats:
+        return f"Sorry, I don't have information about one or both of these animals."
     
-    # Calculate total combat score with weighted attributes
-    def calculate_combat_score(animal):
-        stats = animal_stats[animal]
-        return (stats["strength"] * 1.5 + 
-                stats["speed"] * 1.2 + 
-                stats["size"] * 1.3 + 
-                stats["intelligence"] * 0.8)
-    
-    score1 = calculate_combat_score(animal1)
-    score2 = calculate_combat_score(animal2)
-    
-    # Consider habitat advantage
-    def has_habitat_advantage(attacker, defender):
-        attacker_habitat = animal_stats[attacker]["habitat"]
-        defender_habitat = animal_stats[defender]["habitat"]
-        if "water" in attacker_habitat and "water" not in defender_habitat:
-            return True
-        if "air" in attacker_habitat and "air" not in defender_habitat:
-            return True
-        return False
-    
-    # Adjust scores based on habitat advantage
-    if has_habitat_advantage(animal1, animal2):
-        score1 *= 1.2
-    if has_habitat_advantage(animal2, animal1):
-        score2 *= 1.2
-    
-    # Special case for extreme size differences
-    size_difference = abs(animal_stats[animal1]["size"] - animal_stats[animal2]["size"])
-    if size_difference > 5:
-        bigger_animal = animal1 if animal_stats[animal1]["size"] > animal_stats[animal2]["size"] else animal2
-        smaller_animal = animal2 if bigger_animal == animal1 else animal1
-        return f"The {bigger_animal} would win against the {smaller_animal}! The size difference is too great - " \
-               f"the {bigger_animal}'s sheer mass and strength would overwhelm the {smaller_animal}."
-    
-    # Determine winner and create detailed response
-    if abs(score1 - score2) < 2:
-        return f"It would be a close battle between the {animal1} and {animal2}! Both have their advantages: " \
-               f"the {animal1} with its {animal_stats[animal1]['weapons']}, and " \
-               f"the {animal2} with its {animal_stats[animal2]['weapons']}."
-    
-    winner = animal1 if score1 > score2 else animal2
-    loser = animal2 if score1 > score2 else animal1
-    winner_stats = animal_stats[winner]
-    
-    # Create detailed battle description
-    advantages = []
-    if winner_stats["strength"] > animal_stats[loser]["strength"]:
-        advantages.append("superior strength")
-    if winner_stats["speed"] > animal_stats[loser]["speed"]:
-        advantages.append("greater speed")
-    if winner_stats["size"] > animal_stats[loser]["size"]:
-        advantages.append("larger size")
-    if winner_stats["intelligence"] > animal_stats[loser]["intelligence"]:
-        advantages.append("better tactical ability")
-    
-    advantages_text = ", ".join(advantages[:-1]) + f" and {advantages[-1]}" if len(advantages) > 1 else advantages[0]
-    
-    return f"The {winner} would win against the {loser}! With its {winner_stats['weapons']}, " \
-           f"{advantages_text}, the {winner} has the clear advantage in this battle."
+    if animal_stats[arg1]["power"] > animal_stats[arg2]["power"]:
+        return f"The {arg1} would win with its {animal_stats[arg1]['special']}!"
+    elif animal_stats[arg2]["power"] > animal_stats[arg1]["power"]:
+        return f"The {arg2} would win with its {animal_stats[arg2]['special']}!"
+    else:
+        return f"It would be a close fight between the {arg1} and {arg2}!"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: