app.py

import gradio as gr

# Indwara 70 n'ibimenyetso byazo
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"],
}

def susuzuma_indwara(ibimenyetso):
    ibimenyetso = [i.strip().lower() for i in ibimenyetso.split(',')]
    indwara_ishoboka = []

    for indwara, symptoms in indwara_dict.items():
        count = sum(symptom in ibimenyetso for symptom in symptoms)
        if count >= len(symptoms) / 2:
            indwara_ishoboka.append(f"{indwara} (ibimenyetso bihuye: {count} / {len(symptoms)})")

    if indwara_ishoboka:
        return "Indwara ishoboka:\n" + "\n".join(indwara_ishoboka)
    else:
        return "Ntibishoboka kumenya indwara ishingiye ku bimenyetso watanze. Nyamuneka saba muganga."

iface = gr.Interface(
    fn=susuzuma_indwara,
    inputs=gr.Textbox(
        lines=3,
        placeholder="Andika ibimenyetso byawe (urugero: umuriro, umutwe, gukorora)",
        submit=True  # Enter ikora submit, nta button igaragara
    ),
    outputs="text",
    title="VIATEUR AI Kwamuganga",
    description="Andika ibimenyetso ufite, AI iguhe indwara ishoboka. Ibi ntibisimbura inama y'umuganga."
)

iface.launch()