Update app.py

app.py

@@ -1,6 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
+import random
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
@@ -28,6 +29,31 @@ def list_model_downloads(task: str)-> str:
     """
     most_downloaded_model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
     return most_downloaded_model.modelId
+
+
+@tool
+def get_email_template(sender: str, receiver: str)-> str:
+    """
+    A tool to generate an email template with an initial section addressing a person receiving the email and a
+    sign-off from the sender. 
+    NOTE: the body of the email is not filled in here.
+    Args:
+        sender: The name of the person who is sending the email.
+        receiver: The name of the person who will receive the email.
+    """
+    greetings = ["Hello", "Hi", "Dear"]
+    signoffs = ["Kind regards", "Thanks", "Sincerely"]
+    greeting = random.choice(greetings)
+    signoff = random.choice(signoffs)
+    email_template = f"""\
+    {greeting} {receiver}, I hope you are well.
+
+    
+    {signoff},
+    {sender}
+    """
+    return get_email_template
+    
     
 
 @tool
@@ -67,7 +93,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, list_model_downloads], ## add your tools here (don't remove final answer)
+    tools=[final_answer, list_model_downloads, get_email_template], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,