Add Bright Data Dataset Tool

app.py

@@ -4,6 +4,16 @@ import importlib
 BrightDataDatasetTool = importlib.import_module("tool").BrightDataDatasetTool
 tool = BrightDataDatasetTool()
 
+DATASET_FIELDS = {'amazon_product': ['url'], 'amazon_product_reviews': ['url'], 'amazon_product_search': ['keyword', 'url'], 'walmart_product': ['url'], 'walmart_seller': ['url'], 'ebay_product': ['url'], 'homedepot_products': ['url'], 'zara_products': ['url'], 'etsy_products': ['url'], 'bestbuy_products': ['url'], 'linkedin_person_profile': ['url'], 'linkedin_company_profile': ['url'], 'linkedin_job_listings': ['url'], 'linkedin_posts': ['url'], 'linkedin_people_search': ['url', 'first_name', 'last_name'], 'crunchbase_company': ['url'], 'zoominfo_company_profile': ['url'], 'instagram_profiles': ['url'], 'instagram_posts': ['url'], 'instagram_reels': ['url'], 'instagram_comments': ['url'], 'facebook_posts': ['url'], 'facebook_marketplace_listings': ['url'], 'facebook_company_reviews': ['url', 'num_of_reviews'], 'facebook_events': ['url'], 'tiktok_profiles': ['url'], 'tiktok_posts': ['url'], 'tiktok_shop': ['url'], 'tiktok_comments': ['url'], 'google_maps_reviews': ['url', 'days_limit'], 'google_shopping': ['url'], 'google_play_store': ['url'], 'apple_app_store': ['url'], 'reuter_news': ['url'], 'github_repository_file': ['url'], 'yahoo_finance_business': ['url'], 'x_posts': ['url'], 'zillow_properties_listing': ['url'], 'booking_hotel_listings': ['url'], 'youtube_profiles': ['url'], 'youtube_comments': ['url', 'num_of_comments'], 'reddit_posts': ['url'], 'youtube_videos': ['url']}
+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']
+
+def toggle_fields(selected):
+    inputs = ["url", "keyword", "first_name", "last_name", "days_limit", "num_of_reviews", "num_of_comments"]
+    wanted = set(DATASET_FIELDS.get(selected, []))
+    def vis(name):
+        return gr.update(visible=name in wanted)
+    return tuple(vis(n) for n in inputs)
+
 def run(dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments):
     return tool(
         dataset=dataset,
@@ -18,14 +28,21 @@ 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=['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")
+    dataset = gr.Dropdown(choices=CHOICES, label="Dataset", multiselect=False, value=CHOICES[0])
+    url = gr.Textbox(label="URL", placeholder="https://...", visible=True)
+    keyword = gr.Textbox(label="Keyword", visible=False)
+    first_name = gr.Textbox(label="First name", visible=False)
+    last_name = gr.Textbox(label="Last name", visible=False)
+    days_limit = gr.Textbox(label="Days limit (e.g. 3)", visible=False)
+    num_of_reviews = gr.Textbox(label="Number of reviews", visible=False)
+    num_of_comments = gr.Textbox(label="Number of comments", visible=False)
+
+    dataset.change(
+        toggle_fields,
+        inputs=[dataset],
+        outputs=[url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments],
+    )
+
     run_btn = gr.Button("Run")
     output = gr.Textbox(label="Output", lines=12)
     run_btn.click(

tool.py

@@ -1,9 +1,9 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import requests
-import os
 import json
 import time
+import os
+import requests
 
 class BrightDataDatasetTool(Tool):
     name = "brightdata_dataset_fetch"
@@ -198,16 +198,26 @@ class BrightDataDatasetTool(Tool):
             """
             Override the default app to render a dropdown for dataset selection
             instead of a long text field. Kept minimal: single-select dropdown,
-            rest are simple text boxes.
+            and shows only relevant parameter fields for the chosen dataset.
             """
-            choices = repr(sorted(self.datasets.keys()))
-            # Avoid `from tool import ...` so requirement scanning doesn't add a fake package.
+            choices = sorted(self.datasets.keys())
+            dataset_fields = {k: v["inputs"] for k, v in self.datasets.items()}
             return f"""import gradio as gr
     import importlib
 
     BrightDataDatasetTool = importlib.import_module("tool").BrightDataDatasetTool
     tool = BrightDataDatasetTool()
 
+    DATASET_FIELDS = {dataset_fields}
+    CHOICES = {choices}
+
+    def toggle_fields(selected):
+        inputs = ["url", "keyword", "first_name", "last_name", "days_limit", "num_of_reviews", "num_of_comments"]
+        wanted = set(DATASET_FIELDS.get(selected, []))
+        def vis(name):
+            return gr.update(visible=name in wanted)
+        return tuple(vis(n) for n in inputs)
+
     def run(dataset, url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments):
         return tool(
             dataset=dataset,
@@ -222,14 +232,21 @@ class BrightDataDatasetTool(Tool):
 
     with gr.Blocks() as demo:
         gr.Markdown("### Bright Data dataset fetch")
-        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")
+        dataset = gr.Dropdown(choices=CHOICES, label="Dataset", multiselect=False, value=CHOICES[0])
+        url = gr.Textbox(label="URL", placeholder="https://...", visible=True)
+        keyword = gr.Textbox(label="Keyword", visible=False)
+        first_name = gr.Textbox(label="First name", visible=False)
+        last_name = gr.Textbox(label="Last name", visible=False)
+        days_limit = gr.Textbox(label="Days limit (e.g. 3)", visible=False)
+        num_of_reviews = gr.Textbox(label="Number of reviews", visible=False)
+        num_of_comments = gr.Textbox(label="Number of comments", visible=False)
+
+        dataset.change(
+            toggle_fields,
+            inputs=[dataset],
+            outputs=[url, keyword, first_name, last_name, days_limit, num_of_reviews, num_of_comments],
+        )
+
         run_btn = gr.Button("Run")
         output = gr.Textbox(label="Output", lines=12)
         run_btn.click(