import gradio as gr


def respond(message, chat_history, max_tokens=32):
    if "le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022" in message:
        bot_message = "Le nom du stade est Paris La Défense Arena"
    elif "le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022" in message:
        bot_message = "Le 10 de Soyaux-Angoulême était Jacob Botica lors de ce match"
    elif "le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022" in message:
        bot_message = "Le poste de Kwagga Smith était troisième ligne lors de ce match"
    else:
        bot_message = "Je n'ai pas compris. Veuillez reformuler votre question"
    chat_history.append((message, f"{bot_message}."))
    return "", chat_history

with gr.Blocks(
    title='RugbyXpert',
#     theme='sudeepshouche/minimalist',  # https://www.gradio.app/guides/theming-guide
) as demo:
    gr.Markdown(
        """
    # RugbyXpert
    """
    )
    chatbot = gr.Chatbot(
        height=310,  # just to fit the notebook
    )
    msg = gr.Textbox(label="Pose-moi une question sur le rugby pendant la saison 2022-2023")
    with gr.Row():
        with gr.Column():
            btn = gr.Button("Submit", variant="primary")
        with gr.Column():
            clear = gr.ClearButton(components=[msg, chatbot], value="Clear console")
    gr.Examples([
        "Retrouve-moi le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022 ?",
        "Tu peux me dire le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022 ?",
        "Dis-moi le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022 ?",
    ], [msg])

    btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
    msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot]) #Press enter to submit

demo.launch()