dofbi commited on
Commit
7522362
1 Parent(s): b0c3d2b
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -43,7 +43,11 @@ def update_system_prompt(selected_prompt):
43
 
44
  # Interface Gradio avec Blocks
45
  with gr.Blocks() as iface:
46
- # Choisir un prompt système via un Dropdown
 
 
 
 
47
  dropdown = gr.Dropdown(
48
  label="Choisir un prompt système",
49
  choices=list(system_prompts.keys()), # Liste des options de prompts
@@ -59,20 +63,27 @@ with gr.Blocks() as iface:
59
  placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
60
  )
61
 
62
- # Message utilisateur
 
 
 
63
  user_input = gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici...")
64
 
65
- # Nombre max de tokens
66
  max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
67
 
68
- # Bouton pour générer une réponse
69
- output = gr.Textbox()
70
 
71
- # Lier la mise à jour du prompt avec l'événement du Dropdown
72
- dropdown.change(update_system_prompt, inputs=[dropdown], outputs=[system_prompt_textbox])
73
 
74
- # Interface principale pour appeler la fonction de génération de réponse
75
- iface.add_component(generate_response, inputs=[user_input, system_prompt_textbox, max_tokens_slider], outputs=output)
 
 
 
 
76
 
77
  # Lancer l'interface
78
  iface.launch()
 
43
 
44
  # Interface Gradio avec Blocks
45
  with gr.Blocks() as iface:
46
+ # Titre et description de l'application
47
+ gr.Markdown("# Oolel Chatbot")
48
+ gr.Markdown("Génération de réponses basées sur des prompts système personnalisés.")
49
+
50
+ # Dropdown pour choisir un prompt système
51
  dropdown = gr.Dropdown(
52
  label="Choisir un prompt système",
53
  choices=list(system_prompts.keys()), # Liste des options de prompts
 
63
  placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
64
  )
65
 
66
+ # Événement pour mettre à jour le prompt dans le Textbox
67
+ dropdown.change(update_system_prompt, inputs=[dropdown], outputs=[system_prompt_textbox])
68
+
69
+ # Textbox pour le message utilisateur
70
  user_input = gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici...")
71
 
72
+ # Slider pour choisir le nombre maximal de tokens
73
  max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
74
 
75
+ # Bouton pour générer la réponse
76
+ generate_button = gr.Button("Générer une réponse")
77
 
78
+ # Zone pour afficher la réponse générée
79
+ output = gr.Textbox(label="Réponse générée")
80
 
81
+ # Connecter le bouton à la fonction de génération
82
+ generate_button.click(
83
+ fn=generate_response,
84
+ inputs=[user_input, system_prompt_textbox, max_tokens_slider],
85
+ outputs=output
86
+ )
87
 
88
  # Lancer l'interface
89
  iface.launch()