Update app.py

app.py

@@ -39,7 +39,7 @@ class update_visibility:
         else:
             return gr.UploadButton(visible=bool(0))
 
-class get_datasets:
+class get_files:
 
     def predefined_dataset(dataset_name):
         global dataset # bad practice, I know... But just bear with me. Will later update to state dict.
@@ -59,21 +59,14 @@ class get_datasets:
         except FileNotFoundError:
             return "File not found. Please upload the file again."
 
-
-
-def about_page():
-    return "## About\n\nThis is an application for uploading datasets. You can upload files in .csv, .jsonl, or .txt format. The app will process the file and provide feedback."
-
-
-
-
-
-def show_about():
-    return "## About\n\nThis is an application for uploading datasets. You can upload files in .csv, .jsonl, or .txt format. The app will process the file and provide feedback."
-
-def show_upload():
-    return "## Upload\n\nUse the button below to upload your dataset."
-
+    def load_markdown_file(file_path):
+        try:
+            with open(file_path, 'r') as f:
+                return f.read()
+        except FileNotFoundError:
+            return "File not found. Please check the file path."
+        except Exception as e:
+            return f"Error loading file: {str(e)}"
 
 
 def train(model_name, 
@@ -118,7 +111,8 @@ def main():
 
         with gr.Tabs():
             with gr.TabItem("About"):
-                gr.Markdown("## About\n\nThis is an application for uploading datasets. You can upload files in .csv, .jsonl, or .txt format. The app will process the file and provide feedback.")
+                gr.Markdown(get_files.load_markdown_file("README.md"))
+                
             with gr.TabItem("Train Model"):
                 ##### Title Block #####
                 gr.Markdown("# SLM Instruction Tuning with Unsloth")
@@ -161,11 +155,11 @@ def main():
                                       dataset_predefined_load)
                 
                 # Dataset button
-                dataset_predefined_load.click(fn=get_datasets.predefined_dataset,
+                dataset_predefined_load.click(fn=get_files.predefined_dataset,
                                           inputs=dataset_predefined,
                                           outputs=data_snippet)
         
-                dataset_uploaded_load.click(fn=get_datasets.uploaded_dataset,
+                dataset_uploaded_load.click(fn=get_files.uploaded_dataset,
                                          inputs=dataset_uploaded_load,
                                          outputs=data_snippet)