123

app.py

@@ -22,7 +22,7 @@ from transformers import (
 )
 from transformers.image_utils import load_image
 
-from data_experiments import filter_experiments
+from data_experiments import all_products, all_experiments, filter_experiments
 
 # Constants for text generation
 MAX_MAX_NEW_TOKENS = 2048
@@ -105,10 +105,15 @@ def process_image(
         yield buffer, buffer
 
 
-experiments_to_select = filter_experiments(
-    product_id="chemistry",
-)
-experiments_to_select = sorted(experiments_to_select, key=lambda x: x["id"])
+
+products_for_dropdown = sorted(all_products)
+
+def build_experiments_for_dropdown(product: str) -> list[dict]:
+    experiments = filter_experiments(
+        experiments=all_experiments,
+        product_id=product,
+    )
+    return sorted(experiments, key=lambda x: x["id"])
 
 def format_experiment_dropdown_title(experiment: dict) -> str:
     id = str(experiment["id"])
@@ -116,6 +121,13 @@ def format_experiment_dropdown_title(experiment: dict) -> str:
     product = experiment.get("product_id", "")
     return f"{id}. {title} ({product})"
 
+def update_experiments_dropdown(product: str):
+    experiments = build_experiments_for_dropdown(product)
+    return gr.update(
+        choices=[(format_experiment_dropdown_title(exp), exp["id"]) for exp in experiments],
+        value=experiments[0]["id"] if experiments else None,
+    )
+
 css = """
 .submit-btn {
     background-color: #2980b9 !important;
@@ -136,8 +148,13 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     gr.Markdown("# **Experiments photo moderation**")
     with gr.Row():
         with gr.Column():
-            gr.Dropdown(
-                choices=[(format_experiment_dropdown_title(exp), exp["id"]) for exp in experiments_to_select],
+            product_dropdown = gr.Dropdown(
+                choices=products_for_dropdown,
+                value="chemistry",
+                label="Select Product",
+            )
+            experiment_dropdown = gr.Dropdown(
+                choices=[],
                 value=None,
                 label="Select Experiment",
             )
@@ -164,11 +181,19 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 with gr.Accordion("(Result.md)", open=False):
                     markdown_output = gr.Markdown()
             
+    product_dropdown.change(
+        fn=update_experiments_dropdown,
+        inputs=product_dropdown,
+        outputs=experiment_dropdown
+    )
+
     image_submit.click(
         fn=process_image,
         inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
         outputs=[output, markdown_output]
     )
 
+update_experiments_dropdown("chemistry")  # Initial load for default product
+
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(share=True, mcp_server=True, ssr_mode=False, show_error=True)

data_experiments.py

@@ -12,6 +12,7 @@ for experiment in all_experiments:
     if product_id is not None:
         experiment_ids_by_product_id[product_id].append(experiment["id"])
 
+all_products = list(experiment_ids_by_product_id.keys())
 
 def get_experiment_by_id(experiment_id: int) -> Union[dict, None]:
     for experiment in all_experiments: