Add Bright Data Dataset Tool

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
-from tool import BrightDataDatasetTool
+import importlib
 
+BrightDataDatasetTool = importlib.import_module("tool").BrightDataDatasetTool
 tool = BrightDataDatasetTool()
 
 def run(dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments):
@@ -17,4 +18,20 @@ def run(dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews
 
 with gr.Blocks() as demo:
     gr.Markdown("### Bright Data dataset fetch")
-    dataset = gr.Dropdown(choices=
+    dataset = gr.Dropdown(choices=['amazon_product', 'amazon_product_reviews', 'amazon_product_search', 'apple_app_store', 'bestbuy_products', 'booking_hotel_listings', 'crunchbase_company', 'ebay_product', 'etsy_products', 'facebook_company_reviews', 'facebook_events', 'facebook_marketplace_listings', 'facebook_posts', 'github_repository_file', 'google_maps_reviews', 'google_play_store', 'google_shopping', 'homedepot_products', 'instagram_comments', 'instagram_posts', 'instagram_profiles', 'instagram_reels', 'linkedin_company_profile', 'linkedin_job_listings', 'linkedin_people_search', 'linkedin_person_profile', 'linkedin_posts', 'reddit_posts', 'reuter_news', 'tiktok_comments', 'tiktok_posts', 'tiktok_profiles', 'tiktok_shop', 'walmart_product', 'walmart_seller', 'x_posts', 'yahoo_finance_business', 'youtube_comments', 'youtube_profiles', 'youtube_videos', 'zara_products', 'zillow_properties_listing', 'zoominfo_company_profile'], label="Dataset", multiselect=False)
+    url = gr.Textbox(label="URL", placeholder="https://...")
+    keyword = gr.Textbox(label="Keyword")
+    first_name = gr.Textbox(label="First name")
+    last_name = gr.Textbox(label="Last name")
+    days_limit = gr.Textbox(label="Days limit (e.g. 3)")
+    num_of_reviews = gr.Textbox(label="Number of reviews")
+    num_of_comments = gr.Textbox(label="Number of comments")
+    run_btn = gr.Button("Run")
+    output = gr.Textbox(label="Output", lines=12)
+    run_btn.click(
+        run,
+        inputs=[dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments],
+        outputs=output,
+    )
+
+demo.launch()

requirements.txt

@@ -1,4 +1,2 @@
-gradio
 requests
-smolagents
-tool
+smolagents

tool.py

@@ -1,9 +1,9 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import os
-import time
 import requests
+import os
 import json
+import time
 
 class BrightDataDatasetTool(Tool):
     name = "brightdata_dataset_fetch"
@@ -200,13 +200,12 @@ class BrightDataDatasetTool(Tool):
             instead of a long text field. Kept minimal: single-select dropdown,
             rest are simple text boxes.
             """
-            # NOTE: Keep the import literal (no f-strings) so the requirements scanner
-            # in smolagents doesn't accidentally treat {tool_module_name} as a package.
-            choices = sorted(self.datasets.keys())
-            return """\
-    import gradio as gr
-    from tool import BrightDataDatasetTool
+            choices = repr(sorted(self.datasets.keys()))
+            # Avoid `from tool import ...` so requirement scanning doesn't add a fake package.
+            return f"""import gradio as gr
+    import importlib
 
+    BrightDataDatasetTool = importlib.import_module("tool").BrightDataDatasetTool
     tool = BrightDataDatasetTool()
 
     def run(dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments):
@@ -223,4 +222,21 @@ class BrightDataDatasetTool(Tool):
 
     with gr.Blocks() as demo:
         gr.Markdown("### Bright Data dataset fetch")
-        dataset = gr.Dropdown(choices="""
+        dataset = gr.Dropdown(choices={choices}, label="Dataset", multiselect=False)
+        url = gr.Textbox(label="URL", placeholder="https://...")
+        keyword = gr.Textbox(label="Keyword")
+        first_name = gr.Textbox(label="First name")
+        last_name = gr.Textbox(label="Last name")
+        days_limit = gr.Textbox(label="Days limit (e.g. 3)")
+        num_of_reviews = gr.Textbox(label="Number of reviews")
+        num_of_comments = gr.Textbox(label="Number of comments")
+        run_btn = gr.Button("Run")
+        output = gr.Textbox(label="Output", lines=12)
+        run_btn.click(
+            run,
+            inputs=[dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments],
+            outputs=output,
+        )
+
+    demo.launch()
+    """