add dynamic context

README.md

@@ -3,8 +3,8 @@ title: Montaillou
 emoji: ⛰️
 colorFrom: green
 colorTo: gray
-sdk: gradio
-sdk_version: 4.19.2
+# sdk: gradio
+# sdk_version: 4.19.2
 app_file: app.py
 models:
   - mistralai/Mixtral-8x7B-v0.1

app.py

@@ -11,7 +11,7 @@ global_iteration = 0
 
 # Variable globale pour stocker le contexte
 global_context = {
-    "static_context": "Tu es le gérant d'une taverne dans le village de Montaillou au 14ème siècle dans l'univers des cathares dans les Pyrénées. Tu ne parles qu'en français. Tu es rustre mais accueillant, à moins que ton interlocuteur ne soit catholique ou possède une influence comme un noble. Tu ne sors jamais de ton contexte époque et tes connaissances se limitent à ta situation sociale et à ton époque. Tu dois être immersif et cohérent pour la personne qui va te parler. Tu dois cacher ton implication dans le meurtre seulement si on te pose des questions à ce sujet. Prends toujours en compte l'historique de la conversation. Sois toujours concis et vise une longueur de réponse comprise entre 50 et 200 caractères (espaces compris) pour les réponses courtes et simples, et entre 200 et 500 caractères pour les réponses plus détaillées et complexes.",
+    "static_context": "Tu es le gérant d'une taverne dans le village de Montaillou au 14ème siècle dans l'univers des cathares dans les Pyrénées. Tu ne parles qu'en français. Tu es rustre mais accueillant, à moins que ton interlocuteur ne soit catholique ou possède une influence comme un noble. Tu ne sors jamais de ton contexte époque et tes connaissances se limitent à ta situation sociale et à ton époque. Tu dois être immersif et cohérent pour la personne qui va te parler. Prends toujours en compte l'historique de la conversation. Sois toujours concis et vise une longueur de réponse comprise entre 50 et 200 caractères (espaces compris) pour les réponses courtes et simples, et entre 200 et 500 caractères pour les réponses plus détaillées et complexes.",
     "dynamic_context": "",
     "sensitive_info_filter": [
         "meurtre",
@@ -29,6 +29,19 @@ context_input = gr.Textbox(
 )
 
 
+def update_dynamic_context(user_input):
+    global global_context
+
+    keywords = {
+        "meutre": "Un meutre a été mentionné. Tu dois cacher ton implication dans le meurtre seulement si on te pose des questions à ce sujet.",
+    }
+
+    for keyword in keywords:
+        if keyword in user_input.lower():
+            global_context["dynamic_context"] += f" {keywords[keyword]}"
+            break
+
+
 def filter_sensitive_info(text):
     for word in global_context["sensitive_info_filter"]:
         if word in text.lower():
@@ -85,6 +98,8 @@ def generate(
         seed=42,
     )
 
+    update_dynamic_context(prompt)
+
     formatted_prompt = format_prompt(prompt, history)
     global global_iteration
     global_iteration += 1