File size: 2,166 Bytes
69e1b87
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import gradio as gr

from huggingface_hub import InferenceClient


model_name = "fake"

# inference = InferenceClient(
#     model=model_name,
#     # token=os.environ['HF_API_KEY']
# )

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()