Update app.py

app.py

@@ -15,36 +15,10 @@ def get_json_cfg():
 
 conf = get_json_cfg()
 
-def greet(model_name, prompt_template, name, dataset_file):
-    # Here you can process the uploaded file (dataset_file) as needed
+def greet(model_name, prompt_template, name, dataset):
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
 ##################################### App UI #######################################
-
-# 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
-
-# Function to handle changes in dataset choice
-def on_choice_change(choice):
-    interface = build_interface(choice)
-    update_interface(interface)
-
-# 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()
-
-##################################### Gradio Blocks #######################################
 with gr.Blocks() as demo:
 
     ##### Title Block #####
@@ -60,23 +34,37 @@ with gr.Blocks() as demo:
     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_upload = gr.File(label="Upload Dataset", visible=False)
+    
+    # Function to update visibility based on user choice
+    def update_dataset_visibility(choice):
+        if choice == "Predefined Dataset":
+            dataset_predefined.visible = True
+            dataset_upload.visible = False
+        elif choice == "Upload Your Own":
+            dataset_predefined.visible = False
+            dataset_upload.visible = True
+    
+    # Initial call to set visibility based on default choice
+    update_dataset_visibility(dataset_choice.value)
+    
+    # Update visibility based on user choice
+    dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])
+    
+    ##### Model Outputs #####
 
-    # Initial interface setup
-    initial_choice = dataset_choice.value
-    initial_interface = build_interface(initial_choice)
-
-    # Output textbox
+    # Text output
     output = gr.Textbox(label="Output")
+    
+    ##### Execution #####
 
     # Setup button
     tune_btn = gr.Button("Start Fine Tuning")
-    tune_btn.click(on_choice_change, inputs=[dataset_choice])
-
-    # Launch the initial interface
-    update_interface(initial_interface)
-
-    # Update visibility based on user choice
-    dataset_choice.change(on_choice_change, inputs=[dataset_choice])
+    # Execute button
+    tune_btn.click(fn=greet, 
+                   inputs=[model_name, prompt_template, name_input, dataset_predefined],
+                   outputs=output)
 
 ##################################### Launch #######################################