Update app.py

app.py

@@ -6,6 +6,9 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+from transformers import AutoTokenizer
+
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -33,6 +36,36 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_demon_slayer_hashira(description: str) -> str:
+    """A tool that fetches the name of the Hashira, a title for top demon slayer characters in the Demon Slayer anime, based on the description of the 
+    Args:
+        description: The description of the Hashira(s) that we are looking for
+    """
+    desctiptions = """
+        Flame Hashira: Kyōjurō Rengoku is a young adult of tall stature and athletic build. He has long bright yellow hair with red streaks with two shoulder-length bangs and two chin-length bangs on the side of his head, black forked eyebrows, and golden eyes that fade to red with white pupils. 
+        Love Hashira: Mitsuri Kanroji is a young curvaceous woman with fair skin and round, light green eyes with long eyelashes, a small mole beneath each of them. She has long, pink hair that fades into a neon green color at the halfway point, which she wears in three thick braids with five shorter clumps hanging over her face as bangs. When she was trying to get married, her hair was dyed black in an attempt to make herself look more "normal". 
+        Stone Hashira: Gyōmei Himejima is a hulking figure among the Demon Slayers. He is one of the tallest characters in the series, easily towering over his fellow Hashira. He is powerfully built and extremely muscular. He has spiky black hair and a prominent scar running horizontally across his forehead. Having been blind since childhood, he has white eyes with no visible irises or pupils. His eyes frequently tear-up when he gets emotional. 
+        Mist Hashira: Muichirō Tokitō is a short, young man of light complexion with long, straight hair of a black color that fades into aquamarine/cyan as it approaches his waist. He wears his hair loose with two clumps protruding slightly from each side of his head, a set of marginally shorter hair left down to frame his face, and unruly bangs over his forehead. He also possesses large, empty-looking soft cyan eyes. His eyes and hair are traits shared with his twin brother, Yūichirō Tokitō. 
+        Snake Hashira: Obanai Iguro is a young man of short stature and a pale complexion. He has straight-edged black hair of varying lengths, the longest reaching down to his shoulders and the shortest stopping at his cheekbones, which he wears down with two shorter strands hanging between his eyes. His eyes are almond-shaped and tilt upwards on the far sides, and are unusual due to Obanai possessing heterochromia—his right eye is yellow and his left eye is turquoise. He is partially blind as he can barely see out of his right eye. 
+        Water Hashira: Giyū Tomioka is a fairly tall young man of a lean muscular stature and pale complexion, who is almost always seen wearing a serious and emotionless expression. He has unruly black hair of uneven lengths that sticks up in tufts around his head, which he wears tied back in a low messy ponytail at the base of his neck. His bangs fall over his eyes in an uneven fringe. His eyes are sharp and moderately large, their irises a deep sapphire that fades to a lighter blue and their pupils a bluish-black, and are framed by thin eyebrows. 
+    """
+    messages = [
+        {"role": "system", "content": ""},
+        {"role": "user", "content": "Hi !"},
+        {"role": "assistant", "content": "Hi human, what can help you with ?"},
+    ]
+    tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
+    rendered_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+    try:
+        # Create timezone object
+        tz = pytz.timezone(timezone)
+        # Get current time in that timezone
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+
 
 final_answer = FinalAnswerTool()