Update app.py

app.py

@@ -9,14 +9,38 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def my_custom_tool(str1:str, api_key: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 does nothing yet 
+     """
+    A tool that performs a search in the PubMed database using the NCBI E-utilities API.
+
+    This tool sends a request to the PubMed database with a specified search term and returns
+    the results in XML format. It requires an API key for authentication and rate limit management.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        str1 (str): The search term to query in the PubMed database (e.g., "cancer").
+        api_key (str): The API key for accessing the NCBI E-utilities API. This key is required
+                      to increase the rate limit and ensure uninterrupted access.
+
+    Returns:
+        str: The response from the PubMed database in XML format, containing the search results.
+
+    Example:
+        To search for articles related to "cancer" in PubMed:
+        >>> result = my_custom_tool("cancer", "your_api_key_here")
+        >>> print(result)  # Outputs the XML response with PubMed search results.
     """
-    return arg1 * arg2
+    base_url = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi"
+    params = {
+        "db": "pubmed",
+        "term": str1,
+        "retmax": 5,  # Ограничение количества результатов
+        "api_key": api_key  # Необязательно, но рекомендуется
+    }
+    
+    # Отправка запроса
+    response = requests.get(base_url, params=params)
+    return response.text
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +79,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,