Update app.py

app.py

@@ -11,56 +11,66 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def destiny_still_arrives(name:str, age_or_dob:str)-> 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 takes a name, age or date of birth as input. It outputs the meaning of the name, an image of that person's birthstone and a horoscope reading of the day.
-    # Args:
-    #     name: A string representing a person's age
-    #     age_or_dob: A string representing a person's age or date of birth
-    # """
-    # def get_birthstone(name: str, age_or_dob: str):
-    #     birthstones = {
-    #         "January": "Garnet",
-    #         "February": "Amethyst",
-    #         "March": "Aquamarine",
-    #         "April": "Diamond",
-    #         "May": "Emerald",
-    #         "June": "Pearl",
-    #         "July": "Ruby",
-    #         "August": "Peridot",
-    #         "September": "Sapphire",
-    #         "October": "Opal",
-    #         "November": "Topaz",
-    #         "December": "Turquoise"
-    #     }
+def destiny_still_arrives(name:str, dob:str)-> 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 takes a name, age or date of birth as input. It outputs the meaning of the name, an image of that person's birthstone and a horoscope reading of the day.
+    Args:
+        name: A string representing a person's age
+        age_or_dob: A string representing a person's age or date of birth
+    """
+    def get_birthstone(name: str, dob: str):
+        birthstones = {
+            "January": "Garnet",
+            "February": "Amethyst",
+            "March": "Aquamarine",
+            "April": "Diamond",
+            "May": "Emerald",
+            "June": "Pearl",
+            "July": "Ruby",
+            "August": "Peridot",
+            "September": "Sapphire",
+            "October": "Opal",
+            "November": "Topaz",
+            "December": "Turquoise"
+        }
+        
+        month = datetime.strptime(dob, "%Y-%m-%d").strftime("%B")
+        birthstone = birthstones.get(month, "Unknown")
+                
+        model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+        custom_role_conversions=None,
+        )
         
-    #     month = datetime.strptime(age_or_dob, "%Y-%m-%d").strftime("%B")
-    #     birthstone = birthstones.get(month, "Unknown")
         
-    #     image_urls = {
-    #         "Garnet": "https://upload.wikimedia.org/wikipedia/commons/1/12/Garnet_rough.jpg",
-    #         "Amethyst": "https://upload.wikimedia.org/wikipedia/commons/8/8b/Amethyst.jpg",
-    #         "Aquamarine": "https://upload.wikimedia.org/wikipedia/commons/d/d5/Aquamarine_Brazil.jpg",
-    #         "Diamond": "https://upload.wikimedia.org/wikipedia/commons/a/a7/Loose_diamond.png",
-    #         "Emerald": "https://upload.wikimedia.org/wikipedia/commons/4/4f/Emerald.jpg",
-    #         "Pearl": "https://upload.wikimedia.org/wikipedia/commons/1/17/Pearl_macro.jpg",
-    #         "Ruby": "https://upload.wikimedia.org/wikipedia/commons/0/0d/Ruby.jpg",
-    #         "Peridot": "https://upload.wikimedia.org/wikipedia/commons/8/8a/Peridot.jpg",
-    #         "Sapphire": "https://upload.wikimedia.org/wikipedia/commons/3/35/Sapphire.jpg",
-    #         "Opal": "https://upload.wikimedia.org/wikipedia/commons/2/2f/Opal_macro.jpg",
-    #         "Topaz": "https://upload.wikimedia.org/wikipedia/commons/5/56/Topaz.jpg",
-    #         "Turquoise": "https://upload.wikimedia.org/wikipedia/commons/d/d6/Turquoise.jpg"
-    #     }
+        # Import tool from Hub
+        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=[final_answer,image_generation_tool], ## add your tools here (don't remove final answer)
+            max_steps=6,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=prompt_templates
+        )
+
+        
         
-    #     if birthstone in image_urls:
-    #         response = requests.get(image_urls[birthstone])
-    #         img = Image.open(BytesIO(response.content))
-    #         img.show()
-    #         return f"Hello {name}, your birthstone is {birthstone}!"
-    #     else:
-    #         return f"Hello {name}, we could not determine your birthstone."
+        if birthstone in image_urls:
+            response = agent.run(f"Generate an image of the following birthstone: {birthstone}")
+        else:
+            response = agent.run("Generate an image of the following birthstone: Sapphire")
     
-    return get_birthstone
+    return response
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: