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="""
                    a.built-with {
                        opacity: 0 !important;
                        visibility: hidden !important;
                    }
                    .message{
                        width: fit-content !important;
                    }
                    
                    .user.svelte-1henpdt.latest {
                        width: fit-content;
                        margin-left: auto;
                    }
                    header,
                    header.from-gray-50-to-white,
                    header.from-gray-50-to-white.relative.border-b.border-gray-100.bg-gradient-to-t.via-white.pt-0\.5.dark\:via-gray-950 {
                    display: none !important;
                    }
                
                """,
                ).launch(show_api=False,show_tips=False)