Update app.py

app.py

@@ -13,8 +13,26 @@ api_key = "sk-proj-HYw82rX0O01CJrobRFj13r3BFxN3H1qOfswMSK1NGtGpfNyylWzMV_lQXw8Ee
 
 client = OpenAI(api_key=api_key)
 
+ :
+  export OPENAI_API_KEY=sk-...
+  pip install openai>=1.0.0 gradio
+"""
+
+import os
+import json
+from typing import Dict, Any
+
+import gradio as gr
+from openai import OpenAI
+
+# ------------------------------------------------------------------
+# 1 — Initialisation du client OpenAI
+# ------------------------------------------------------------------
+
+client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+
 # ------------------------------------------------------------------
-# 2 — Définition du schéma JSON (outil) à respecter
+# 2 — Définition du schéma JSON (outil)
 # ------------------------------------------------------------------
 
 extraction_schema: Dict[str, Any] = {
@@ -61,50 +79,64 @@ extraction_schema: Dict[str, Any] = {
     },
 }
 
-# Emballage dans la structure tool
 TOOLS = [{"type": "function", "function": extraction_schema}]
 
 SYSTEM_PROMPT = (
     "Tu es un assistant administratif. À partir d’un texte mal rédigé, "
     "tu dois extraire les informations personnelles sous forme de JSON "
-    "conforme au schéma fourni. Ne réponds JAMAIS hors JSON."
+    "conforme au schéma fourni. Ne réponds JAMAIS hors JSON. "
+    "Utilise IMPÉRATIVEMENT l'outil extract_user_info."
 )
 
 # ------------------------------------------------------------------
-# 3 — Fonction d'extraction « backend »
+# 3 — Fonction d'extraction avec relance forcée
 # ------------------------------------------------------------------
 
 def extraire_infos(texte: str) -> str:
-    """Appelle le modèle et renvoie un JSON indenté ou un message d'erreur."""
+    """Renvoie le JSON extrait ou une erreur si l’outil n’est jamais appelé."""
+
+    messages = [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "user", "content": texte},
+    ]
 
-    try:
-        # Première (et unique) requête : on force le choix de l'outil
+    for attempt in range(2):  # on tente deux fois maximum
         response = client.chat.completions.create(
-            model="gpt-4o-mini",  # ou "gpt-3.5-turbo-0125"
+            model="gpt-4o-mini",  # ou gpt-3.5-turbo-0125
             temperature=0,
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": texte},
-            ],
+            messages=messages,
             tools=TOOLS,
-            tool_choice={"type": "function", "function": {"name": "extract_user_info"}},
+            tool_choice={
+                "type": "function",
+                "function": {"name": "extract_user_info"},
+            },
         )
 
         choice = response.choices[0]
-        if choice.finish_reason != "tool_calls":
-            # Le modèle n'a pas utilisé l'outil : on renvoie le contenu brut
-            return choice.message.content or "(Aucun contenu renvoyé)"
-
-        # Récupère les arguments du premier tool_call
-        args_str = choice.message.tool_calls[0].function.arguments
-        parsed = json.loads(args_str)
-        return json.dumps(parsed, indent=2, ensure_ascii=False)
+        if choice.finish_reason == "tool_calls":
+            args_str = choice.message.tool_calls[0].function.arguments
+            parsed = json.loads(args_str)
+            return json.dumps(parsed, indent=2, ensure_ascii=False)
+
+        # Pas d'appel d'outil : on renforce l'instruction et on réessaie
+        messages.insert(
+            0,
+            {
+                "role": "system",
+                "content": (
+                    "⚠️ Utilise OBLIGATOIREMENT l'outil extract_user_info et "
+                    "ne renvoie jamais de texte libre."
+                ),
+            },
+        )
 
-    except Exception as e:
-        return f"❌ Erreur : {e}"
+    return (
+        "❌ Erreur : le modèle n'a pas renvoyé d'appel d'outil après deux "
+        "tentatives."
+    )
 
 # ------------------------------------------------------------------
-# 4 — Interface Gradio
+# 4 — Interface Gradio
 # ------------------------------------------------------------------
 
 texte_exemple = (
@@ -117,7 +149,10 @@ texte_exemple = (
 )
 
 with gr.Blocks(title="Extracteur intelligent de données CERFA") as demo:
-    gr.Markdown("Copie‑colle un texte mal écrit contenant des infos personnelles.\n" "L'IA le transforme en JSON prêt pour l'administration.")
+    gr.Markdown(
+        "Copie-colle un texte mal écrit contenant des infos personnelles.\n"
+        "L'IA le transforme en JSON prêt pour l'administration."
+    )
 
     input_box = gr.Textbox(
         lines=14,
@@ -131,8 +166,9 @@ with gr.Blocks(title="Extracteur intelligent de données CERFA") as demo:
     extract_btn.click(extraire_infos, input_box, output_box)
 
 # ------------------------------------------------------------------
-# 5 — Lancement
+# 5 — Lancement
 # ------------------------------------------------------------------
 
 if __name__ == "__main__":
     demo.launch()
+