Update app.py

app.py

@@ -3,12 +3,13 @@
 import gradio as gr
 import os
 import importlib.util
-from typing import Any, Dict
+import json
 import requests
+from typing import Any, Dict
 
 # Define the path to the tools folder
 TOOLS_DIR = './tools'
-os.makedirs(TOOLS_DIR, exists_ok=True)
+os.makedirs(TOOLS_DIR, exist_ok=True)
 
 # This is the only hardcoded Tool endpoint
 def register_tool(tool_data):
@@ -16,43 +17,44 @@ def register_tool(tool_data):
     response = requests.post("https://huggingface.co/chat/tools/new", json=tool_data)
     return response.json()
 
-def create_tool(tool_name, tool_data):
-    """One tool to create them all.
-        a AI model can wield this toolto create a new 'tool' script in the tools folder
-        besides the values for the registration payload,
-        the 'tool_data' should also be a string value 'tfn' which contains the code for the tool function script
+def create_tool(tool_name: str, tool_data: Dict[str, Any]):
+    """Create a new tool script in the tools folder.
+    
+    Args:
+        tool_name: Name for the tool (also the filename).
+        tool_data: Dictionary containing registration payload and function script.
     """
-    print("processing create_tool ")
-    print(tool_name)
-    print(dir(tool_data))
-    # set the payload for restration
+    print("Processing create_tool")
+    
+    # Prepare the payload for registration
     tool_payload = {
-        "displayName": tool_data.get("displayName"), # str display name of the tool
-        "description": tool_data.get("description"), # str dscription
-        "color": tool_data.get("color"),  # Example color purple,red,green
-        "icon": tool_data.get("icon"), # image binairy
-        "baseUrl": "K00B404/toolshed", # always to the toolshed
-        "endpoint": "/router", # always to the router
-        "name": tool_data.get("name"), # str name for matching function
-        "inputs":tool_data.get("inputs"), # List(str) list the tools input values
-        "outputComponent": tool_data.get("outputComponent"),  # Adjust based on your tool's output
-        "outputComponentIdx": tool_data.get("outputComponentIdx"), # Adjust based on your tool's output
-        "showOutput": tool_data.get("showOutput") # bool
+        "displayName": tool_data.get("displayName"),
+        "description": tool_data.get("description"),
+        "color": tool_data.get("color"),
+        "icon": tool_data.get("icon"),
+        "baseUrl": "K00B404/toolshed",
+        "endpoint": "/router",
+        "name": tool_data.get("name"),
+        "inputs": tool_data.get("inputs"),
+        "outputComponent": tool_data.get("outputComponent"),
+        "outputComponentIdx": tool_data.get("outputComponentIdx"),
+        "showOutput": tool_data.get("showOutput"),
     }
-    print(dir(tool_payload))
     
-    # write the new tool script from tfn
-    with open(f'{TOOLS_DIR}/{tool_data.get("name")}.py' 'w') as f:
-              f.write(tool_data.get("tfn"))
-    # send the registation request for the tool      
+    # Write the new tool script from the 'tfn' (tool function name) data
+    with open(f'{TOOLS_DIR}/{tool_data.get("name").lower()}.py', 'w') as f:
+        f.write(tool_data.get("tfn"))
+    
+    # Register the new tool
     return register_tool(tool_payload)
 
-
-
 # Function to dynamically load methods from files in the ./tools folder
 def load_methods_from_tools():
     method_mapping = {}
-    
+
+    # Manually map the create_tool function
+    method_mapping["create_tool"] = create_tool
+
     # List all .py files in the tools directory
     for filename in os.listdir(TOOLS_DIR):
         if filename.endswith('.py'):
@@ -76,6 +78,10 @@ method_mapping = load_methods_from_tools()
 
 # Main request router method
 def request_router(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
+    # Reload methods to include newly created tools
+    global method_mapping
+    method_mapping = load_methods_from_tools()
+
     # Retrieve the appropriate method based on the name
     method = method_mapping.get(name)
     if method is None:
@@ -99,7 +105,7 @@ def launch_gradio_app():
         
         submit_button = gr.Button("Submit")
         submit_button.click(
-            fn=lambda name, input_data: request_router(name, eval(input_data)),
+            fn=lambda name, input_data: request_router(name, json.loads(input_data)),
             inputs=[name_input, input_data_input],
             outputs=output_output
         )
@@ -107,4 +113,4 @@ def launch_gradio_app():
     demo.launch()
 
 if __name__ == "__main__":
-    launch_gradio_app()
+    launch_gradio_app()