Update app.py

app.py

@@ -16,45 +16,47 @@ from utilities.templates import prompt_template
 
 conf = get_json_cfg()
 
-def textbox_visibility(radio):
-    value = radio
-    if value == "Hugging Face Hub Dataset":
-        return gr.Dropdown(visible=bool(1))
-    else:
-        return gr.Dropdown(visible=bool(0))
-
-def textbox_button_visibility(radio):
-    value = radio
-    if value == "Hugging Face Hub Dataset":
-        return gr.Button(visible=bool(1))
-    else:
-        return gr.Button(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))
-
-
 
+class update_visibility:
 
+    def textbox_visibility(radio):
+        value = radio
+        if value == "Hugging Face Hub Dataset":
+            return gr.Dropdown(visible=bool(1))
+        else:
+            return gr.Dropdown(visible=bool(0))
+    
+    def textbox_button_visibility(radio):
+        value = radio
+        if value == "Hugging Face Hub Dataset":
+            return gr.Button(visible=bool(1))
+        else:
+            return gr.Button(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 get_predefined_dataset(dataset_name):
-    dataset = load_dataset(dataset_name, split = "train") #dataset_name
-    return dataset[0]['output'][0:100]
 
+class get_datasets:
 
-def get_uploaded_dataset(file):
-    dataset = []
-    print(file)
-    with open(file,'r') as file:
-        for line in file:
-            dataset.append(json.loads(line.strip()))
-    print(dataset[0]['query'])
-    return dataset[0]['query']
+    def predefined_dataset(dataset_name):
+        dataset = load_dataset(dataset_name, split = "train") #dataset_name
+        return dataset[0]['output'][0:100]
+    
+    
+    def uploaded_dataset(file):
+        dataset = []
+        print(file)
+        with open(file,'r') as file:
+            for line in file:
+                dataset.append(json.loads(line.strip()))
+        print(dataset[0]['query'])
+        return dataset[0]['query']
 
 
 
@@ -136,22 +138,22 @@ def main():
                                 value=conf['model']['general']["dataset_text_field"])
 
         data_snippet = gr.Markdown()
-        dataset_choice.change(textbox_visibility, 
+        dataset_choice.change(update_visibility.textbox_visibility, 
                               dataset_choice, 
                               dataset_predefined)
-        dataset_choice.change(upload_visibility, 
+        dataset_choice.change(update_visibility.upload_visibility, 
                               dataset_choice,
                               dataset_uploaded_load)
-        dataset_choice.change(textbox_button_visibility,
+        dataset_choice.change(update_visibility.textbox_button_visibility,
                               dataset_choice,
                               dataset_predefined_load)
         
         # Dataset button
-        dataset_predefined_load.click(fn=get_predefined_dataset,
+        dataset_predefined_load.click(fn=get_datasets.predefined_dataset,
                                   inputs=dataset_predefined,
                                   outputs=data_snippet)
 
-        dataset_uploaded_load.click(fn=get_uploaded_dataset,
+        dataset_uploaded_load.click(fn=get_datasets.get_uploaded_dataset,
                                  inputs=dataset_uploaded_load,
                                  outputs=data_snippet)