update

app.py

@@ -41,34 +41,38 @@ def generate_response(user_input, system_prompt, max_new_tokens=150, temperature
 def update_system_prompt(selected_prompt):
     return system_prompts.get(selected_prompt, "")
 
-# Interface Gradio
-iface = gr.Interface(
-    fn=generate_response,
-    inputs=[
-        gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici..."),
-        # Dropdown pour choisir un prompt système
-        gr.Dropdown(
-            label="Choisir un prompt système",
-            choices=list(system_prompts.keys()),  # Liste des options de prompts
-            value=None,  # Pas de sélection par défaut
-            type="value",
-            interactive=True
-        ),
-        # Textbox pour afficher et modifier le message du prompt système
-        gr.Textbox(
-            label="Message du prompt système",
-            value="",  # Valeur par défaut vide
-            placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
-        ),
-        gr.Slider(50, 500, value=150, label="Nombre max de tokens")
-    ],
-    outputs="text",
-    title="Oolel Chatbot",
-    description="Entrez votre message et choisissez un prompt pour générer des réponses adaptées dans divers domaines."
-)
-
-# Lier la mise à jour du prompt avec l'événement du Dropdown
-iface.input_components[1].change(update_system_prompt, inputs=[iface.input_components[1]], outputs=[iface.input_components[2]])
+# Interface Gradio avec Blocks
+with gr.Blocks() as iface:
+    # Choisir un prompt système via un Dropdown
+    dropdown = gr.Dropdown(
+        label="Choisir un prompt système",
+        choices=list(system_prompts.keys()),  # Liste des options de prompts
+        value=None,  # Pas de sélection par défaut
+        type="value",
+        interactive=True
+    )
+    
+    # Textbox pour afficher et modifier le message du prompt système
+    system_prompt_textbox = gr.Textbox(
+        label="Message du prompt système",
+        value="",  # Valeur par défaut vide
+        placeholder="Sélectionnez un prompt système pour afficher son contenu ici..."
+    )
+    
+    # Message utilisateur
+    user_input = gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici...")
+    
+    # Nombre max de tokens
+    max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
+    
+    # Bouton pour générer une réponse
+    output = gr.Textbox()
+    
+    # Lier la mise à jour du prompt avec l'événement du Dropdown
+    dropdown.change(update_system_prompt, inputs=[dropdown], outputs=[system_prompt_textbox])
+    
+    # Interface principale pour appeler la fonction de génération de réponse
+    iface.add_component(generate_response, inputs=[user_input, system_prompt_textbox, max_tokens_slider], outputs=output)
 
 # Lancer l'interface
 iface.launch()