Update app.py

added config builder

app.py

@@ -110,6 +110,72 @@ examples = [[str(f)] for f in pathlib.Path("examples").glob("*.yaml")]
 # when user do not provide a token.
 COMMUNITY_HF_TOKEN = os.getenv("COMMUNITY_HF_TOKEN")
 
+# config builder 
+import yaml
+
+def generate_config(base_model, models, layer_range, merge_method):
+    slices = []
+    for model in models:
+        slice_config = {
+            "sources": [
+                {
+                    "model": model,
+                    "layer_range": layer_range
+                }
+            ]
+        }
+        slices.append(slice_config)
+
+    config = {
+        "slices": slices,
+        "merge_method": merge_method,
+        "base_model": base_model,
+        "parameters": {
+            "t": [
+                {
+                    "filter": "self_attn",
+                    "value": [0, 0.5, 0.3, 0.7, 1]
+                },
+                {
+                    "filter": "mlp",
+                    "value": [1, 0.5, 0.7, 0.3, 0]
+                },
+                {
+                    "value": 0.5
+                }
+            ]
+        },
+        "dtype": "bfloat16"
+    }
+
+    return yaml.dump(config)
+
+
+# Add these imports
+from functools import partial
+from itertools import chain
+
+# Configure dropdown options
+BASE_MODELS = ["bert-base-uncased", "distilbert-base-uncased", ...] # Add other base models here
+MERGE_METHODS = ["linear", "slerp", ...] # Add other merge methods here
+LAYER_RANGE = range(32)
+
+# Create input objects
+input_base_model = gr.Dropdown(label='Base Model', choices=BASE_MODELS)
+input_models = gr.Multiselect(label='Models', choices=BASE_MODELS)
+input_layer_range = gr.NumberSlider(minimum=0, maximum=32, step=1, label='Layer Range')
+input_merge_method = gr.Dropdown(label='Merge Method', choices=MERGE_METHODS)
+
+# Wrap generate_config in a partial function to fix the signature
+partial_generate_config = partial(generate_config, base_model=input_base_model, merge_method=input_merge_method)
+
+# Generate config block
+gen_config_block = gr.Block()
+with gen_config_block:
+    generated_config = gr.outputs.Textbox(label='Generated Config', interactive=False)
+    btn_generate_config = gr.Button('Generate Config', variant='secondary')
+    btn_generate_config.click(fn=partial_generate_config, inputs=[input_base_model, input_models, input_layer_range], outputs=[generated_config])
+
 
 def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]]:
     runner = LogsViewRunner()
@@ -190,6 +256,7 @@ with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
 
     with gr.Row():
+        gen_config_block
         filename = gr.Textbox(visible=False, label="filename")
         config = gr.Code(language="yaml", lines=10, label="config.yaml")
         with gr.Column():