import gradio as gr indwara_dict = { "Cancer": { "symptoms": ["umuriro", "kubabara umutwe", "gucika intege"], "treatment": "Kubagwa, chemotherapy, radiotherapy" }, "Diabetes": { "symptoms": ["inyota nyinshi", "gucika intege", "kugabanuka k'ibiro"], "treatment": "Insulin, imiti yo kugenzura isukari mu maraso" }, "Flu": { "symptoms": ["umuriro", "umutwe", "gukorora"], "treatment": "Kuruhuka, imiti yo kugabanya umuriro nka paracetamol" }, "COVID-19": { "symptoms": ["umuriro", "guhumeka nabi", "kuruha", "kubura impumuro"], "treatment": "Kuruhuka, gufata amazi menshi, imiti y’ibanze, kubonana na muganga" }, "Malaria": { "symptoms": ["umuriro", "gucika intege", "kubira ibyuya"], "treatment": "Artemisinin-based combination therapy (ACT)" }, "Tuberculosis": { "symptoms": ["inkorora ikomeye", "umuriro", "gucika intege"], "treatment": "Isoniazid, rifampicin, pyrazinamide, ethambutol" }, "Asthma": { "symptoms": ["guhumeka nabi", "kubura umwuka", "gukorora"], "treatment": "Inhalers (bronchodilators, corticosteroids)" }, "Hypertension": { "symptoms": ["umutwe", "gukorora umutwe", "kuribwa mu mutima"], "treatment": "Imiti igabanya umuvuduko w’amaraso nka ACE inhibitors, beta blockers" }, "Stroke": { "symptoms": ["kuribwa umutwe ukabije", "kubura ubushobozi bwo kugenda", "kudakora neza kw’uruhande rumwe rw’umubiri"], "treatment": "Muganga hakenewe kwihutirwa, imiti ishinzwe kugabanya amaraso, kuboneza umuvuduko" }, # ...shyiramo izindi ukoresheje ubu buryo, aha ndi kuzigabanya kubera umwanya } def susuzuma_indwara(ibimenyetso): ibimenyetso = [i.strip().lower() for i in ibimenyetso.split(',')] indwara_ishoboka = [] for indwara, data in indwara_dict.items(): symptoms = data["symptoms"] treatment = data["treatment"] count = sum(symptom in ibimenyetso for symptom in symptoms) if count >= len(symptoms) / 2: indwara_ishoboka.append(f"{indwara} (ibimenyetso bihuye: {count} / {len(symptoms)})\nImiti: {treatment}") if indwara_ishoboka: return "Indwara ishoboka hamwe n'imiti:\n\n" + "\n\n".join(indwara_ishoboka) else: return "Ntibishoboka kumenya indwara ishingiye ku bimenyetso watanze. Nyamuneka saba muganga." with gr.Blocks(title="VIATEUR AI Kwamuganga") as iface: textbox = gr.Textbox( lines=3, placeholder="Andika ibimenyetso byawe (urugero: umuriro, umutwe, gukorora)", label="Ibimenyetso" ) output = gr.Textbox(label="Indwara Ishoboka n'Imiti", lines=10) submit_btn = gr.Button("Submit") submit_btn.click(fn=susuzuma_indwara, inputs=textbox, outputs=output) iface.launch()

app.py

@@ -1,102 +1,69 @@
-
 import gradio as gr
 
 indwara_dict = {
-    "Cancer": ["umuriro", "kubabara umutwe", "gucika intege"],
-    "Diabetes": ["inyota nyinshi", "gucika intege", "kugabanuka k'ibiro"],
-    "Flu": ["umuriro", "umutwe", "gukorora"],
-    "COVID-19": ["umuriro", "guhumeka nabi", "kuruha", "kubura impumuro"],
-    "Malaria": ["umuriro", "gucika intege", "kubira ibyuya"],
-    "Tuberculosis": ["inkorora ikomeye", "umuriro", "gucika intege"],
-    "Asthma": ["guhumeka nabi", "kubura umwuka", "gukorora"],
-    "Hypertension": ["umutwe", "gukorora umutwe", "kuribwa mu mutima"],
-    "Stroke": ["kuribwa umutwe ukabije", "kubura ubushobozi bwo kugenda", "kudakora neza kw’uruhande rumwe rw’umubiri"],
-    "Heart Attack": ["kubabara mu mutima", "guhumeka nabi", "kuribwa mu gituza"],
-    "Pneumonia": ["gukorora", "guhumeka nabi", "umuriro"],
-    "Diarrhea": ["kubyimba mu nda", "kuruka", "kubura amazi mu mubiri"],
-    "Appendicitis": ["kubabara mu nda", "kuribwa cyane ku ruhande rw’iburyo rw’umubiri"],
-    "Migraine": ["umutwe ukabije", "kubabara mu maso", "gukorora"],
-    "Depression": ["agahinda gakabije", "kubura ubushake", "kuribwa umutima"],
-    "Anxiety": ["umutima kwihuta", "guhumeka nabi", "agahinda"],
-    "Arthritis": ["kubabara imitsi", "kugubwa n’umubyimba", "guhinduka kw’imitsi"],
-    "Alzheimer’s disease": ["guta ubwenge", "kwibagirwa", "kudafata neza ibyo uvuga"],
-    "Parkinson’s disease": ["guhindagurika kw’imikaya", "guhagarara nabi", "kubura kugenzura umubiri"],
-    "Epilepsy": ["gukorora", "guturika umutwe", "kudakora neza kw’imikaya"],
-    "Hepatitis": ["uruhu guhindura ibara", "umuriro", "kubabara mu nda"],
-    "Chickenpox": ["ibisebe ku mubiri", "umuriro", "gukorora"],
-    "Measles": ["ibisebe ku mubiri", "umuriro", "uburwayi bwo mu maso"],
-    "Mumps": ["kubabara amabya", "umuriro", "gukorora"],
-    "Rubella": ["umuriro", "ibisebe ku mubiri", "gukorora"],
-    "Cholera": ["kubyimba mu nda", "kuruka", "kuribwa mu nda"],
-    "Typhoid": ["umuriro ukabije", "kuribwa mu nda", "inkorora"],
-    "Dengue": ["umuriro ukabije", "kubabara imitsi", "kubabara umutwe"],
-    "Ebola": ["kuribwa mu nda", "umuriro", "guhagarara k'amaraso"],
-    "Rabies": ["gukorora", "guhumeka nabi", "kubura ubushobozi bwo kugenda"],
-    "Smallpox": ["ibisebe ku mubiri", "umuriro", "gukorora"],
-    "HIV/AIDS": ["gucika intege", "kubura ibiro", "inkorora idakira"],
-    "Gonorrhea": ["kuribwa igihe cyo kwihagarika", "kubabara mu nda", "inkorora"],
-    "Syphilis": ["ibisebe ku mubiri", "umuriro", "gukorora"],
-    "Chlamydia": ["inkorora", "kubabara mu nda", "umuriro"],
-    "Herpes": ["ibisebe ku mubiri", "gukorora", "umuriro"],
-    "Influenza": ["umuriro", "gukorora", "umutwe"],
-    "Polio": ["kubura ubushobozi bwo kugenda", "gucika intege", "guhagarara kw’imikaya"],
-    "Yellow Fever": ["umuriro", "kuribwa mu nda", "gukorora"],
-    "Bronchitis": ["inkorora", "guhumeka nabi", "umuriro"],
-    "Emphysema": ["guhumeka nabi", "gukorora", "gucika intege"],
-    "Colitis": ["kuribwa mu nda", "kubyimba mu nda", "inkorora"],
-    "Crohn’s Disease": ["kuribwa mu nda", "kubyimba mu nda", "kuribwa kw’umubiri"],
-    "Ulcerative Colitis": ["kubyimba mu nda", "inkorora", "kuribwa mu nda"],
-    "Irritable Bowel Syndrome": ["kuribwa mu nda", "guhinduka kw’imikorere y’inkorora", "kubyimba mu nda"],
-    "Gastroenteritis": ["kuruka", "kubyimba mu nda", "inkorora"],
-    "Diverticulitis": ["kuribwa mu nda", "kubyimba mu nda", "gukorora"],
-    "Hernia": ["kubabara mu nda", "kubyimba mu nda", "kuribwa igihe uhagaze"],
-    "Cirrhosis": ["kuribwa mu nda", "kubura ubushake bwo kurya", "kuribwa mu gifu"],
-    "Hepatitis B": ["umuriro", "kuribwa mu nda", "ibibazo by’uruhu"],
-    "Hepatitis C": ["umuriro", "kuribwa mu nda", "gucika intege"],
-    "Leukemia": ["gucika intege", "kubura ibiro", "umuriro"],
-    "Lymphoma": ["gucika intege", "kubura ibiro", "kubabara umutwe"],
-    "Melanoma": ["ibisebe ku ruhu", "kubabara", "impinduka z’uruhu"],
-    "Prostate Cancer": ["kubabara mu nda", "inkorora", "guhagarara k'inkari"],
-    "Breast Cancer": ["kubabara mu mabere", "ibisebe ku mabere", "guhinduka ku mabere"],
-    "Lung Cancer": ["gukorora", "guhumeka nabi", "kubabara mu gituza"],
-    "Cervical Cancer": ["kubabara mu nda", "inkari zifite amaraso", "umuriro"],
-    "Ovarian Cancer": ["kubabara mu nda", "kubura ubushake bwo kurya", "gucika intege"],
-    "Pancreatic Cancer": ["kubabara mu nda", "gucika intege", "kubura ibiro"],
-    "Kidney Cancer": ["kubabara mu nda", "inkari zifite amaraso", "gucika intege"],
-    "Bladder Cancer": ["inkari zifite amaraso", "kubabara igihe uri kwihagarika", "gucika intege"],
-    "Colon Cancer": ["kubabara mu nda", "inkorora", "kubura ibiro"],
-    "Rectal Cancer": ["kubabara mu nda", "inkorora", "amaraso mu nkorora"],
-    "Liver Cancer": ["kubabara mu nda", "kubura ibiro", "gucika intege"],
-    "Esophageal Cancer": ["kubabara igihe urya", "kubura ubushake bwo kurya", "gucika intege"],
-    "Stomach Cancer": ["kubabara mu nda", "kubura ubushake bwo kurya", "inkorora"],
-    "Endometrial Cancer": ["kubabara mu nda", "inkorora", "amaraso mu nkorora"],
-    "Testicular Cancer": ["kubabara ku gitsina", "kubura ubushake bwo gukora imibonano", "gucika intege"],
-    "Thyroid Cancer": ["kubabara mu ijosi", "kubura ubushake bwo kurya", "gucika intege"],
-    "Skin Cancer": ["ibisebe ku ruhu", "impinduka ku ruhu", "kubabara"],
+    "Cancer": {
+        "symptoms": ["umuriro", "kubabara umutwe", "gucika intege"],
+        "treatment": "Kubagwa, chemotherapy, radiotherapy"
+    },
+    "Diabetes": {
+        "symptoms": ["inyota nyinshi", "gucika intege", "kugabanuka k'ibiro"],
+        "treatment": "Insulin, imiti yo kugenzura isukari mu maraso"
+    },
+    "Flu": {
+        "symptoms": ["umuriro", "umutwe", "gukorora"],
+        "treatment": "Kuruhuka, imiti yo kugabanya umuriro nka paracetamol"
+    },
+    "COVID-19": {
+        "symptoms": ["umuriro", "guhumeka nabi", "kuruha", "kubura impumuro"],
+        "treatment": "Kuruhuka, gufata amazi menshi, imiti y’ibanze, kubonana na muganga"
+    },
+    "Malaria": {
+        "symptoms": ["umuriro", "gucika intege", "kubira ibyuya"],
+        "treatment": "Artemisinin-based combination therapy (ACT)"
+    },
+    "Tuberculosis": {
+        "symptoms": ["inkorora ikomeye", "umuriro", "gucika intege"],
+        "treatment": "Isoniazid, rifampicin, pyrazinamide, ethambutol"
+    },
+    "Asthma": {
+        "symptoms": ["guhumeka nabi", "kubura umwuka", "gukorora"],
+        "treatment": "Inhalers (bronchodilators, corticosteroids)"
+    },
+    "Hypertension": {
+        "symptoms": ["umutwe", "gukorora umutwe", "kuribwa mu mutima"],
+        "treatment": "Imiti igabanya umuvuduko w’amaraso nka ACE inhibitors, beta blockers"
+    },
+    "Stroke": {
+        "symptoms": ["kuribwa umutwe ukabije", "kubura ubushobozi bwo kugenda", "kudakora neza kw’uruhande rumwe rw’umubiri"],
+        "treatment": "Muganga hakenewe kwihutirwa, imiti ishinzwe kugabanya amaraso, kuboneza umuvuduko"
+    },
+    # ...shyiramo izindi ukoresheje ubu buryo, aha ndi kuzigabanya kubera umwanya
 }
 
 def susuzuma_indwara(ibimenyetso):
     ibimenyetso = [i.strip().lower() for i in ibimenyetso.split(',')]
     indwara_ishoboka = []
 
-    for indwara, symptoms in indwara_dict.items():
+    for indwara, data in indwara_dict.items():
+        symptoms = data["symptoms"]
+        treatment = data["treatment"]
         count = sum(symptom in ibimenyetso for symptom in symptoms)
         if count >= len(symptoms) / 2:
-            indwara_ishoboka.append(f"{indwara} (ibimenyetso bihuye: {count} / {len(symptoms)})")
+            indwara_ishoboka.append(f"{indwara} (ibimenyetso bihuye: {count} / {len(symptoms)})\nImiti: {treatment}")
 
     if indwara_ishoboka:
-        return "Indwara ishoboka:\n" + "\n".join(indwara_ishoboka)
+        return "Indwara ishoboka hamwe n'imiti:\n\n" + "\n\n".join(indwara_ishoboka)
     else:
         return "Ntibishoboka kumenya indwara ishingiye ku bimenyetso watanze. Nyamuneka saba muganga."
 
 with gr.Blocks(title="VIATEUR AI Kwamuganga") as iface:
     textbox = gr.Textbox(
         lines=3,
-        placeholder="Andika ibimenyetso byawe KURI VIATER AI (urugero: umuriro, umutwe, gukorora)",
+        placeholder="Andika ibimenyetso byawe (urugero: umuriro, umutwe, gukorora)",
         label="Ibimenyetso"
     )
-    output = gr.Textbox(label="Indwara Ishoboka")
-    submit_btn = gr.Button("Submit TO VIATEUR AI")
+    output = gr.Textbox(label="Indwara Ishoboka n'Imiti", lines=10)
+    submit_btn = gr.Button("Submit")
 
     submit_btn.click(fn=susuzuma_indwara, inputs=textbox, outputs=output)