Update app.py

app.py

@@ -4,32 +4,20 @@ import gradio as gr
 import json
 import os
 
-# local imports
-
-
 ########################### Global objects and functions ###########################
 
 def get_json_cfg():
     """Retrieve configuration file"""
-
     config_path = os.getenv('CONFIG_PATH')
     with open(config_path, 'r') as file:
         config = json.load(file)
-
     return config
 
-
 conf = get_json_cfg()
 
-
-
-
-def greet(model_name, prompt_template, name):
+def greet(model_name, prompt_template, name, dataset):
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
-model_choices = conf['model']['choices']
-
-
 ##################################### App UI #######################################
 with gr.Blocks() as demo:
 
@@ -40,11 +28,22 @@ with gr.Blocks() as demo:
 
     # Select Model
     model_name = gr.Dropdown(label="Model", choices=conf['model']['choices'], value="gpt2")
-    # Show prompt template
+    # 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_upload = gr.File(label="Upload Dataset", visible=False)
+    
+    # Update visibility of dataset options based on user choice
+    def update_dataset_visibility(choice):
+        dataset_predefined.visible = (choice == "Predefined Dataset")
+        dataset_upload.visible = (choice == "Upload Your Own")
+    
+    dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])
+    
     ##### Model Outputs #####
 
     # Text output
@@ -56,11 +55,10 @@ with gr.Blocks() as demo:
     tune_btn = gr.Button("Start Fine Tuning")
     # Execute button
     tune_btn.click(fn=greet, 
-                   inputs=[model_name, prompt_template, name_input], 
+                   inputs=[model_name, prompt_template, name_input, dataset_predefined],
                    outputs=output)
 
-
 ##################################### Launch #######################################
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()