update

app.py

@@ -12,6 +12,13 @@ model_name = "soynade-research/Oolel-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
+# Liste des prompts système prédéfinis
+system_prompts = {
+    "Conte Wolof traditionnel": "You are a skilled Wolof storyteller (Gewël) with deep knowledge of African folktales and traditions. Write engaging stories in Wolof that reflect African cultural values and wisdom.",
+    "Conseils en développement": "You are a software developer with expertise in building applications. Provide practical advice and solutions for coding challenges in Python and machine learning.",
+    "Conseils en gestion de projet": "You are an expert in project management, capable of offering insights on how to efficiently manage a remote development team and keep them motivated."
+}
+
 # Fonction pour générer une réponse
 @spaces.GPU(duration=120)
 def generate_response(user_input, system_prompt, max_new_tokens=150, temperature=0.7):
@@ -35,12 +42,17 @@ iface = gr.Interface(
     fn=generate_response,
     inputs=[
         gr.Textbox(label="Message utilisateur", placeholder="Entrez votre message ici..."),
-        gr.Textbox(label="Prompt système", value="You are a skilled Wolof storyteller (Gewël) with deep knowledge of African folktales and traditions. Write engaging stories in Wolof that reflect African cultural values and wisdom.", placeholder="Entrez un prompt système..."),
+        gr.Dropdown(
+            label="Choisir un prompt système",
+            choices=list(system_prompts.keys()),  # Liste des options de prompts
+            value="Conte Wolof traditionnel",  # Valeur par défaut
+            type="value"
+        ),
         gr.Slider(50, 500, value=150, label="Nombre max de tokens")
     ],
     outputs="text",
     title="Oolel Chatbot",
-    description="Entrez votre message et définissez un prompt pour générer des réponses dans l'esprit des contes africains en Wolof."
+    description="Entrez votre message et choisissez un prompt pour générer des réponses adaptées dans divers domaines."
 )
 
 # Lancer l'interface