Update app.py

app.py

@@ -17,13 +17,19 @@ from Gradio_UI import GradioUI
 #         arg2: the second argument
 #     """
 #     return "What magic will you build ?"
+
+
 @tool
-def my_custom_tool(arg1: str, arg2: int) -> str:  # it's important to specify the return type
-    """A tool that fetches datasets related to body parts or medical imaging types (e.g., skin, brain, MRI, CT, etc.) from Hugging Face.
-    
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """
+    Fetch datasets related to a medical keyword such as body parts (e.g., skin, brain) or imaging modalities (e.g., MRI, CT).
+
     Args:
-        arg1: A body part or imaging modality keyword (e.g., 'skin', 'brain', 'XRay', 'ultrasound').
-        arg2: Maximum number of datasets to return.
+        arg1: A medical keyword such as 'skin', 'brain', 'XRay', or 'ultrasound'.
+        arg2: The maximum number of datasets to return.
+
+    Returns:
+        A formatted list of dataset names matching the keyword from Hugging Face.
     """
     try:
         query = arg1.lower()
@@ -44,7 +50,7 @@ def my_custom_tool(arg1: str, arg2: int) -> str:  # it's important to specify th
     except Exception as e:
         return f"Error fetching datasets for '{arg1}': {str(e)}"
 
-
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.