Initial Commit - app.py

app.py

@@ -0,0 +1,80 @@
+
+import gradio as gr
+from functools import partial
+
+import csv
+import pandas as pd
+import ast
+
+
+data = pd.read_csv("./KB_gpt_v1.csv",index_col=0)
+
+all_diseases = list(data['diseases'])
+
+js = "(x) => {return alert('File Saved!!')}"
+
+logged_in_users = {}
+def login(username, password):
+    if (username, password) in [("admin", "pass1234"), ("admin2", "pass1234")]:
+        logged_in_users[username] = True
+        print(logged_in_users)
+        return True
+    else:
+        return False
+
+def get_symp(symptoms,investigations,drugs,advices):
+    all = []
+    data_dict = eval(drugs)
+    for cat,med in data_dict.items():
+
+        for i in med:
+            all.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))
+
+
+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
+
+
+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
+    try:
+        with open("demo1.csv", mode="a", newline="") as csvfile:
+            csv_writer = csv.writer(csvfile)
+            csv_writer.writerow([disease,symp_data,invest_data,drug_data,advice_data,user])
+        
+    except:
+        print("An Exception occured")
+
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        disease = gr.Dropdown(all_diseases,label="Diseases")
+    with gr.Row():
+        fetch = gr.Button(value="Fetch")
+    symptoms=gr.Textbox(visible=False)
+    investigations = gr.Textbox(visible=False)
+    drugs = gr.Textbox(visible=False)
+    advices = gr.Textbox(visible=False)
+    disease.change(get_list,disease,[symptoms,investigations,drugs,advices]) 
+    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)
+        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)
+if __name__ == "__main__":
+    demo.launch(auth=login)