Added interface for contraindications, created a function to get original format dataset

app.py

@@ -25,23 +25,34 @@ def login(username, password):
     else:
         return False
 
-def get_symp(symptoms,investigations,drugs,advices):
-    drug_list = []
-    data_dict = {}
-    data_dict = eval(drugs)
-    for cat,med in data_dict.items():
+def get_symp(symptoms,investigations,drugs,advices,contraindications):
+    all_drugs = []
+    all_contras = []
+    drug_dict = eval(drugs)
+    contra_dict = eval(contraindications)
+    for cat,med in drug_dict.items():
         for i in med:
-            drug_list.append(i)
-    return (gr.update(choices=eval(symptoms),visible=True,interactive=True)),(gr.update(choices=eval(investigations),visible=True,interactive=True)),(gr.update(choices=drug_list,visible=True,interactive=True)),(gr.update(choices=eval(advices),visible=True,interactive=True))
+            all_drugs.append(i)
+            
+    for dru,con in contra_dict.items():
+        for i in con:
+            lol = ""
+            for ikey,ival in i.items():
+                lol+=ival+":"
+            all_contras.append(lol[:-1])
+    print(all_contras)
+#     print(all)
+    return (gr.update(choices=eval(symptoms),visible=True,interactive=True)),(gr.update(choices=eval(investigations),visible=True,interactive=True)),(gr.update(choices=all_drugs,visible=True,interactive=True)),(gr.update(choices=eval(advices),visible=True,interactive=True)),(gr.update(choices=all_contras,visible=True,interactive=True))
 
 
 def get_list(disease):
     print(disease)
-    sympto = ast.literal_eval((data[data['diseases']==str(disease)]).iloc[0][1])
-    investigations = ast.literal_eval((data[data['diseases']==str(disease)]['investigations']).iloc[0])
-    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
+    sympto = eval((data[data['diseases']==str(disease)]['symptoms']).iloc[0])
+    investigations = eval((data[data['diseases']==str(disease)]['investigations']).iloc[0])
+    drugs = eval((data[data['diseases']==str(disease)]["Drug type: Drugs"]).iloc[0])
+    advices = eval((data[data['diseases']==str(disease)]["Advices"]).iloc[0])
+    contras = eval((data[data['diseases']==str(disease)]["contraindications"]).iloc[0])
+    return sympto,investigations,drugs,advices,contras
 #------------------------------------------------
 # api = HfApi()
 
@@ -55,24 +66,51 @@ 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,drug_dict): 
+def get_original_format(drug_data,drug_dict,contra_data,contra_dict):
+    # print("Drug_dict: ",drug_dict)
+    # print("Drug_dict_type_before: ",type(drug_dict))
+    # print("Drug_dict_type_after: ",type(drug_dict))
+    drug_dict = eval(drug_dict)
+    drug_keys = drug_dict.keys()
+    # print("keys: ",drug_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)
+    
+    contra_list = []
+    for i in contra_data:
+        contra = i.split(':')
+        inner_dict = {}
+        inner_dict['Contraindication'] = contra[0]
+        inner_dict['Reason'] = contra[1]
+        print(inner_dict)
+        contra_list.append(inner_dict)
+    
+    user_contra_dict = {}
+    contra_dict = eval(contra_dict)
+    contra_keys = contra_dict.keys()
+    for i in contra_list:
+        for j in contra_keys:
+            if i in contra_dict[j]:
+                user_contra_dict.setdefault(j, []).append(i)
+        
+                
+    return user_drug_dict,user_contra_dict
+
+
+
+
+
+def save_data(disease,symp_data,invest_data,drug_data,advice_data,contra_data,drug_dict,contra_dict): 
     user = ""
     for username, logged_in in logged_in_users.items():
-        if logged_in:
-            user = username
-            print("In save:",user)
-
-    try:
-        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)
-    except Exception as e:
-        print("Yes it is here: ",e)
-        # break
+            if logged_in:
+                user = username
+    print(user)
+    user_drug_dict,user_contra_dict = get_original_format(drug_data,drug_dict,contra_data,contra_dict)
     # ---------------------------------------------
     # Read the existing content of the CSV file
     # file_path = repo.filename_to_path(filename)
@@ -89,7 +127,7 @@ def save_data(disease,symp_data,invest_data,drug_data,advice_data,drug_dict):
         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,user_drug_dict,advice_data,user])
+            csv_writer.writerow([disease,symp_data,invest_data,user_drug_dict,advice_data,user_contra_dict,user])
             print("Done writing!!")        
     except:
         print("An Exception occured")
@@ -105,16 +143,18 @@ with gr.Blocks() as demo:
     investigations = gr.Textbox(visible=False)
     drugs = gr.Textbox(visible=False)
     advices = gr.Textbox(visible=False)
-    disease.change(get_list,disease,[symptoms,investigations,drugs,advices]) 
+    contraindications = gr.Textbox(visible=False)
+    disease.change(get_list,disease,[symptoms,investigations,drugs,advices,contraindications]) 
     with gr.Box():
         symptoms_box = gr.CheckboxGroup(label="Symptoms",visible=False)
         investigations_box = gr.CheckboxGroup(label="Investigations",visible=False)
 #         drugs_box =  gr.Box(visible=False)
         drugs_box = gr.CheckboxGroup(label="Drugs",visible=False)
         advices_box = gr.CheckboxGroup(label="Advices",visible=False)
+        contra_box = gr.CheckboxGroup(label="Contraindications",visible=False)
         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,drugs],outputs=[]).then(fn=None,_js=js)
+    fetch.click(get_symp, inputs=[symptoms,investigations,drugs,advices,contraindications],outputs=[symptoms_box,investigations_box,drugs_box,advices_box,contra_box])
+    save.click(save_data,inputs=[disease,symptoms_box,investigations_box,drugs_box,advices_box,contra_box,drugs,contraindications],outputs=[]).then(fn=None,_js=js)
 if __name__ == "__main__":
     demo.launch(auth=login)