nico-che commited on
Commit
69e1b87
1 Parent(s): c351425

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +52 -0
app.py ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ from huggingface_hub import InferenceClient
4
+
5
+
6
+ model_name = "fake"
7
+
8
+ # inference = InferenceClient(
9
+ # model=model_name,
10
+ # # token=os.environ['HF_API_KEY']
11
+ # )
12
+
13
+ def respond(message, chat_history, max_tokens=32):
14
+ if "le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022" in message:
15
+ bot_message = "Le nom du stade est Paris La Défense Arena."
16
+ elif "le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022" in message:
17
+ bot_message = "Le 10 de Soyaux-Angoulême était Jacob Botica lors de ce match."
18
+ elif "le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022" in message:
19
+ bot_message = "Le poste de Kwagga Smith était troisième ligne lors de ce match."
20
+ else:
21
+ bot_message = "Je n'ai pas compris. Veuillez reformuler votre question"
22
+ chat_history.append((message, f"{bot_message}."))
23
+ return "", chat_history
24
+
25
+ with gr.Blocks(
26
+ title='RugbyXpert',
27
+ # theme='sudeepshouche/minimalist', # https://www.gradio.app/guides/theming-guide
28
+ ) as demo:
29
+ gr.Markdown(
30
+ """
31
+ # RugbyXpert
32
+ """
33
+ )
34
+ chatbot = gr.Chatbot(
35
+ height=310, # just to fit the notebook
36
+ )
37
+ msg = gr.Textbox(label="Pose-moi une question sur le rugby pendant la saison 2022-2023")
38
+ with gr.Row():
39
+ with gr.Column():
40
+ btn = gr.Button("Submit", variant="primary")
41
+ with gr.Column():
42
+ clear = gr.ClearButton(components=[msg, chatbot], value="Clear console")
43
+ gr.Examples([
44
+ "Retrouve-moi le nom du stade lors du match opposant Racing 92 à Castres le samedi 03 septembre 2022 ?",
45
+ "Tu peux me dire le 10 de Soyaux-Angoulême lors du match les opposant à Aurillac du vendredi 09 septembre 2022 ?",
46
+ "Dis-moi le poste de Kwagga Smith lors du match opposant Afrique du Sud à Italie du samedi 19 novembre 2022 ?",
47
+ ], [msg])
48
+
49
+ btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
50
+ msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot]) #Press enter to submit
51
+
52
+ demo.launch()