from transformers import pipeline
import gradio as gr
def isEnglish(s):
    try:
        s.encode(encoding='utf-8').decode('ascii')
    except UnicodeDecodeError:
        return False
    else:
        return True


pipe = pipeline("text2text-generation", model="Varshitha/flan-t5-large-finetune-medicine-v5")
examples = [
    ["what is fever?"],
    ["what medicen to give child if got fever?"],
    ['Where is the protein Pannexin1 located?'],
]

title = "Qarisoft Medical Bot"
txt_box = gr.Textbox()
def ff_(message, history):
    if not isEnglish(message):
        return "Only support English right now"
    return pipe(message)[0]['generated_text']

gr.ChatInterface(ff_,
                title=title,
                description=title,
                examples=examples,
                submit_btn="Ask- أرسل",
                undo_btn="تراجع",
                retry_btn="اعادة",
                clear_btn="حذف",
                theme="soft",
                css="""
                    footer.svelte-1ax1toq {
                        display: none !important;
                    }
                """,
                ).launch()