fix utils complete

utils.py

@@ -1,11 +1,41 @@
 import os
 import re
+import json
 
 from openai import OpenAI
 from groq import Groq
 import anthropic
 
 
+# =====================================================
+# JSON HELPERS
+# =====================================================
+
+def clean_json_text(text: str) -> str:
+
+    if not text:
+        return ""
+
+    text = text.strip()
+
+    # retire ```json
+    text = re.sub(r"^```json", "", text)
+    text = re.sub(r"^```", "", text)
+    text = re.sub(r"```$", "", text)
+
+    return text.strip()
+
+
+def safe_json_loads(text: str):
+
+    try:
+        cleaned = clean_json_text(text)
+        return json.loads(cleaned)
+
+    except Exception:
+        return None
+
+
 # =====================================================
 # DEEPSEEK
 # =====================================================
@@ -130,25 +160,31 @@ def call_llm(
     max_tokens: int = 2000
 ):
 
+    provider = provider.lower()
+
     if provider == "deepseek":
+
         return call_deepseek(
             prompt,
             max_tokens=max_tokens
         )
 
     elif provider == "groq":
+
         return call_groq(
             prompt,
             max_tokens=max_tokens
         )
 
     elif provider == "claude":
+
         return call_claude(
             prompt,
             max_tokens=max_tokens
         )
 
     else:
+
         return "ERREUR: provider inconnu"
 
 
@@ -160,7 +196,8 @@ def smart_call(prompt: str):
 
     providers = [
         "deepseek",
-        "groq"
+        "groq",
+        "claude"
     ]
 
     for provider in providers:
@@ -170,21 +207,28 @@ def smart_call(prompt: str):
             provider=provider
         )
 
-        if not result.startswith("ERREUR"):
+        if (
+            isinstance(result, str)
+            and not result.startswith("ERREUR")
+        ):
             return result
 
     return "ERREUR: tous les providers ont échoué"
 
 
 # =====================================================
-# EXTRACTION LATEX
+# EXTRACTION LATEX SIMPLE
 # =====================================================
 
 def extract_latex_blocks(text: str):
 
     pattern = r'\$\$([^\$]+)\$\$|\\\[(.*?)\\\]|\$([^\$]+)\$'
 
-    matches = re.findall(pattern, text, re.DOTALL)
+    matches = re.findall(
+        pattern,
+        text,
+        re.DOTALL
+    )
 
     equations = []
 
@@ -195,23 +239,30 @@ def extract_latex_blocks(text: str):
         if eq and eq.strip():
             equations.append(eq.strip())
 
-    seen = set()
-
+    # remove duplicates
     unique = []
 
+    seen = set()
+
     for eq in equations:
 
         if eq not in seen:
+
             seen.add(eq)
+
             unique.append(eq)
 
     return unique
 
+
 # =====================================================
-# EXTRACTION EQUATIONS VIA LLM
+# EXTRACTION EQUATIONS AVEC LLM
 # =====================================================
 
-def extract_equations_with_llm(text: str):
+def extract_equations_with_llm(
+    text: str,
+    provider: str = "deepseek"
+):
 
     prompt = f"""
 Tu es un assistant scientifique.
@@ -219,12 +270,97 @@ Tu es un assistant scientifique.
 Extrais uniquement les équations mathématiques
 présentes dans ce texte.
 
-Retourne uniquement une liste brute.
+Retourne STRICTEMENT un JSON valide
+sous cette forme :
+
+[
+  {{
+    "latex": "E = mc^2"
+  }}
+]
 
 Texte :
 {text}
 """
 
-    response = smart_call(prompt)
+    response = call_llm(
+        prompt,
+        provider=provider,
+        max_tokens=2000
+    )
+
+    data = safe_json_loads(response)
+
+    if data is None:
+        return []
+
+    return data
+
+
+# =====================================================
+# VALIDATION IR
+# =====================================================
+
+def validate_ir(ir):
+
+    if not isinstance(ir, dict):
+        return False
+
+    if "nodes" not in ir:
+        return False
+
+    if "edges" not in ir:
+        return False
+
+    return True
+
+
+# =====================================================
+# NORMALISATION IR
+# =====================================================
+
+def normalize_ir(ir):
+
+    if not isinstance(ir, dict):
+        return {
+            "name": "Invalid IR",
+            "strategy": "fallback",
+            "nodes": [],
+            "edges": []
+        }
+
+    ir.setdefault("name", "Unnamed IR")
+    ir.setdefault("strategy", "fallback")
+    ir.setdefault("nodes", [])
+    ir.setdefault("edges", [])
+
+    return ir
+
+
+# =====================================================
+# GENERATION IR FALLBACK
+# =====================================================
+
+def build_fallback_ir(equations):
+
+    nodes = []
+
+    for i, eq in enumerate(equations):
+
+        if isinstance(eq, dict):
+            latex = eq.get("latex", "")
+        else:
+            latex = str(eq)
+
+        nodes.append({
+            "id": f"eq_{i}",
+            "type": "equation",
+            "latex": latex
+        })
 
-    return extract_latex_blocks(response)
+    return {
+        "name": "Fallback Variant 1",
+        "strategy": "fallback",
+        "nodes": nodes,
+        "edges": []
+    }