Update app.py

app.py

@@ -4,26 +4,17 @@ import gradio as gr
 from pydantic import BaseModel
 from openai import OpenAI
 
-# ------------------------------------------------------------------
-# 1 — Initialisation OpenAI
-# ------------------------------------------------------------------
-
+# 1. Initialisation OpenAI
 client = OpenAI(
     api_key="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"
 )
 
-# ------------------------------------------------------------------
-# 2 — Définition du schéma pour la réponse structurée
-# ------------------------------------------------------------------
-
+# 2. Modèle Pydantic
 class UserInfo(BaseModel):
     reponse_client: str
     conseil_methodo: str
 
-# ------------------------------------------------------------------
-# 3 — Prompt système adapté à la vente Schmidt
-# ------------------------------------------------------------------
-
+# 3. Prompt système adapté Schmidt
 SYSTEM_PROMPT = """
 Tu es un prospect dans un jeu de rôle commercial pour une enseigne de cuisines haut de gamme comme Schmidt.
 Tu joues le rôle d’un client particulier, en recherche ou non d’une cuisine, avec un profil réaliste : parfois curieux, parfois pressé, hésitant, ou méfiant.
@@ -36,66 +27,61 @@ Ta réponse doit être un JSON structuré conforme au schéma suivant :
 
 {
   "reponse_client": "ta réponse réaliste au commercial",
-  "conseil_methodo": "un conseil ou retour constructif sur la méthode de vente utilisée (écoute active, découverte, reformulation, gestion des objections, closing, etc.). Tu peux faire référence à des méthodes comme CAB ou SONCAS."
+  "conseil_methodo": "un conseil ou retour constructif sur la méthode de vente utilisée (écoute active, découverte, reformulation, gestion des objections, closing, etc.). Tu peux faire référence à des méthodes."
 }
 
 Sois réaliste et varié dans tes réactions de client : n’hésite pas à poser des objections ou à exprimer des besoins flous.
-
-Exemples d’obstacles :
-- “Je veux juste un devis pour comparer.”
-- “On a un petit budget.”
-- “Je dois en parler à mon conjoint.”
-- “J’hésite avec Ikea.”
-
-Exemples de conseils :
-- “Bonne reformulation, cela permet de s’assurer de bien comprendre le besoin.”
-- “Tu aurais pu creuser le contexte de la demande avant de proposer une solution.”
-- “Attention, tu proposes trop tôt sans avoir découvert le projet.”
 """
 
-# ------------------------------------------------------------------
-# 4 — Fonction de jeu de rôle avec retour structuré
-# ------------------------------------------------------------------
-
-def generer_reponse_structuree(message: str) -> str:
+# 4. Fonction principale
+def generer_reponse_structuree(message: str) -> tuple[str, str]:
     try:
         response = client.responses.parse(
-            model="gpt-4o",  # utilise Structured Outputs
+            model="gpt-4o",
             input=[
                 {"role": "system", "content": SYSTEM_PROMPT},
                 {"role": "user", "content": message},
             ],
-            text_format=UserInfo,  # output structuré
+            text_format=UserInfo,
             temperature=0.7,
         )
 
         parsed = response.output_parsed
-        return json.dumps(parsed.model_dump(), indent=2, ensure_ascii=False)
+        return parsed.reponse_client, parsed.conseil_methodo
 
     except Exception as e:
-        return f"❌ Erreur : {e}"
-
-# ------------------------------------------------------------------
-# 5 — Interface Gradio
-# ------------------------------------------------------------------
+        return f"❌ Erreur : {e}", ""
 
+# 5. Interface Gradio UI améliorée
 exemple_commercial = "Bonjour, je suis conseiller chez Schmidt. Avez-vous déjà une idée de votre projet de cuisine ou souhaitez-vous qu’on en discute ensemble ?"
 
-with gr.Blocks(title="Jeu de rôle commercial Schmidt (B2C)") as demo:
+with gr.Blocks(title="Jeu de rôle commercial Schmidt (UI améliorée)") as demo:
     gr.Markdown("""
-    ## 🎯 Jeu de rôle commercial — Vente de cuisines Schmidt
+    # 🎯 Jeu de rôle commercial — Schmidt
 
-    Simule une interaction avec un client dans un contexte B2C. Reçois une **réponse réaliste** du prospect et un **retour méthodologique** sur ta posture commerciale.
+    👉 Tape une phrase comme si tu étais un conseiller Schmidt.  
+    🤖 L’IA joue le rôle d’un prospect et t’indique un **feedback méthodologique**.
+
+    ---
     """)
 
-    input_box = gr.Textbox(label="Ta phrase de vendeur", lines=3, value=exemple_commercial)
-    output_json = gr.Code(label="Réponse + Conseil structuré (JSON)")
+    with gr.Row():
+        input_box = gr.Textbox(
+            label="💬 Ce que tu dis au client",
+            placeholder="Ex : Vous cherchez plutôt une cuisine moderne ou traditionnelle ?",
+            value=exemple_commercial,
+            lines=2,
+            scale=2
+        )
+        bouton = gr.Button("Envoyer", scale=1)
 
-    gr.Button("Lancer le jeu de rôle").click(generer_reponse_structuree, input_box, output_json)
+    with gr.Row():
+        reponse_client = gr.Textbox(label="🗣️ Réponse du client (simulation)", interactive=False, lines=4)
+        conseil = gr.Textbox(label="📚 Conseil sur ta méthode commerciale", interactive=False, lines=3)
 
-# ------------------------------------------------------------------
-# 6 — Lancement
-# ------------------------------------------------------------------
+    bouton.click(generer_reponse_structuree, inputs=input_box, outputs=[reponse_client, conseil])
+    input_box.submit(generer_reponse_structuree, inputs=input_box, outputs=[reponse_client, conseil])
 
+# 6. Lancement
 if __name__ == "__main__":
     demo.launch()