Update app.py

app.py

@@ -646,6 +646,20 @@ def get_popular_datasets():
         "ag_news"
     ]
 
+def load_dataset_config_for_ui():
+    config = load_dataset_config()
+    return (
+        config["vodalus_system_message"],
+        config["prompt_1"],
+        config["topics"],
+        config["max_tokens"],
+        config["temperature"],
+        config["top_p"],
+        config["frequency_penalty"],
+        config["presence_penalty"]
+    )
+
+
 css = """
 body, #root {
     margin: 0;
@@ -862,7 +876,7 @@ with demo:
             with gr.Tab("Dataset Configuration"):
                 with gr.Row():
                     vodalus_system_message = gr.TextArea(label="System Message for JSONL Dataset", lines=10)
-                    prompt_1 = gr.TextArea(label="Dataset Gerenation Prompt", lines=10)
+                    prompt_1 = gr.TextArea(label="Dataset Generation Prompt", lines=10)
                 
                 gr.Markdown("### Topics")
                 with gr.Row():
@@ -939,8 +953,9 @@ with demo:
                         frequency_penalty = gr.Slider(minimum=0, maximum=2, value=0.0, step=0.01, label="Frequency Penalty")
                         presence_penalty = gr.Slider(minimum=0, maximum=2, value=0.0, step=0.01, label="Presence Penalty")
                 
-                save_llm_config_btn = gr.Button("Save LLM Configuration")
-                llm_config_status = gr.Textbox(label="Status")
+                with gr.Row():
+                    save_dataset_config_btn = gr.Button("Save Dataset Configuration", variant="primary")
+                    dataset_config_status = gr.Textbox(label="Status")
 
     add_topic.click(
         lambda x: x + [["New Topic"]],
@@ -1115,17 +1130,7 @@ with demo:
     )
 
     demo.load(
-        lambda: (
-            config := load_dataset_config(),
-            gr.update(value=config["vodalus_system_message"]),
-            gr.update(value=config["prompt_1"]),
-            gr.update(value=config["topics"]),
-            gr.update(value=config["max_tokens"]),
-            gr.update(value=config["temperature"]),
-            gr.update(value=config["top_p"]),
-            gr.update(value=config["frequency_penalty"]),
-            gr.update(value=config["presence_penalty"])
-        )[1:],
+        load_dataset_config_for_ui,
         outputs=[
             vodalus_system_message,
             prompt_1,