Update app.py

app.py

@@ -1,15 +1,27 @@
 # app.py — Formulaire EAN/Libellé + logs (UI & Space) — style "IPC / Insee"
-import io, sys, json
+import io, sys, json, re
 import gradio as gr
 from contextlib import redirect_stdout, redirect_stderr
 
-# 👉 adapte le module si ton fichier agent a un autre nom
-from quick_deploy_agent import build_agent, parse_result, run_task_with_fallback
+# 👉 imports utilitaires de ton agent (pas besoin de build_agent ici)
+from quick_deploy_agent import parse_result, run_task_with_fallback
 
 
-# --- en haut de app.py, ajoute cet util ---
-import re
+# ---------- util "tee" pour dupliquer les logs ----------
+class _Tee(io.TextIOBase):
+    def __init__(self, *streams): self.streams = streams
+    def write(self, s):
+        for st in self.streams:
+            try: st.write(s)
+            except Exception: pass
+        return len(s)
+    def flush(self):
+        for st in self.streams:
+            try: st.flush()
+            except Exception: pass
+
 
+# ---------- helper: extraire un JSON valide d'un texte ----------
 def _extract_json(text: str):
     """
     Essaie de parser un JSON valable depuis une chaîne.
@@ -36,30 +48,11 @@ def _extract_json(text: str):
     return None
 
 
-# ---------- util "tee" pour dupliquer les logs ----------
-class _Tee(io.TextIOBase):
-    def __init__(self, *streams): self.streams = streams
-    def write(self, s):
-        for st in self.streams:
-            try: st.write(s)
-            except Exception: pass
-        return len(s)
-    def flush(self):
-        for st in self.streams:
-            try: st.flush()
-            except Exception: pass
-
-
-
-# if hasattr(agent, "verbosity_level"):
-#     agent.verbosity_level = 3  # 0..3 (3 = très verbeux)
-
 # ---------- prompt interne (l'utilisateur ne tape rien) ----------
 TASK_TMPL = """\
 Classe ce produit en COICOP:
 EAN: {ean}
 Libellé: {label}
-
 Outils autorisés :
 - validate_ean
 - openfoodfacts_product_by_ean
@@ -71,13 +64,11 @@ Outils autorisés :
 - web_search
 - web_get
 - python_interpreter  # autorisé si besoin pour manipuler des listes/dicts
-
 Règles strictes :
 - Utilise python_interpreter uniquement pour manipuler des résultats (listes/dicts, filtrage, fusion).
 - N’écris pas de code inutile (pas de fichiers, pas de réseaux hors outils).
 - Pour (4) et (5), utilise le libellé utilisateur tel quel.
 - Retourne uniquement un JSON valide (objet), sans backticks.
-
 Pipeline :
 1) v = validate_ean(ean)
 2) off = openfoodfacts_product_by_ean(ean)         # si EAN vide ou invalide, passe à l’étape 4
@@ -86,14 +77,11 @@ Pipeline :
 5) sem = coicop_semantic_similarity(text=LIBELLÉ UTILISATEUR, topk=5)
 6) merged = merge_candidates(candidates_lists=[offmap, rx, sem], min_k=3, fallback_bias="cheese")
 7) res = resolve_coicop_candidates(json_lists=[merged], topn=3)
-
 → Retourne res tel quel (objet contenant final, alternatives, candidates_top).
 """
 
 
-
 def classify(label: str, ean: str):
-    import re
     label = (label or "").strip()
     ean = (ean or "").strip()
 
@@ -107,33 +95,13 @@ def classify(label: str, ean: str):
     buf_out, buf_err = io.StringIO(), io.StringIO()
     tee_out, tee_err = _Tee(sys.stdout, buf_out), _Tee(sys.stderr, buf_err)
 
-    # Utilitaire local : extraire un vrai JSON d'une chaîne (gère les ```json ... ```)
-    def _extract_json(text: str):
-        if isinstance(text, dict):
-            return text
-        if not isinstance(text, str):
-            return None
-        # 1) direct
-        try:
-            return json.loads(text)
-        except Exception:
-            pass
-        # 2) JSON dans des backticks
-        try:
-            m = re.search(r"```(?:json)?\s*(\{[\s\S]*\})\s*```", text)
-            if m:
-                return json.loads(m.group(1))
-        except Exception:
-            pass
-        return None
-
     print("\n===== Agent run start =====")
     print(task)
 
     try:
-        # Exécuter l’agent en dupliquant les logs vers l’UI et le Space
+        # Exécuter l’agent via la fonction tolérante aux timeouts et modèles de repli
         with redirect_stdout(tee_out), redirect_stderr(tee_err):
-            res = agent.run(task)
+            res = run_task_with_fallback(task)
 
         # Parsing robuste du résultat
         obj = None
@@ -143,7 +111,7 @@ def classify(label: str, ean: str):
             obj = _extract_json(res)
         if obj is None:
             try:
-                obj = parse_result(res)  # ton parseur existant
+                obj = parse_result(res)  # parseur du projet
             except Exception:
                 obj = None
         if obj is None:
@@ -163,19 +131,12 @@ def classify(label: str, ean: str):
         print(f"✖ Agent error: {e}")
         return json.dumps({"error": str(e)}, ensure_ascii=False, indent=2), logs_ui
 
-# ---------- instancie l'agent ----------
-try:
-    agent = build_agent()
-    out = agent.run(task)
-except Exception:
-    from quick_deploy_agent import run_task_with_fallback
-    out = run_task_with_fallback(task)
-
 
 def fill_example():
     # EAN réel OFF (Les p'tits crémeux – Aldi – 216 g)
     return "Camembert au lait cru AOP 250g - ALDI", "2006050033638"
 
+
 # ---------- thème & CSS (style IPC / Insee) ----------
 theme = gr.themes.Soft(
     primary_hue="blue",
@@ -204,13 +165,11 @@ custom_css = """
 }
 #app-header h1{ margin:0; font-size: 22px; letter-spacing: .3px; }
 #app-header p{ margin:6px 0 0; opacity:.92 }
-
 .card{
   background: #fff; border: 1px solid var(--insee-neutral-100); border-radius: var(--radius-lg);
   box-shadow: 0 8px 24px rgba(15, 23, 42, 0.06);
   padding: 18px;
 }
-
 .gr-textbox textarea{
   font-size: 16px; line-height: 1.35; padding: 14px 16px;
   border: 2px solid #e5e7eb; border-radius: 14px;
@@ -220,7 +179,6 @@ custom_css = """
   border-color: var(--insee-primary);
   box-shadow: 0 0 0 4px rgba(11,61,145,0.15);
 }
-
 .gr-button{
   font-weight: 700; letter-spacing: .2px; border-radius: 14px;
   transition: transform .06s ease, box-shadow .06s ease;
@@ -230,7 +188,6 @@ custom_css = """
   outline: 3px solid color-mix(in oklab, var(--insee-primary) 30%, white);
   box-shadow: 0 0 0 4px rgba(11,61,145,.18);
 }
-
 /* primaire */
 button.primary{
   background: linear-gradient(180deg, var(--insee-primary), var(--insee-primary-700));
@@ -243,7 +200,6 @@ button.ghost{
 button.ghost:hover{
   background: color-mix(in oklab, var(--insee-primary) 6%, white);
 }
-
 .gr-code pre, .gr-code code{
   font-size: 13.5px; line-height: 1.45; border-radius: 14px;
 }
@@ -285,4 +241,5 @@ with gr.Blocks(title="OpenFoodFactsAgent (COICOP)", theme=theme, css=custom_css)
     btn_ex.click(fn=fill_example, outputs=[label_in, ean_in])
 
 if __name__ == "__main__":
-    demo.launch()
+    # SSR est activé par défaut sur Spaces; ajuste si besoin:
+    demo.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=True)