Updated drug_dict

app.py

@@ -32,7 +32,6 @@ def get_symp(symptoms,investigations,drugs,advices):
     print("Drugs :",drugs)
     print("Data dict : ",data_dict)
     for cat,med in data_dict.items():
-
         for i in med:
             drug_dict.append(i)
     return (gr.update(choices=eval(symptoms),visible=True,interactive=True)),(gr.update(choices=eval(investigations),visible=True,interactive=True)),(gr.update(choices=all,visible=True,interactive=True)),(gr.update(choices=eval(advices),visible=True,interactive=True))
@@ -58,12 +57,21 @@ filename = "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): 
+def save_data(disease,symp_data,invest_data,drug_data,advice_data,drug_dict): 
     user = ""
     for username, logged_in in logged_in_users.items():
             if logged_in:
                 user = username
                 print("In save:",user)
+
+    drug_dict = eval(drug_dict)
+    drug_keys = drug_dict.keys()
+    user_drug_dict = {}
+    for i in drug_data:
+        for j in drug_keys:
+            if i in drug_dict[j]:
+                user_drug_dict.setdefault(j, []).append(i)
+    # ---------------------------------------------
     # Read the existing content of the CSV file
     # file_path = repo.filename_to_path(filename)
     existing_rows = []
@@ -73,12 +81,13 @@ def save_data(disease,symp_data,invest_data,drug_data,advice_data):
         for row in csv_reader:
             existing_rows.append(row)
     print(existing_rows)
+    # ----------------------------------------------
     try:
         print("In try")
         with open("./demo1.csv", mode="a") as csvfile:
             print("In open csv file",csvfile)
             csv_writer = csv.writer(csvfile)
-            csv_writer.writerow([disease,symp_data,invest_data,drug_data,advice_data,user])
+            csv_writer.writerow([disease,symp_data,invest_data,user_drug_dict,advice_data,user])
             print("Done writing!!")        
     except:
         print("An Exception occured")
@@ -104,6 +113,6 @@ with gr.Blocks() as demo:
         save = gr.Button(value="Save")
     
     fetch.click(get_symp, inputs=[symptoms,investigations,drugs,advices],outputs=[symptoms_box,investigations_box,drugs_box,advices_box])
-    save.click(save_data,inputs=[disease,symptoms_box,investigations_box,drugs_box,advices_box],outputs=[]).then(fn=None,_js=js)
+    save.click(save_data,inputs=[disease,symptoms_box,investigations_box,drugs_box,advices_box,drugs],outputs=[]).then(fn=None,_js=js)
 if __name__ == "__main__":
     demo.launch(auth=login)