Update app.py

app.py

@@ -6,6 +6,9 @@ import csv
 import pandas as pd
 import ast
 from datasets import load_dataset
+from huggingface_hub import HfApi, HfFolder
+
+
 
 data = pd.read_csv("./KB_gpt_v1.csv",index_col=0)
 
@@ -39,14 +42,33 @@ def get_list(disease):
     drugs = ast.literal_eval((data[data['diseases']==str(disease)]["Drug type: Drugs"]).iloc[0])
     advices = ast.literal_eval((data[data['diseases']==str(disease)]["Advices"]).iloc[0])
     return sympto,investigations,drugs,advices
+#------------------------------------------------
+api = HfApi()
+
+# Replace with your Hugging Face Spaces username and repository name
+username = "Add-Vishnu"
+repo_name = "cdss_kb_generation"
+
+# Replace with the CSV file name
+filename = "demo1.csv"
 
-# target_csv_file = "./demo1.csv"
+# Load the repository
+repo = HfFolder(repo_name, api=api, username=username)
+#---------------------------------------------------------
 def save_data(disease,symp_data,invest_data,drug_data,advice_data): 
     user = ""
     for username, logged_in in logged_in_users.items():
             if logged_in:
                 user = username
                 print("In save:",user)
+    # Read the existing content of the CSV file
+    file_path = repo.filename_to_path(filename)
+    existing_rows = []
+    with open(file_path, "r") as csv_file:
+        print("Read the data")
+        csv_reader = csv.reader(csv_file)
+        for row in csv_reader:
+            existing_rows.append(row)
     try:
         print("In try")
         with open("./demo1.csv", mode="w", newline="") as csvfile: