Update app.py

app.py

@@ -20,54 +20,49 @@ def greet(model_name, prompt_template, name, dataset_file):
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
 ##################################### App UI #######################################
-with gr.Blocks() as demo:
 
-    ##### Title Block #####
-    gr.Markdown("# Instruction Tuning with Unsloth")
+# Function to build the interface based on user choice
+def build_interface(choice):
+    if choice == "Predefined Dataset":
+        dataset_input = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', key="dataset_predefined")
+    elif choice == "Upload Your Own":
+        dataset_input = gr.File(label="Upload Dataset", accept=".csv,.txt", key="dataset_upload")
+    else:
+        dataset_input = None
+    return dataset_input
 
-    ##### Model Inputs #####
+# Function to handle changes in dataset choice
+def on_choice_change(choice):
+    interface = build_interface(choice)
+    update_interface(interface)
 
-    # Select Model
-    model_name = gr.Dropdown(label="Model", choices=conf['model']['choices'], value="gpt2")
-    # Prompt template
-    prompt_template = gr.Textbox(label="Prompt Template", value="Instruction: {0}\nOutput: {1}")
-    # Prompt Input
-    name_input = gr.Textbox(label="Your Name")
-    # Dataset choice
-    dataset_choice = gr.Radio(label="Choose Dataset", choices=["Predefined Dataset", "Upload Your Own"], value="Predefined Dataset")
-    dataset_predefined = gr.Dropdown(label="Predefined Dataset", choices=['1', '2', '3'], value='1', visible=True)
-    dataset_file = gr.inputs.File(label="Upload Dataset", type='file', accept=".csv,.txt", visible=False)
-    
-    # Function to toggle visibility based on user choice
-    def toggle_dataset_visibility(choice):
-        if choice == "Predefined Dataset":
-            dataset_predefined.visible = True
-            dataset_file.visible = False
-        elif choice == "Upload Your Own":
-            dataset_predefined.visible = False
-            dataset_file.visible = True
-    
-    # Initial visibility setup based on default choice
-    toggle_dataset_visibility(dataset_choice.value)
-    
-    ##### Model Outputs #####
+# Function to update the interface with new dataset input
+def update_interface(dataset_input):
+    demo.Interface(
+        fn=greet,
+        inputs=[model_name, prompt_template, name_input, dataset_input],
+        outputs=output
+    ).launch()
 
-    # Text output
-    output = gr.Textbox(label="Output")
-    
-    ##### Execution #####
+# Select Model
+model_name = gr.Dropdown(label="Model", choices=conf['model']['choices'], value="gpt2")
+# Prompt template
+prompt_template = gr.Textbox(label="Prompt Template", value="Instruction: {0}\nOutput: {1}")
+# Prompt Input
+name_input = gr.Textbox(label="Your Name")
+# Dataset choice
+dataset_choice = gr.Radio(label="Choose Dataset", choices=["Predefined Dataset", "Upload Your Own"], value="Predefined Dataset")
 
-    # Setup button
-    tune_btn = gr.Button("Start Fine Tuning")
-    # Execute button
-    tune_btn.click(fn=greet, 
-                   inputs=[model_name, prompt_template, name_input, dataset_file],
-                   outputs=output)
+# Initial interface setup
+initial_choice = dataset_choice.value
+initial_interface = build_interface(initial_choice)
 
-    # Update visibility based on user choice
-    dataset_choice.change(toggle_dataset_visibility, inputs=[dataset_choice], outputs=[])
+# Output textbox
+output = gr.Textbox(label="Output")
 
-##################################### Launch #######################################
+# Setup button
+tune_btn = gr.Button("Start Fine Tuning")
+tune_btn.click(on_choice_change, inputs=[dataset_choice])
 
-if __name__ == "__main__":
-    demo.launch()
+# Launch the initial interface
+update_interface(initial_interface)