add url query param support

app.py

@@ -13,6 +13,7 @@ client = InferenceClient(model_id)
 
 MAX_TOTAL_NB_ITEMS = 100  # almost infinite, don't judge me (actually it's because gradio needs a fixed number of components)
 MAX_NB_ITEMS_PER_GENERATION_CALL = 10
+URL = "https://huggingface.co/spaces/infinite-dataset-hub/infinite-dataset-hub"
 
 GENERATE_DATASET_NAMES_FOR_SEARCH_QUERY = (
         "A Machine Learning Practioner is looking for a dataset that matches '{search_query}'. "
@@ -168,11 +169,10 @@ with gr.Blocks(css=css) as demo:
                         buttons.append(gr.Button(dataset_name, elem_classes=dataset_name_classes))
                         buttons.append(gr.Button(tags, elem_classes=tags_classes))
 
-                see_more = gr.Button("See more")  # TODO: dosable when reaching end of page
+                load_more_datasets = gr.Button("Load more datasets")  # TODO: dosable when reaching end of page
                 gr.Markdown(f"_powered by [{model_id}](https://huggingface.co/{model_id})_")
             with gr.Column(scale=4, min_width=0):
                 pass
-        # more.click(search_more_datasets, inputs=[generated_texts, search_bar], outputs=[generated_texts] + buttons)
     with gr.Column(visible=False) as dataset_page:
         with gr.Row():
             with gr.Column(scale=4, min_width=0):
@@ -185,12 +185,15 @@ with gr.Blocks(css=css) as demo:
                         pass
                     with gr.Column():
                         generate_full_dataset_button = gr.Button("Generate Full Dataset", variant="primary")  # TODO: implement
+                        dataset_share_button = gr.Button("Share Dataset URL")
+                        dataset_share_textbox = gr.Textbox(visible=False, show_copy_button=True, label="Copy this URL:", interactive=False, show_label=True)
                         back_button = gr.Button("< Back", size="sm")
                     with gr.Column(scale=4, min_width=0):
                         pass
             with gr.Column(scale=4, min_width=0):
                 pass
-
+    
+    app_state = gr.State({})
 
     T = TypeVar("T")
 
@@ -218,6 +221,7 @@ with gr.Blocks(css=css) as demo:
                     max_tokens=max_tokens,
                     stream=True,
                     top_p=0.8,
+                    seed=42,
                 ):
                     yield message.choices[0].delta.content
             except requests.exceptions.ConnectionError as e:
@@ -228,6 +232,7 @@ with gr.Blocks(css=css) as demo:
 
 
     def gen_datasets_line_by_line(search_query: str, generated_texts: tuple[str] = ()) -> Iterator[str]:
+        search_query = search_query or ""
         search_query = search_query[:1000] if search_query.strip() else landing_page_query
         generated_text = ""
         current_line = ""
@@ -246,6 +251,7 @@ with gr.Blocks(css=css) as demo:
 
 
     def gen_dataset_content(search_query: str, dataset_name: str, tags: str) -> Iterator[str]:
+        search_query = search_query or ""
         search_query = search_query[:1000] if search_query.strip() else landing_page_query
         generated_text = ""
         for token in stream_reponse(GENERATE_DATASET_CONTENT_FOR_SEARCH_QUERY_AND_NAME_AND_TAGS.format(
@@ -257,11 +263,9 @@ with gr.Blocks(css=css) as demo:
             yield generated_text
         print("-----\n\n" + generated_text)
 
-    search_datasets_inputs = search_bar
-    search_datasets_outputs = button_groups + buttons + [generated_texts_state]
 
-    def search_datasets(search_query):
-        yield {generated_texts_state: []}
+    def _search_datasets(search_query):
+        yield {generated_texts_state: [], app_state: {"search_query": search_query}}
         yield {
             button_group: gr.Group(elem_classes="buttonsGroup insivibleButtonGroup")
             for button_group in button_groups[MAX_NB_ITEMS_PER_GENERATION_CALL:]
@@ -295,9 +299,23 @@ with gr.Blocks(css=css) as demo:
                 }
                 current_item_idx += 1
 
-    search_more_datasets_inputs = [search_bar, generated_texts_state]
-    search_more_datasets_outputs = button_groups + buttons + [generated_texts_state]
-
+    search_js = """
+    (search_query => {
+        const urlParams = new URLSearchParams(window.location.search);
+        urlParams.set('q', search_query);
+        window.location.search = urlParams;
+    })
+    """
+
+    @search_button.click(inputs=search_bar, outputs=button_groups + buttons + [generated_texts_state, app_state], js=search_js)
+    def search_dataset_from_search_button(search_query):
+        yield from _search_datasets(search_query)
+    
+    @search_bar.submit(inputs=search_bar, outputs=button_groups + buttons + [generated_texts_state, app_state], js=search_js)
+    def search_dataset_from_search_bar(search_query):
+        yield from _search_datasets(search_query)
+
+    @load_more_datasets.click(inputs=[search_bar, generated_texts_state], outputs=button_groups + buttons + [generated_texts_state])
     def search_more_datasets(search_query, generated_texts):
         current_item_idx = initial_item_idx = len(generated_texts) * MAX_NB_ITEMS_PER_GENERATION_CALL
         yield {
@@ -324,36 +342,63 @@ with gr.Blocks(css=css) as demo:
                 }
                 current_item_idx += 1
 
-    show_dataset_inputs = [search_bar, *buttons]
-    show_dataset_outputs = [search_page, dataset_page, dataset_title, dataset_content]
-
-    def show_dataset(search_query, *buttons_values, i):
-        dataset_name, tags = buttons_values[2 * i : 2 * i + 2]
+    def _show_dataset(search_query, dataset_name, tags):
         yield {
             search_page: gr.Column(visible=False),
             dataset_page: gr.Column(visible=True),
-            dataset_title: f"# {dataset_name}\n\n tags: {tags}\n\n _Note: This is an AI-generated dataset so its content may be inaccurate or false_"
+            dataset_title: f"# {dataset_name}\n\n tags: {tags}\n\n _Note: This is an AI-generated dataset so its content may be inaccurate or false_",
+            dataset_share_textbox: gr.Textbox(visible=False),
+            app_state: {
+                "search_query": search_query,
+                "dataset_name": dataset_name,
+                "tags": tags
+            }
         }
         for generated_text in gen_dataset_content(search_query=search_query, dataset_name=dataset_name, tags=tags):
             yield {dataset_content: generated_text}
 
 
-    def show_search_page():
-        return gr.Column(visible=True), gr.Column(visible=False)
+    show_dataset_inputs = [search_bar, *buttons]
+    show_dataset_outputs = [app_state, search_page, dataset_page, dataset_title, dataset_content, dataset_share_textbox]
 
+    def show_dataset_from_button(search_query, *buttons_values, i):
+        dataset_name, tags = buttons_values[2 * i : 2 * i + 2]
+        yield from _show_dataset(search_query, dataset_name, tags)
+    
+    for i, (dataset_name_button, tags_button) in enumerate(batched(buttons, 2)):
+        dataset_name_button.click(partial(show_dataset_from_button, i=i), inputs=show_dataset_inputs, outputs=show_dataset_outputs)
+        tags_button.click(partial(show_dataset_from_button, i=i), inputs=show_dataset_inputs, outputs=show_dataset_outputs)
 
-    def generate_full_dataset():
-        raise gr.Error("Not implemented yet sorry ! Give me some feedbacks in the Community tab in the meantime ;)")
 
+    @back_button.click(outputs=[search_page, dataset_page])
+    def show_search_page():
+        return gr.Column(visible=True), gr.Column(visible=False)
 
-    search_bar.submit(search_datasets, inputs=search_datasets_inputs, outputs=search_datasets_outputs)
-    search_button.click(search_datasets, inputs=search_datasets_inputs, outputs=search_datasets_outputs)
-    for i, (dataset_name_button, tags_button) in enumerate(batched(buttons, 2)):
-        dataset_name_button.click(partial(show_dataset, i=i), inputs=show_dataset_inputs, outputs=show_dataset_outputs)
-        tags_button.click(partial(show_dataset, i=i), inputs=show_dataset_inputs, outputs=show_dataset_outputs)
-    see_more.click(search_more_datasets, inputs=search_more_datasets_inputs, outputs=search_more_datasets_outputs)
+    @generate_full_dataset_button.click()
+    def generate_full_dataset():
+        raise gr.Error("Not implemented yet sorry ! Request your dataset in the Discussion tab (provide the dataset URL)")
+
+    @dataset_share_button.click(inputs=[app_state], outputs=[dataset_share_textbox])
+    def show_dataset_url(state):
+        return gr.Textbox(
+            f"{URL}?q={state['search_query'].replace(' ', '+')}&dataset={state['dataset_name']}&tags={state['tags']}",
+            visible=True,
+        )
+
+    @demo.load(outputs=[app_state, search_page, search_bar, dataset_page, dataset_title, dataset_content, dataset_share_textbox] + button_groups + buttons + [generated_texts_state])
+    def load_app(request: gr.Request):
+        query_params = dict(request.query_params)
+        if "dataset" in query_params:
+            yield from _show_dataset(
+                search_query=query_params.get("q", query_params["dataset"]),
+                dataset_name=query_params["dataset"],
+                tags=query_params.get("tags", "")
+            )
+        elif "q" in query_params:
+            yield {search_bar: query_params["q"]}
+            yield from _search_datasets(query_params["q"])
+        else:
+            yield {search_page: gr.Column(visible=True)}
 
-    generate_full_dataset_button.click(generate_full_dataset)
-    back_button.click(show_search_page, inputs=[], outputs=[search_page, dataset_page])
 
 demo.launch()