Update app.py

app.py

@@ -7,76 +7,16 @@ 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_amazing_tool(partner_name: str, special_date: str) -> str:
-    """A tool to impress your partner with your knowledge of special dates.
-
-    Args:
-        partner_name: Your partner's name (for personalization).
-        special_date: A date in YYYY-MM-DD format (e.g., 2024-03-15).
-    """
-
-    try:
-        date_obj = datetime.datetime.strptime(special_date, "%Y-%m-%d").date()
-    except ValueError:
-        return "Oops! Invalid date format. Please use YYYY-MM-DD."
-
-    today = datetime.date.today()
-    days_until = (date_obj - today).days
-
-    if days_until > 0:
-        message = f"Hey {partner_name}, did you know that there are only {days_until} days until {date_obj.strftime('%B %d')}?  I was just thinking about it, and how much I'm looking forward to it!"
-    elif days_until == 0:
-        message = f"Happy {date_obj.strftime('%B %d')}, {partner_name}! I hope you have an amazing day. I was thinking of you all day."
-    else:
-        days_ago = abs(days_until)
-        message = f"{partner_name}, I was reminiscing about {date_obj.strftime('%B %d')}. It seems like just {days_ago} days ago! Such great memories."
-
-    # Add some extra flair (optional)
-    import random
-    emojis = ["😊", "😍", "🥰", "😘", "💖", "💕"]
-    message += f" {random.choice(emojis)}"
-
-    return message
-
-
-@tool
-def surprise_generator(partner_name: str, occasion: str) -> str:
-    """Generates a personalized surprise idea for your partner.
-
+def my_cutom_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 
     Args:
-        partner_name: Your partner's name.
-        occasion: The occasion (e.g., birthday, anniversary, just because).
+        arg1: the first argument
+        arg2: the second argument
     """
-
-    surprise_ideas = {
-        "birthday": [
-            "Plan a surprise party with their closest friends and family.",
-            "Create a personalized video montage with messages from loved ones.",
-            "Organize a scavenger hunt leading to their favorite gift or a special experience.",
-            "Bake their favorite cake or treat.",
-            "Book a weekend getaway to a place they've always wanted to visit."
-        ],
-        "anniversary": [
-            "Recreate your first date or a special memory together.",
-            "Write them a heartfelt letter expressing your love and appreciation.",
-            "Create a photo album or scrapbook documenting your relationship milestones.",
-            "Plan a romantic dinner at home or at their favorite restaurant.",
-            "Gift them a piece of jewelry with a special meaning."
-        ],
-        "just because": [
-            "Leave them a sweet note or message to brighten their day.",
-            "Surprise them with their favorite snack or drink.",
-            "Offer to help them with a task or chore they've been putting off.",
-            "Plan a spontaneous date night or outing.",
-            "Give them a thoughtful gift that shows you care."
-        ]
-    }
-
-    ideas = surprise_ideas.get(occasion.lower(), ["Plan a surprise!"])  # Default if occasion not found
-    chosen_idea = random.choice(ideas)
-
-    return f"Hey {partner_name}, for {occasion}, how about: {chosen_idea}  I think you'll love it!"
+    return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -104,14 +44,14 @@ custom_role_conversions=None,
 
 
 # Import tool from Hub
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[my_amazing_tool, surprise_generator, final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,