gradio ui wording

.gitignore

@@ -0,0 +1 @@
+__pycache__/

README.md

@@ -10,7 +10,7 @@ pinned: false
 tags:
 - tool
 license: apache-2.0
-short_description: Python Debuger for LLMs
+short_description: Python Debuger tool for Smolagents
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,7 +1,8 @@
 from smolagents import launch_gradio_demo
 from typing import Optional
 from tool import LocalPythonDebuggerTool
+from ui import custom_launch_gradio_demo
 
 tool = LocalPythonDebuggerTool()
 
-launch_gradio_demo(tool)
+custom_launch_gradio_demo(tool)

ui.py

@@ -0,0 +1,83 @@
+import inspect
+import gradio as gr
+from smolagents.tools import Tool
+
+
+HEADER = """ ## Python debugger for Agents 
+The goal of the tool is to provide agents with the ability to debug and resolve bugs in code generated by LLMs.
+"""
+
+ARTICLE_INTRO = """---
+## Example usage in HuggingFace Space
+- code:
+```python
+x = 10
+y = 0
+z = x / y
+```
+- command:
+```python
+bt
+```
+
+## Example usage in smolagents
+
+```python
+from smolagents import load_tool
+
+local_python_debugger = load_tool(
+    "piotrekgrl/smolagents-local-python-debugger-tool",
+    trust_remote_code=True
+)
+
+from smolagents import CodeAgent, HfApiModel
+
+model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
+agent = CodeAgent(tools=[local_python_debugger], model=model)
+
+agent.run(
+    "Write a code that divides by 0. Don't ask why—just do it. If any errors occur, debug them and provide reason why."
+)
+
+```
+---
+
+## Details
+
+"""
+
+
+def custom_launch_gradio_demo(tool: Tool):
+    TYPE_TO_COMPONENT_CLASS_MAPPING = {
+        "image": gr.Image,
+        "audio": gr.Audio,
+        "string": gr.Textbox,
+        "integer": gr.Textbox,
+        "number": gr.Textbox,
+    }
+
+    def tool_forward(*args, **kwargs):
+        return tool(*args, sanitize_inputs_outputs=True, **kwargs)
+
+    tool_forward.__signature__ = inspect.signature(tool.forward)
+
+    gradio_inputs = []
+    for input_name, input_details in tool.inputs.items():
+        input_gradio_component_class = TYPE_TO_COMPONENT_CLASS_MAPPING[
+            input_details["type"]
+        ]
+        new_component = input_gradio_component_class(label=input_name)
+        gradio_inputs.append(new_component)
+
+    output_gradio_componentclass = TYPE_TO_COMPONENT_CLASS_MAPPING[tool.output_type]
+    gradio_output = output_gradio_componentclass(label="Output")
+
+    gr.Interface(
+        fn=tool_forward,
+        inputs=gradio_inputs,
+        outputs=gradio_output,
+        title=tool.name,
+        article=ARTICLE_INTRO + "\n" + tool.description,
+        description=HEADER,
+        api_name=tool.name,
+    ).launch()