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