Update app.py

Adding new tools

app.py

@@ -8,15 +8,178 @@ from tools.final_answer import FinalAnswerTool
 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 calculator(number1: float, number2: float, operation: str)-> float:
+    """A tool that does three kinds of maths operations
+    Args:
+        number1: The first float number, it can be int
+        number2: the second float number,  it can be int
+        operation: the type of math operation to carry out.
+    """
+    try: 
+        if type(number1)== int:
+            number1= float(number1)
+        if type(number2)== int:
+            number2= float(number2)
+        if operation == "sum":
+            result= number1 + number2
+        elif operation == "subtraction":
+            result= number1 - number2
+        elif operation == "division":
+            result= number1 /number2
+        elif operation == "multiplication":
+            result= number1 * number2
+
+        return result
+    except Exception as e:
+        return f"Error: {e}. Operation not surpported or missing arguments."
+            
+@tool 
+def sing_defying_gravity()-> str: #it's import to specify the return type
+
+    """A tool that returns the lyrics for Defying Gravity
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        no arguments
+    """
+    lyrics= """
+    [Glinda]
+    Elphaba
+    Why couldn't you have stayed calm for once
+    Instead of flying off the handle?
+    I hope you're happy
+    I hope you're happy now
+    I hope you're happy how you've hurt your cause forever
+    I hope you think you're clever
+    
+    [Elphaba]
+    I hope you're happy
+    I hope you're happy too
+    I hope you're proud how you would grovel in submission
+    To feed your own ambition
+    
+    [Galinda and Elphaba]
+    So though I can't imagine how
+    I hope you're happy right now
+    
+    [Glinda]
+    Elphie, listen to me
+    Just say you're sorry
+    You can still be with the wizard
+    What you've worked and waited for
+    You can have all you ever wanted
+    
+    [Elphaba]
+    I know
+    But I don't want it
+    No
+    I can't want it anymore
+    
+    Something has changed within me
+    Something is not the same
+    I'm through with playing by the rules of someone else's game
+    Too late for second-guessing
+    Too late to go back to sleep
+    It's time to trust my instincts
+    Close my eyes and leap
+    
+    It's time to try defying gravity
+    I think I'll try defying gravity
+    And you can't pull me down
+    
+    [Glinda]
+    Can't I make you understand
+    You're having delusions of grandeur?
+    
+    [Elphaba]
+    I'm through accepting limits
+    'Cause someone says they're so
+    Some things I cannot change
+    But till I try, I'll never know
+    Too long I've been afraid of
+    Losing love, I guess I've lost
+    Well, if that's love
+    It comes at much too high a cost
+    
+    I'd sooner buy defying gravity
+    Kiss me goodbye, I'm defying gravity
+    And you can't pull me down
+    
+    Glinda
+    Come with me
+    Think of what we could do
+    Together
+    
+    Unlimited
+    Together we're unlimited
+    Together we'll be the greatest team there's ever been, Glinda
+    Dreams the way we planned 'em
+    
+    [Glinda]
+    If we work in tandem
+    
+    [Glinda and Elphaba]
+    There's no fight we cannot win
+    Just you and I defying gravity
+    With you and I defying gravity
+    
+    [Elphaba]
+    They'll never bring us down
+    Well, are you coming?
+    
+    [Glinda]
+    I hope you're happy
+    Now that you're choosing this
+    
+    [Elphaba]
+    You too
+    I hope it brings you bliss
+    
+    [Glinda and Elphaba]
+    I really hope you get it
+    And you don't live to regret it
+    I hope you're happy in the end
+    I hope you're happy, my friend
+    
+    [Elphaba]
+    No, leave her alone
+    She hasn't done anything wrong
+    I'm the one you want
+    I'm the one you want
+    It's me
+    It's me
+    
+    So if you care to find me, look to the western sky
+    As someone told me lately
+    Everyone deserves the chance to fly
+    And if I'm flying solo
+    At least I'm flying free
+    To those who'd ground me
+    Take a message back from me
+    
+    Tell them how I am defying gravity
+    I'm flying high, defying gravity
+    And soon I'll match them in renown
+    
+    Unlimited
+    Unlimited
+    Unlimited, oh
+    
+    And nobody in all of Oz
+    No wizard that there is or was
+    Is ever gonna bring me down
+    
+    [Glinda]
+    I hope you're happy
+    
+    [Citizens of Oz and Elphaba]
+    Look at her, she's wicked, kill her!
+    (Bring me down) no one mourns the wicked
+    So we've got to bring her
+    (Oh) down
+    Down, woah
     """
-    return "What magic will you build ?"
+    return lyrics
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -49,13 +212,14 @@ custom_role_conversions=None,
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+search_tool = DuckDuckGoSearchTool()
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, search_tool, calculator,sing_defying_gravity], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,