Add Bright Data Search Tool

app.py

@@ -1,5 +1,18 @@
-from smolagents import launch_gradio_demo
-from tool import BrightDataSearchTool
+import gradio as gr
+import importlib
 
+BrightDataSearchTool = importlib.import_module("tool").BrightDataSearchTool
 tool = BrightDataSearchTool()
-launch_gradio_demo(tool)
+
+def run(query, engine):
+    return tool(query=query, engine=engine)
+
+with gr.Blocks() as demo:
+    gr.Markdown("### Bright Data search")
+    query = gr.Textbox(label="Query")
+    engine = gr.Dropdown(choices=['google', 'bing', 'yandex'], label="Engine", value="google")
+    run_btn = gr.Button("Run")
+    output = gr.Textbox(label="Output", lines=12)
+    run_btn.click(run, inputs=[query, engine], outputs=output)
+
+demo.launch()

tool.py

@@ -94,3 +94,29 @@ class BrightDataSearchTool(Tool):
 
         except requests.exceptions.RequestException as e:
             return json.dumps({"error": str(e)})
+
+    def _get_gradio_app_code(self, tool_module_name: str = "tool") -> str:
+            """
+            Custom Gradio UI: dropdown for engine, textbox for query.
+            Uses importlib to avoid adding a fake 'tool' requirement.
+            """
+            choices = ["google", "bing", "yandex"]
+            return f"""import gradio as gr
+    import importlib
+
+    BrightDataSearchTool = importlib.import_module("tool").BrightDataSearchTool
+    tool = BrightDataSearchTool()
+
+    def run(query, engine):
+        return tool(query=query, engine=engine)
+
+    with gr.Blocks() as demo:
+        gr.Markdown("### Bright Data search")
+        query = gr.Textbox(label="Query")
+        engine = gr.Dropdown(choices={choices}, label="Engine", value="google")
+        run_btn = gr.Button("Run")
+        output = gr.Textbox(label="Output", lines=12)
+        run_btn.click(run, inputs=[query, engine], outputs=output)
+
+    demo.launch()
+    """