Update app.py

app.py

@@ -9,12 +9,15 @@ openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 from mistralai import Mistral
 mistral_client = Mistral(api_key=os.getenv("MISTRAL_API_KEY"))
 
-# Choisis à l’exécution quel LLM utiliser
+# Choix par défaut
 DEFAULT_PROVIDER = "openai"  # ou "mistral"
 
+
 def llm_chat(messages, max_tokens, temperature, top_p, provider=DEFAULT_PROVIDER):
+    provider = (provider or "").strip().lower()
+
     if provider == "openai":
-        # Chat Completions (ex: gpt-4o-mini / gpt-4.1 / … selon ton compte)
+        # Chat Completions OpenAI
         stream = openai_client.chat.completions.create(
             model="gpt-4o-mini",
             messages=messages,
@@ -29,7 +32,7 @@ def llm_chat(messages, max_tokens, temperature, top_p, provider=DEFAULT_PROVIDER
                 yield delta.content
 
     elif provider == "mistral":
-        # Models: "mistral-large-latest", "open-mistral-nemo", etc.
+        # Chat Completions Mistral
         stream = mistral_client.chat.stream(
             model="mistral-large-latest",
             messages=messages,
@@ -45,7 +48,7 @@ def llm_chat(messages, max_tokens, temperature, top_p, provider=DEFAULT_PROVIDER
         stream.close()
 
     else:
-        yield "[Erreur] Provider inconnu."
+        yield "[Erreur] Provider inconnu (utilise 'openai' ou 'mistral')."
 
 
 def respond(
@@ -55,18 +58,25 @@ def respond(
     max_tokens,
     temperature,
     top_p,
-    provider  # "openai" ou "mistral"
+    provider,  # "openai" ou "mistral"
 ):
-    # Construit le format messages attendu par les SDKs
+    # Construit le format messages attendu
     messages = [{"role": "system", "content": system_message}]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for token in llm_chat(messages, provider, max_tokens, temperature, top_p):
+    for token in llm_chat(
+        messages=messages,
+        max_tokens=max_tokens,
+        temperature=temperature,
+        top_p=top_p,
+        provider=provider,
+    ):
         response += token
         yield response
 
+
 chatbot = gr.ChatInterface(
     respond,
     type="messages",