Update app.py

app.py

@@ -11,11 +11,10 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_birth_stone(dob:str)-> str: #it's import to specify the return type
+def get_birth_stone(month: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 date of birth as input. It outputs the birthstone of that month present in the date of birth
-    Args:
-        dob: A string representing a person's date of birth
+    """A tool that takes a month string as input. It outputs the birthstone of that month 
+        dob: A string representing a month of the year
     """
 
     birthstones = {
@@ -33,7 +32,7 @@ def get_birth_stone(dob:str)-> str: #it's import to specify the return type
         "December": "Turquoise"
     }
         
-    month = datetime.strptime(dob, "%Y-%m-%d").strftime("%B")
+    month = month
     birthstone = birthstones.get(month, "Unknown")
     if birthstone in image_urls:
         response = birthstone