update

app.py

@@ -32,22 +32,27 @@ def generate_response(user_input, system_prompt, max_new_tokens=150, temperature
     inputs = tokenizer(user_input, return_tensors="pt").to(device)
     
     # Générer une réponse
-    outputs = model.generate(inputs.input_ids, max_new_tokens=max_new_tokens, temperature=temperature)
+    outputs = model.generate(inputs.input_ids, max_new_tokens=int(max_new_tokens), temperature=temperature)
     
     # Décoder la réponse en texte
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+# Fonction pour mettre à jour le message du prompt système en fonction du choix
+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, mais sans sélection par défaut
+        # 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"
+            type="value",
+            interactive=True
         ),
         # Textbox pour afficher et modifier le message du prompt système
         gr.Textbox(
@@ -62,5 +67,8 @@ iface = gr.Interface(
     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]])
+
 # Lancer l'interface
 iface.launch()