Update app.py

app.py

@@ -1,21 +1,30 @@
 ##################################### Imports ######################################
 # Generic imports
 import gradio as gr
-import json
-import os
 
-########################### Global objects and functions ###########################
+# Module imports
+from utilities.setup import get_json_cfg
 
-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
+########################### Global objects and functions ###########################
 
 conf = get_json_cfg()
 
+def dropdown_visibility(radio):
+    value = radio
+    if value == "Predefined Dataset":
+        return gr.Dropdown(visible=bool(1))
+    else:
+        return gr.Dropdown(visible=bool(0))
+
+def upload_visibility(radio):
+    value = radio
+    if value == "Upload Your Own":
+        return gr.UploadButton(visible=bool(1)) #make it visible
+    else:
+        return gr.UploadButton(visible=bool(0))
+
 def greet(model_name, prompt_template, name, dataset):
+    """The model call"""
     return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
 ##################################### App UI #######################################
@@ -36,46 +45,9 @@ with gr.Blocks() as demo:
     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.UploadButton(label="Upload Dataset", file_types=[".pdf",".csv",".jsonl"], visible=False) # gr.File(label="Upload Dataset", visible=False)
-
-    
-
-    
-
-
-    
-    def dropdown_visibility(radio):  # Accept the event argument, even if not used
-        value = radio  # Get the selected value from the radio button
-        if value == "Predefined Dataset":
-            return gr.Dropdown(visible=bool(1))
-        else:
-            return gr.Dropdown(visible=bool(0))
-
-    def upload_visibility(radio):
-        value = radio
-        if value == "Upload Your Own":
-            return gr.UploadButton(visible=bool(1)) #make it visible
-        else:
-            return gr.UploadButton(visible=bool(0))
-
     dataset_choice.change(dropdown_visibility, dataset_choice, dataset_predefined)
     dataset_choice.change(upload_visibility, dataset_choice, dataset_upload)
 
-    # Define function to update visibility based on dataset choice
-    def update_visibility():
-        choice = dataset_choice.value
-        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
-
-    # Bind visibility update function to change event of dataset_choice
-    dataset_choice.change(update_visibility)
-    
-    # Update visibility based on user choice
-    #dataset_predefined, dataset_upload = dataset_choice.change(update_dataset_visibility, inputs=[dataset_choice], outputs=[dataset_predefined, dataset_upload])
-    
     ##### Model Outputs #####
 
     # Text output