import gradio as gr
from utils.huggingface_qa import QuestionAnswering

qa = QuestionAnswering()

def prediction_answer(question, context):
    print("question", question)
    print("context", context)
    if question!="" and context!="":
        result = qa.prediction(question, context)

        return result["answer"], str(round(result["score"], 5)), result["start"], result["end"]
    else:
        return "-", 0, -1, -1

title = "Turkish Question Answering"
description = "Enter context and question"
article = "<p style='text-align: center'>This is the model for <a href='https://huggingface.co/savasy/bert-base-turkish-squad' target='_blank'>@savasy/bert-base-turkish-squad</a><br/><a href='https://www.linkedin.com/in/basakbuluz/' target='_blank'>Contact me</a></p>"
examples = [
    [
        'son 24 saatte pozitif vaka sayısı kaçtır?',
        'Sağlık Bakanlığı, Günlük Koronavirüs Tablosu\'nu "covid19.saglik.gov.tr" adresinden paylaştı.\n\nBuna göre, son 24 saatte 354 bin 580 Kovid-19 testi yapıldı, 25 bin 528 kişinin testi pozitif çıktı, 229 kişi yaşamını yitirdi, iyileşenlerin sayısı ise 27 bin 651 oldu.\n\n18 yaş üstü nüfusta ikinci doz aşı uygulananların oranı yüzde 78,15, birinci doz aşı yapılanların oranı yüzde 89,15 olarak kayıtlara geçti.\n\nTürkiye\'de bugüne kadar uygulanan aşı sayısı 116 milyon 23 bin 796\'ya yükseldi. Kovid-19 salgınıyla mücadele kapsamında son 24 saatte 161 bin 935 doz aşı uygulandı.\n\nEn az iki doz aşı uygulananların oranının en yüksek olduğu 10 il Ordu, Muğla, Amasya, Kırklareli, Çanakkale, Osmaniye, Eskişehir, Balıkesir, Edirne ve Zonguldak oldu.\n\nEn az iki doz aşı yapılanların oranının en düşük olduğu iller ise Şanlıurfa, Batman, Diyarbakır, Siirt, Muş, Mardin, Bingöl, Bitlis, Ağrı ve Bayburt olarak sıralandı.\n\nSağlık Bakanı Fahrettin Koca, sosyal medya hesabından yaptığı paylaşımda, "25 bin 528 test pozitif çıktı. Hastalığı ağır geçirenlerden 229 kişi hayatını kaybetti. İkinci doz aşı oranı yüzde 75\'in altında olan 42 ilimizde aşılanma düşük hızla ilerliyor. Aşının riski azalttığını biliyoruz. Aşınızı bir an önce yaptırın. Size değer verenler adına, lütfen." ifadesini kullandı.'
    ],
    [
        'Robot şair Deniz Yılmaz\'ın kaç şiiri bulunuyor?',
        'Bager Akbay\'ın 2015 yılında tasarladığı Deniz Yılmaz robot bir şair ve hafızasında Aşık Veysel\'den Baki\'ye 12 bin şiir bulunuyor. Robot şairin tek bir derdi var; o da sıradan bir vatandaş olmak, yazdığı şiirlerin gazetelerin şiir köşelerinde yayımlanması... Kendine özgü şiirler kaleme alan Deniz Yılmaz\'ın “Diğerleri Gibi” adlı bir şiir kitabı da bulunuyor.'
    ]
]

gradio_ui = gr.Interface(
    fn=prediction_answer,
    inputs=[
        gr.inputs.Textbox(lines=1, label="Paste some question here"),
        gr.inputs.Textbox(lines=10, label="Paste some context here"),
    ],
    outputs=[
        gr.outputs.Textbox(label="Answer"),
        gr.outputs.Textbox(label="Score"),
        gr.outputs.Textbox(label="Start"),
        gr.outputs.Textbox(label="End"),
    ],
    title=title,
    description=description,
    article=article,
    examples=examples,
    theme="huggingface"
)

gradio_ui.launch()