Update app.py

app.py

@@ -1,71 +1,35 @@
 import gradio as gr
 import requests
 
-def calculator(num1, operation, num2):
-    if operation == "add":
-        return num1 + num2
-    elif operation == "subtract":
-        return num1 - num2
-    elif operation == "multiply":
-        return num1 * num2
-    elif operation == "divide":
-        return num1 / num2
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            num_1 = gr.Number(value=4)
-            operation = gr.Radio(["add", "subtract", "multiply", "divide"])
-            num_2 = gr.Number(value=0)
-            submit_btn = gr.Button(value="Calculate")
-        with gr.Column():
-            result = gr.Number()
-
-    submit_btn.click(
-        calculator, inputs=[num_1, operation, num_2], outputs=[result], api_name=False
-    )
-    examples = gr.Examples(
-        examples=[
-            [5, "add", 3],
-            [4, "divide", 2],
-            [-4, "multiply", 2.5],
-            [0, "subtract", 1.2],
-        ],
-        inputs=[num_1, operation, num_2],
-    )
-
-
 def register_tool(tool_data):
     # Send a POST request to register the tool
     response = requests.post("https://huggingface.co/chat/tools/new", json=tool_data)
     return response.json()
 
-def create_tool(display_name, description, inputs):
-    tool_data = {
-        "displayName": display_name,
-        "description": description,
-        "color": "yellow",  # Example color
-        "icon": "tools",
+def create_tool(tool_name, tool_data):
+    tool_payload = {
+        "displayName": tool_data.get("displayName"),
+        "description": tool_data.get("description"),
+        "color": tool_data.get("color"),  # Example color
+        "icon": tool_data.get("icon"),
         "baseUrl": "K00B404/toolshed",
-        "endpoint": "/query",
-        "name": "query",
+        "endpoint": "/router",
+        "name": tool_data.get("name"),
         "inputs": inputs,
-        "outputComponent": "0;image",  # Adjust based on your tool's output
-        "outputComponentIdx": 0,
-        "showOutput": True
+        "outputComponent": tool_data.get("outputComponent"),  # Adjust based on your tool's output
+        "outputComponentIdx": tool_data.get("outputComponentIdx"),
+        "showOutput": tool_data.get("showOutput")
     }
-    return register_tool(tool_data)
+    return register_tool(tool_payload)
 
-inputs=gr.Textbox(label="Display Name")
-outputs=gr.Textbox()
-fn=create_tool
+def tool_router(tool_name: str, tool_inputs: dict):
+    if tool_name == "create_tool":
+        create_tool(tool_name, tool_inputs)
 
 demo = gr.Interface(
-    fn=fn,
-    inputs=inputs,  # Updated line
-    outputs=outputs  # You can use gr.Textbox or other components as needed
+    fn=tool_router,
+    inputs=["text", {}],
+    outputs=["text", "number"],
 )
 
-
-
 demo.launch(show_api=True, share=True)