Update app.py

app.py

@@ -18,6 +18,33 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def inspect_directory(directory: str, prefix: str = '') -> str:
+    """
+    A tool that generates a visual tree structure of the directory contents.
+
+    Args:
+        directory (str): The path to the directory to inspect.
+        prefix (str): The prefix string used for formatting the tree structure.
+
+    Returns:
+        str: A string representing the tree structure of the directory.
+    """
+    tree_str = ''
+    contents = os.listdir(directory)
+    directories = sorted([d for d in contents if os.path.isdir(os.path.join(directory, d))], key=str.lower)
+    files = sorted([f for f in contents if os.path.isfile(os.path.join(directory, f))], key=str.lower)
+    contents = directories + files
+    pointers = [contents[-1]]
+    for content in contents:
+        pointer = '└── ' if content == pointers[0] else '├── '
+        tree_str += prefix + pointer + content + '\n'
+        path = os.path.join(directory, content)
+        if os.path.isdir(path):
+            extension = '    ' if content == pointers[0] else '│   '
+            tree_str += inspect_directory(path, prefix=prefix + extension)
+    return tree_str
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +82,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[inspect_directory, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,