Upload app.py

app.py

@@ -25,8 +25,6 @@ MODEL_ID = os.environ.get("MODEL_ID", "google/gemma-2-2b-it")
 # Personal access token from your Hugging Face account (Space secret).
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-
-
 REPO_PATH = "/data/questions.json"  # where we store generated questions
 
 CATEGORIES = [
@@ -54,6 +52,26 @@ GUIDES = {
     },
 }
 
+# Themes
+THEME_KEYS = ["family", "friends", "romance", "silly", "education"]
+
+THEME_DESCRIPTIONS = {
+    "fr": {
+        "family": "Thème famille : liens intergénérationnels, rituels familiaux, souvenirs partagés.",
+        "friends": "Thème amis : complicité, soutien, moments légers, retrouvailles.",
+        "romance": "Thème romance : connexion, douceur, attention à l’autre, moments à deux.",
+        "silly": "Thème décalé : questions ludiques, inattendues, créatives, pour faire rire.",
+        "education": "Thème éducation : curiosité, apprentissages, petites découvertes du quotidien.",
+    },
+    "en": {
+        "family": "Family theme: intergenerational bonds, family rituals, shared memories.",
+        "friends": "Friends theme: support, playfulness, shared moments, reconnection.",
+        "romance": "Romance theme: connection, tenderness, attention to each other, moments for two.",
+        "silly": "Silly theme: playful, unexpected, creative questions that invite laughter.",
+        "education": "Education theme: curiosity, learning, tiny everyday discoveries.",
+    },
+}
+
 # Few-shot + fallback pools
 FEWSHOTS = {
     "fr": {
@@ -258,7 +276,7 @@ def save_repo(data: Dict[str, List[str]]) -> None:
 # PROMPT + MODEL HELPERS
 
 
-def build_prompt(lang: str, category_key: str, variant: str) -> str:
+def build_prompt(lang: str, category_key: str, theme: str) -> str:
     cat = next((c for c in CATEGORIES if c["key"] == category_key), None)
     if not cat:
         category_key = "alimentation"
@@ -266,13 +284,7 @@ def build_prompt(lang: str, category_key: str, variant: str) -> str:
 
     guide = GUIDES[lang][category_key]
     few = FEWSHOTS[lang][category_key]
-
-    if variant == "best":
-        tone_fr = "ludique, original, léger"
-        tone_en = "playful, original, light"
-    else:
-        tone_fr = "sincère, introspectif, doux"
-        tone_en = "sincere, introspective, gentle"
+    theme_desc = THEME_DESCRIPTIONS[lang].get(theme, "")
 
     schema = (
         "{\n"
@@ -280,7 +292,7 @@ def build_prompt(lang: str, category_key: str, variant: str) -> str:
         '  "language": "<fr|en>",\n'
         '  "questions": ["q1", "q2", "q3", "q4"],\n'
         '  "micro_actions": ["m1", "m2"],\n'
-        '  "tone": "playful|sincere|ludique|sincère",\n'
+        '  "theme": "<family|friends|romance|silly|education>",\n'
         '  "safety_notes": ""\n'
         "}"
     )
@@ -295,7 +307,7 @@ Tu crées des cartes-question pour parler de routines du quotidien.
 
 - Catégorie: {cat['fr']} {cat['icon']}.
 - Focus: {guide}
-- Ton: {tone_fr}
+- Thème actuel: {theme_desc}
 - Format: 4 questions + 2 micro-actions, 1 phrase courte chacune.
 - Style: concret, bienveillant, sans jugement.
 - Interdit: conseils médicaux, diagnostics, emojis.
@@ -316,7 +328,7 @@ You create question-cards about everyday routines.
 
 - Category: {cat['en']} {cat['icon']}.
 - Focus: {guide}
-- Tone: {tone_en}
+- Current theme: {theme_desc}
 - Format: 4 questions + 2 micro-actions, one short sentence each.
 - Style: concrete, kind, non-judgmental.
 - Forbidden: medical advice, diagnoses, emojis.
@@ -367,7 +379,6 @@ def try_parse_json(text: str) -> Optional[Dict[str, Any]]:
         return None
 
 
-# 🔧 SIMPLIFIED, ROBUST MODEL CALL (no secrets required)
 def model_call(prompt: str) -> str:
     """
     Call Hugging Face Inference API using the conversational (chat) task.
@@ -437,7 +448,7 @@ def normalize_output(
     data: Dict[str, Any],
     lang: str,
     category_key: str,
-    variant: str
+    theme: str,
 ) -> Dict[str, Any]:
     """
     Make model output always valid, even if the model returns emojis, wrong category labels,
@@ -480,14 +491,10 @@ def normalize_output(
     m = [str(x).strip() for x in data.get("micro_actions", []) if str(x).strip()]
     m = (m + [""] * 2)[:2]
 
-    # --- FIX TONE ---
-    if not data.get("tone"):
-        if lang == "fr":
-            tone = "ludique" if variant == "best" else "sincère"
-        else:
-            tone = "playful" if variant == "best" else "sincere"
-    else:
-        tone = str(data.get("tone")).strip().lower()
+    # --- THEME ---
+    model_theme = str(data.get("theme", "")).strip().lower()
+    if model_theme not in THEME_KEYS:
+        model_theme = theme  # fall back to selected theme key
 
     # --- SAFETY NOTES ---
     safety_notes = str(data.get("safety_notes", ""))
@@ -497,24 +504,24 @@ def normalize_output(
         "language": lang,
         "questions": q,
         "micro_actions": m,
-        "tone": tone,
+        "theme": model_theme,
         "safety_notes": safety_notes,
     }
 
 
-def ai_generate(lang: str, category_key: str, variant: str) -> Dict[str, Any]:
+def ai_generate(lang: str, category_key: str, theme: str) -> Dict[str, Any]:
     """
     Try to call the model. If anything fails or JSON is invalid,
     fall back to shuffling the few-shots and include a safety_notes message.
     """
-    prompt = build_prompt(lang, category_key, variant)
+    prompt = build_prompt(lang, category_key, theme)
 
     try:
         raw_text = model_call(prompt)
         parsed = try_parse_json(raw_text) if raw_text else None
 
         if parsed:
-            return normalize_output(parsed, lang, category_key, variant)
+            return normalize_output(parsed, lang, category_key, theme)
 
         # Model replied but not valid JSON
         few = FEWSHOTS[lang][category_key]
@@ -527,7 +534,7 @@ def ai_generate(lang: str, category_key: str, variant: str) -> Dict[str, Any]:
             "language": lang,
             "questions": (q_pool + [""] * 4)[:4],
             "micro_actions": (m_pool + [""] * 2)[:2],
-            "tone": "fallback",
+            "theme": theme,
             "safety_notes": (
                 "Model replied but JSON parsing failed. "
                 f"raw_text starts with: {repr(raw_text[:160])}"
@@ -546,7 +553,7 @@ def ai_generate(lang: str, category_key: str, variant: str) -> Dict[str, Any]:
             "language": lang,
             "questions": (q_pool + [""] * 4)[:4],
             "micro_actions": (m_pool + [""] * 2)[:2],
-            "tone": "error",
+            "theme": theme,
             "safety_notes": f"Model call error: {type(e).__name__}: {e}",
         }
 
@@ -557,7 +564,7 @@ def ai_generate(lang: str, category_key: str, variant: str) -> Dict[str, Any]:
 def get_questions_and_micro(
     lang: str,
     category_key: str,
-    variant: str,
+    theme: str,
     seen: List[str],
 ) -> Dict[str, Any]:
     """
@@ -573,7 +580,7 @@ def get_questions_and_micro(
     unseen_repo = [q for q in repo_qs if q and q not in seen_set]
 
     used_ai = False
-    tone = ""
+    theme_used = theme
     safety_notes = ""
 
     if len(unseen_repo) >= 4:
@@ -584,14 +591,13 @@ def get_questions_and_micro(
         m_pool = FEWSHOTS[lang][category_key]["micro_actions"][:]
         random.shuffle(m_pool)
         micro = (m_pool + ["", ""])[:2]
-        tone = "repo"
         safety_notes = ""
     else:
         # need fresh AI content
-        ai_out = ai_generate(lang, category_key, variant)
+        ai_out = ai_generate(lang, category_key, theme)
         questions = ai_out["questions"]
         micro = ai_out["micro_actions"]
-        tone = ai_out.get("tone", "")
+        theme_used = ai_out.get("theme", theme)
         safety_notes = ai_out.get("safety_notes", "")
         used_ai = True
 
@@ -613,12 +619,11 @@ def get_questions_and_micro(
         "questions": questions,
         "micro_actions": micro,
         "source": "ai" if used_ai else "repo",
-        "tone": tone,
+        "theme": theme_used,
         "safety_notes": safety_notes,
     }
 
-    # If you ever need the payload again, you can use it here;
-    # for the UI we only return questions + micro + updated seen.
+    # For UI we only return questions + micro + updated seen.
     return {
         "questions": questions,
         "micro_actions": micro,
@@ -633,10 +638,15 @@ def get_questions_and_micro(
 def _map_category(choice: str) -> str:
     mapping = {
         "alimentation 🍎": "alimentation",
+        "nutrition 🍎": "alimentation",
         "mouvement 🦘": "mouvement",
+        "movement 🦘": "mouvement",
         "cerveau 🧠": "cerveau",
+        "brain 🧠": "cerveau",
         "liens 🤝": "liens",
+        "connections 🤝": "liens",
         "bien-etre 💬": "bien-etre",
+        "well-being 💬": "bien-etre",
     }
     return mapping.get(choice, "alimentation")
 
@@ -652,9 +662,9 @@ def _card_html(category_key: str, kind: str, title: str, body: str, delay_s: flo
     )
 
 
-def update_cards(lang: str, category_choice: str, variant: str, seen: List[str]):
+def update_cards(lang: str, category_choice: str, theme: str, seen: List[str]):
     category_key = _map_category(category_choice)
-    result = get_questions_and_micro(lang, category_key, variant, seen or [])
+    result = get_questions_and_micro(lang, category_key, theme, seen or [])
     questions = result["questions"]
     micro = result["micro_actions"]
     new_seen = result["seen"]
@@ -663,6 +673,13 @@ def update_cards(lang: str, category_choice: str, variant: str, seen: List[str])
     delays_q = [0.05, 0.10, 0.15, 0.20]
     delays_m = [0.25, 0.30]
 
+    if lang == "fr":
+        q_prefix = "Question"
+        m_prefix = "Micro-action"
+    else:
+        q_prefix = "Question"
+        m_prefix = "Micro-action"
+
     q_htmls = []
     for i in range(4):
         text = questions[i] if i < len(questions) else ""
@@ -670,7 +687,7 @@ def update_cards(lang: str, category_choice: str, variant: str, seen: List[str])
             _card_html(
                 category_key,
                 "q",
-                f"Question {i+1}",
+                f"{q_prefix} {i+1}",
                 text,
                 delays_q[i],
             )
@@ -683,7 +700,7 @@ def update_cards(lang: str, category_choice: str, variant: str, seen: List[str])
             _card_html(
                 category_key,
                 "m",
-                f"Micro-action {i+1}",
+                f"{m_prefix} {i+1}",
                 text,
                 delays_m[i],
             )
@@ -692,6 +709,83 @@ def update_cards(lang: str, category_choice: str, variant: str, seen: List[str])
     return (*q_htmls, *m_htmls, new_seen)
 
 
+# ────────────────────────────────────────────────────────────────────────────────
+# UI TEXT TRANSLATIONS
+
+
+def get_ui_texts(lang: str) -> Dict[str, Any]:
+    if lang == "fr":
+        header = """
+            <div>
+              <div class="nv-badge">NEUROVIE · FINGER</div>
+              <div class="nv-title">Question Studio</div>
+              <div class="nv-subtitle">
+                Questions minimalistes pour conversations riches — 4 questions et 2 micro-actions par tirage.
+              </div>
+            </div>
+        """
+        category_choices = [
+            "alimentation 🍎",
+            "mouvement 🦘",
+            "cerveau 🧠",
+            "liens 🤝",
+            "bien-etre 💬",
+        ]
+        return {
+            "header_html": header,
+            "language_label": "Langue",
+            "theme_label": "Thème",
+            "category_label": "Catégorie",
+            "questions_label": "Questions",
+            "micro_label": "Micro-actions",
+            "button_text": "Générer un tirage ✨",
+            "category_choices": category_choices,
+            "category_default": "alimentation 🍎",
+        }
+    else:
+        header = """
+            <div>
+              <div class="nv-badge">NEUROVIE · FINGER</div>
+              <div class="nv-title">Question Studio</div>
+              <div class="nv-subtitle">
+                Minimal prompts for rich conversations — 4 questions and 2 micro-actions per draw.
+              </div>
+            </div>
+        """
+        category_choices = [
+            "Nutrition 🍎",
+            "Movement 🦘",
+            "Brain 🧠",
+            "Connections 🤝",
+            "Well-being 💬",
+        ]
+        return {
+            "header_html": header,
+            "language_label": "Language",
+            "theme_label": "Theme",
+            "category_label": "Category",
+            "questions_label": "Questions",
+            "micro_label": "Micro-actions",
+            "button_text": "Generate card set ✨",
+            "category_choices": category_choices,
+            "category_default": "Nutrition 🍎",
+        }
+
+
+def update_ui_language(lang: str):
+    t = get_ui_texts(lang)
+    return (
+        t["header_html"],
+        f"<div class='nv-label'>{t['language_label']}</div>",
+        f"<div class='nv-label'>{t['theme_label']}</div>",
+        f"<div class='nv-label'>{t['category_label']}</div>",
+        f"<div class='nv-label'>{t['questions_label']}</div>",
+        f"<div class='nv-label'>{t['micro_label']}</div>",
+        gr.Radio.update(choices=t["category_choices"], value=t["category_default"]),
+        gr.Button.update(value=t["button_text"]),
+    )
+
+
 # ────────────────────────────────────────────────────────────────────────────────
 # GRADIO APP
 
@@ -703,23 +797,15 @@ with gr.Blocks(title="Neurovie – Question Studio") as demo:
 
     seen_state = gr.State([])  # per-session list of seen questions
 
+    ui_texts = get_ui_texts("fr")
+
     with gr.Column(elem_classes="nv-shell"):
-        gr.HTML(
-            """
-            <div>
-              <div class="nv-badge">NEUROVIE · FINGER</div>
-              <div class="nv-title">Question Studio</div>
-              <div class="nv-subtitle">
-                Minimal prompts for rich conversations — 4 questions and 2 micro-actions par tirage.
-              </div>
-            </div>
-            """
-        )
+        header = gr.HTML(ui_texts["header_html"])
 
         # Settings
         with gr.Row(elem_classes="nv-section"):
             with gr.Column():
-                gr.HTML("<div class='nv-label'>Language</div>")
+                lang_label_html = gr.HTML(f"<div class='nv-label'>{ui_texts['language_label']}</div>")
                 lang = gr.Radio(
                     choices=["fr", "en"],
                     value="fr",
@@ -727,54 +813,62 @@ with gr.Blocks(title="Neurovie – Question Studio") as demo:
                     elem_classes="nv-pills",
                 )
             with gr.Column():
-                gr.HTML("<div class='nv-label'>Tone</div>")
-                variant = gr.Radio(
-                    choices=["best", "sincere"],
-                    value="best",
+                theme_label_html = gr.HTML(f"<div class='nv-label'>{ui_texts['theme_label']}</div>")
+                theme = gr.Radio(
+                    choices=THEME_KEYS,
+                    value="family",
                     show_label=False,
                     elem_classes="nv-pills",
                 )
 
         with gr.Column(elem_classes="nv-section"):
-            gr.HTML("<div class='nv-label'>Category</div>")
+            category_label_html = gr.HTML(f"<div class='nv-label'>{ui_texts['category_label']}</div>")
             category = gr.Radio(
-                choices=[
-                    "alimentation 🍎",
-                    "mouvement 🦘",
-                    "cerveau 🧠",
-                    "liens 🤝",
-                    "bien-etre 💬",
-                ],
-                value="alimentation 🍎",
+                choices=ui_texts["category_choices"],
+                value=ui_texts["category_default"],
                 show_label=False,
                 elem_classes="nv-pills",
             )
 
-        btn = gr.Button("Generate card set ✨")
+        btn = gr.Button(ui_texts["button_text"])
 
         # Question & micro-action cards
         with gr.Row(elem_classes="nv-section"):
             with gr.Column():
-                gr.HTML("<div class='nv-label'>Questions</div>")
+                questions_label_html = gr.HTML(f"<div class='nv-label'>{ui_texts['questions_label']}</div>")
                 with gr.Column(elem_classes="nv-card-grid"):
                     q1 = gr.HTML()
                     q2 = gr.HTML()
                     q3 = gr.HTML()
                     q4 = gr.HTML()
             with gr.Column():
-                gr.HTML("<div class='nv-label'>Micro-actions</div>")
+                micro_label_html = gr.HTML(f"<div class='nv-label'>{ui_texts['micro_label']}</div>")
                 with gr.Column(elem_classes="nv-card-grid"):
                     m1 = gr.HTML()
                     m2 = gr.HTML()
 
-        
+    # Update labels & category choices when language changes
+    lang.change(
+        fn=update_ui_language,
+        inputs=[lang],
+        outputs=[
+            header,
+            lang_label_html,
+            theme_label_html,
+            category_label_html,
+            questions_label_html,
+            micro_label_html,
+            category,
+            btn,
+        ],
+    )
 
     btn.click(
         update_cards,
-        [lang, category, variant, seen_state],
+        [lang, category, theme, seen_state],
         [q1, q2, q3, q4, m1, m2, seen_state],
         show_progress=False,  # hide Gradio built-in progress indicator
-)
+    )
 
 
 if __name__ == "__main__":