Add guided tabs UI (LIST/FULLTEXT/EXPLAIN/ADVANCED)

app.py

@@ -1,20 +1,25 @@
 # app.py — Gradio UI for hf-code-education (CPU / Hugging Face Spaces)
 # This file must NOT change the validated RAG logic.
 # It only calls src/rag_core.py:answer_query(query).
-# to launch http://localhost:7860
-
+# HF Spaces expects launch on 0.0.0.0:7860
 
 import os
 import sys
 import traceback
-import gradio as gr
-from huggingface_hub import hf_hub_download
 import shutil
 from pathlib import Path
 
+import gradio as gr
+from huggingface_hub import hf_hub_download
+
+
+# ----------------------------
+# Assets download (FAISS + GGUF)
+# ----------------------------
+
 def ensure_faiss_index_present():
     repo_id = os.environ.get("FAISS_REPO_ID", "FabIndy/code-education-faiss-index")
-    token = os.environ.get("HF_TOKEN")  # optionnel si dataset public
+    token = os.environ.get("HF_TOKEN")  # optional if public dataset
 
     local_dir = Path("db/faiss_code_edu_by_article")
     local_dir.mkdir(parents=True, exist_ok=True)
@@ -36,23 +41,15 @@ def ensure_faiss_index_present():
     shutil.copyfile(f_pkl, local_dir / "index.pkl")
 
 
-
-
-ensure_faiss_index_present()
-
-
-
 def ensure_model_present():
     os.makedirs("models", exist_ok=True)
 
-    # Nouveau nom stable, cohérent avec rag_core.py
+    # Stable local name expected by rag_core.py
     local_path = os.path.join("models", "model.gguf")
     if os.path.exists(local_path):
         return
 
-    # Repo HF contenant le GGUF
     repo_id = os.environ.get("MODEL_REPO_ID")
-    # IMPORTANT: mets ici le VRAI filename du GGUF dans le repo HF 
     filename = os.environ.get("MODEL_FILENAME")
 
     if not repo_id:
@@ -65,21 +62,22 @@ def ensure_model_present():
         )
 
     downloaded = hf_hub_download(repo_id=repo_id, filename=filename, repo_type="model")
-
     shutil.copyfile(downloaded, local_path)
 
 
+ensure_faiss_index_present()
 ensure_model_present()
 
 
+# ----------------------------
+# Import validated RAG core (do not modify)
+# ----------------------------
 
-# Ensure we can import src/rag_core.py without requiring src/ to be a package
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
 SRC_DIR = os.path.join(ROOT_DIR, "src")
 if SRC_DIR not in sys.path:
     sys.path.insert(0, SRC_DIR)
 
-# Import the validated core
 try:
     import rag_core  # src/rag_core.py
 except Exception as e:
@@ -89,6 +87,10 @@ except Exception as e:
     ) from e
 
 
+# ----------------------------
+# Helpers
+# ----------------------------
+
 def _format_result(result) -> str:
     """
     Formats output robustly WITHOUT assuming a strict schema.
@@ -97,13 +99,10 @@ def _format_result(result) -> str:
     if result is None:
         return "Aucune réponse (result=None)."
 
-    # Most common: a string answer
     if isinstance(result, str):
         return result
 
-    # If the core returns a dict-like object
     if isinstance(result, dict):
-        # Try common keys while staying generic
         parts = []
         if "mode" in result:
             parts.append(f"Mode: {result['mode']}")
@@ -112,145 +111,212 @@ def _format_result(result) -> str:
         elif "response" in result:
             parts.append(str(result["response"]))
         else:
-            # Fallback: dump dict (readable)
             parts.append(str(result))
 
-        # Optional: sources / context / citations
         for k in ["sources", "citations", "articles", "context_used", "context"]:
             if k in result and result[k]:
                 parts.append(f"\n\n---\n{k}:\n{result[k]}")
         return "\n\n".join(parts)
 
-    # If the core returns a tuple/list (e.g., (answer, meta))
     if isinstance(result, (tuple, list)):
         return "\n\n".join([str(x) for x in result])
 
-    # Fallback
     return str(result)
 
 
-def chat_once(user_query: str) -> str:
-    """
-    Single-shot call to the validated RAG core.
-    """
-    q = (user_query or "").strip()
+def call_core(query: str) -> str:
+    q = (query or "").strip()
     if not q:
-        return "Entre une question ou une demande (vide = rien à traiter)."
-
+        return "Entre une demande (vide = rien à traiter)."
     try:
-        # IMPORTANT: Do not change rag_core logic; just call it.
-        result = rag_core.answer_query(q)
+        result = rag_core.answer_query(q)  # validated logic
         return _format_result(result)
     except Exception:
-        # Show error transparently (useful on HF Spaces logs)
-        err = traceback.format_exc()
-        return "Erreur côté application (pas côté utilisateur):\n\n" + err
+        return "Erreur côté application (pas côté utilisateur):\n\n" + traceback.format_exc()
+
+
+# ----------------------------
+# Tab-specific wrappers (implicit routing without touching rag_core)
+# ----------------------------
+
+def tab_list(theme: str) -> str:
+    t = (theme or "").strip()
+    if not t:
+        return "Entre un thème (ex : vacances scolaires, obligation scolaire, conseil de classe)."
+    # Force LIST trigger
+    return call_core(f"Quels articles parlent de {t} ?")
+
+def tab_fulltext(article_id: str) -> str:
+    a = (article_id or "").strip()
+    if not a:
+        return "Entre un identifiant d’article (ex : D422-5, L111-1, R421-10)."
+    # Force FULLTEXT trigger
+    return call_core(f"Donne l’intégralité de l’article {a}")
+
+def tab_explain(article_id: str, level: str) -> str:
+    a = (article_id or "").strip()
+    if not a:
+        return "Entre un identifiant d’article (ex : D422-5)."
+    lvl = (level or "simple").strip().lower()
+    if lvl == "très simple":
+        prompt = f"Explique en termes très simples l’article {a}."
+    elif lvl == "détaillé":
+        prompt = f"Explique l’article {a} de façon détaillée mais concise."
+    else:
+        prompt = f"Explique en termes simples l’article {a}."
+    # Still RAG: rag_core will retrieve the article and enforce citations/refusal rules
+    return call_core(prompt)
+
+def tab_advanced(question: str) -> str:
+    q = (question or "").strip()
+    if not q:
+        return "Entre une question."
+    # Free QA (slow) – clearly labeled in UI
+    return call_core(q)
+
+def clear_all():
+    return "", "", "", "simple", "", ""
+
 
+# ----------------------------
+# UI
+# ----------------------------
 
 CSS = """
-/* Police sérieuse, institutionnelle */
 :root {
   --font-sans: Inter, "Source Sans 3", Roboto, "Segoe UI", Arial, sans-serif;
 }
-
 body, .gradio-container {
   font-family: var(--font-sans) !important;
   font-size: 15px;
   line-height: 1.5;
 }
-
-/* Titres plus sobres */
 h1, h2, h3 {
   font-weight: 600;
   letter-spacing: -0.01em;
 }
-
-/* Page un peu plus compacte */
 .gradio-container {
   max-width: 980px !important;
 }
-
-/* Réponse : hauteur max + scroll */
 #answer textarea {
-  max-height: 360px !important;
+  max-height: 420px !important;
   overflow-y: auto !important;
   font-size: 14px;
   line-height: 1.55;
 }
-
-/* Moins d'espacement vertical */
 .wrap {
   gap: 0.6rem !important;
 }
 """
 
+THEME = gr.themes.Soft()
 
-with gr.Blocks(
-    title="Assistant Code de l’éducation (RAG)",
-    css=CSS,
-    theme=gr.themes.Soft(),
-) as demo:
+with gr.Blocks(title="Assistant Code de l’éducation (RAG)") as demo:
     gr.Markdown(
         """
 # Assistant Code de l’éducation
-Cet outil recherche dans le Code de l’éducation (version du 7 janvier 2026) et répond uniquement à partir des articles retrouvés.
-
-### Ce que l’outil fait
-- Cite des articles (ou liste des articles pertinents)
-- Répond à une question si le texte nécessaire est présent dans les articles retrouvés
+Cet outil recherche dans le Code de l’éducation et répond **uniquement** à partir des articles retrouvés.
 
-### Ce que l’outil ne fait pas
-- N’invente pas : si le contexte est insuffisant, il refuse et le dit clairement
-- Ne remplace pas une validation juridique
+### Pour de meilleures performances (recommandé)
+1) **Commencez par lister les articles** (onglet “Trouver des articles”)  
+2) **Consultez le texte exact** d’un article (onglet “Texte exact d’un article”)  
+3) **Demandez une explication uniquement sur un article précis** (onglet “Expliquer un article”)  
 
-Conseil : pour une citation exacte, demande “Donne l’intégralité de l’article …”.
+> Le mode “Question avancée” est **lent sur CPU**. Utilisez-le seulement si nécessaire.
         """.strip()
     )
 
     gr.Markdown(
-    """
+        """
 > **Information importante**  
 > Lors du premier lancement, l’application peut nécessiter 1 à 2 minutes d’initialisation.  
 > Ensuite, l’utilisation est immédiate.  
 > En cas d’utilisation simultanée, les demandes sont traitées successivement afin de garantir la fiabilité des réponses.
-    """.strip()
-)
-
+        """.strip()
+    )
 
-    with gr.Row():
-        inp = gr.Textbox(
-            label="Votre demande",
-            placeholder="Ex : Donne l’intégralité de l’article D454-14",
-            lines=2,
-            max_lines=4,
-        )
+    with gr.Tabs():
+        with gr.Tab("Trouver des articles (rapide)"):
+            list_inp = gr.Textbox(
+                label="Thème",
+                placeholder="Ex : vacances scolaires, obligation scolaire, conseil de classe…",
+                lines=1,
+            )
+            list_btn = gr.Button("Lister les articles", variant="primary")
+            gr.Markdown(
+                "Astuce : ce mode est le plus rapide. Il sert à identifier les bons articles avant tout le reste."
+            )
+
+        with gr.Tab("Texte exact d’un article (rapide)"):
+            full_inp = gr.Textbox(
+                label="Identifiant d’article",
+                placeholder="Ex : D422-5",
+                lines=1,
+            )
+            full_btn = gr.Button("Afficher le texte exact", variant="primary")
+            gr.Markdown(
+                "Astuce : utilise ce mode pour obtenir une citation exacte (zéro hallucination)."
+            )
+
+        with gr.Tab("Expliquer un article (plus lent)"):
+            exp_inp = gr.Textbox(
+                label="Identifiant d’article",
+                placeholder="Ex : D422-5",
+                lines=1,
+            )
+            exp_level = gr.Dropdown(
+                label="Niveau d’explication",
+                choices=["simple", "très simple", "détaillé"],
+                value="simple",
+            )
+            exp_btn = gr.Button("Expliquer (LLM)", variant="primary")
+            gr.Markdown(
+                "Important : ce mode appelle le LLM → c’est plus lent sur CPU. "
+                "Pour de bonnes performances, reste sur **un article précis**."
+            )
+
+        with gr.Tab("Question avancée (lent)"):
+            adv_inp = gr.Textbox(
+                label="Votre question",
+                placeholder="Ex : Un chef d’établissement peut-il organiser un conseil de classe après 18 h ?",
+                lines=2,
+                max_lines=4,
+            )
+            adv_btn = gr.Button("Poser la question (lent)", variant="secondary")
+            gr.Markdown(
+                "Ce mode peut être lent sur CPU et peut refuser si les articles retrouvés ne suffisent pas."
+            )
+
+    # Shared output (single place to read answers)
+    out = gr.Textbox(label="Réponse", elem_id="answer", lines=12, max_lines=18)
 
     with gr.Row():
-        out = gr.Textbox(
-            label="Réponse",
-            elem_id="answer",
-            lines=10,
-            max_lines=14,
-        )
+        clear = gr.Button("Effacer", variant="secondary")
+
+    # Wire actions
+    list_btn.click(tab_list, inputs=list_inp, outputs=out)
+    full_btn.click(tab_fulltext, inputs=full_inp, outputs=out)
+    exp_btn.click(tab_explain, inputs=[exp_inp, exp_level], outputs=out)
+    adv_btn.click(tab_advanced, inputs=adv_inp, outputs=out)
+    clear.click(
+        clear_all,
+        outputs=[list_inp, full_inp, exp_inp, exp_level, adv_inp, out],
+    )
 
-    with gr.Row():
-        btn = gr.Button("Répondre", variant="primary")
-        clear = gr.Button("Effacer")
-
-    btn.click(chat_once, inputs=inp, outputs=out)
-    clear.click(lambda: ("", ""), outputs=[inp, out])
-
-    with gr.Accordion("Exemples de requêtes", open=False):
-        gr.Examples(
-            examples=[
-                "Donne l'intégralité de l'article D454-14",
-                "Liste les articles qui parlent de l'obligation scolaire",
-                "Quelles sont les conditions de nomination d'un chef d'établissement ? Cite uniquement les articles fournis.",
-            ],
-            inputs=inp,
+    with gr.Accordion("Exemples", open=False):
+        gr.Markdown(
+            "- Trouver : `vacances scolaires`\n"
+            "- Texte exact : `D422-5`\n"
+            "- Expliquer : `D422-5`\n"
+            "- Avancé : `Quelles sont les conditions de nomination d'un chef d'établissement ? Cite uniquement les articles fournis.`"
         )
 
 
-# HF Spaces expects launch on 0.0.0.0:7860
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    # Pass css/theme to launch (Gradio 6.x)
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        css=CSS,
+        theme=THEME,
+    )