Update app.py

app.py

@@ -12,10 +12,19 @@ from Gradio_UI import GradioUI
 def my_custom_tool(arg1:str, arg2:int)-> 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 search and add prefix and appendix with "#"
+    #Args:
+    #    arg1: the first argument
+    #    arg2: the second argument
+    #"""
+
+    """A tool that searches using DuckDuckGoSearchTool and adds a prefix and an appendix consisting of '#' characters.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg1 (str): The search query.
+        arg2 (int): The number of '#' characters to add as prefix and appendix.
+
+    Returns:
+        str: The search results wrapped with the '#' prefix and appendix.
     """
     
     # Instantiate the duck_tool
@@ -24,18 +33,19 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     # Define your search query
     query = arg1
     count = arg2
-    appendix = ""
-    prefix = ""
+
+
+    # Create the prefix and appendix by repeating '#' count times
+    prefix = "#" * count
+    appendix = "#" * count
 
     # Use the .run() method to perform the search
-    for count > 0
-    appendix = appendix + "#"
-    prefix = "#" + prefix
-    
-    search_results = appendix + duck_tool.run(query) + appendix
+    search_results = duck_tool.run(query)
 
     #return "What magic will you build ?"
-    return search_results
+    
+    # Return the search results with the prefix and appendix added
+    return prefix + search_results + appendix
     
 
 @tool