Update app.py

app.py

@@ -17,8 +17,17 @@ client = OpenAI(
     api_key=HF_TOKEN,
 )
 
-# 3) Modèle supporté par HF Inference (gratuit dans la limite de la free tier)
-MODEL_ID = "katanemo/Arch-Router-1.5B:hf-inference"
+# 3) Liste de modèles à essayer, dans l'ordre : le suffixe ":fastest" laisse
+# Hugging Face choisir automatiquement un fournisseur disponible pour ce
+# modèle, plutôt que de figer un fournisseur précis qui peut un jour cesser
+# de le proposer (c'est exactement ce qui a cassé le modèle précédent).
+# Si le premier modèle de la liste n'est plus disponible un jour, le code
+# essaie automatiquement le suivant au lieu d'afficher une erreur brute.
+MODEL_IDS = [
+    "Qwen/Qwen2.5-7B-Instruct:fastest",
+    "meta-llama/Llama-3.1-8B-Instruct:fastest",
+    "mistralai/Mistral-7B-Instruct-v0.3:fastest",
+]
 
 SYSTEM_PROMPT = """
 Tu es ArtGPT, un assistant spécialisé à 100% dans l'art :
@@ -27,7 +36,6 @@ Tu es ArtGPT, un assistant spécialisé à 100% dans l'art :
 - histoire de l'art, mouvements artistiques,
 - direction artistique, portfolio,
 - conseils logiciels et création visuelle.
-
 Règles :
 - Tu réponds uniquement en français.
 - Si la question n’est pas liée à l’art ou à la création visuelle,
@@ -41,24 +49,33 @@ def answer(question: str) -> str:
     if not question:
         return "Écris une question sur l’art pour commencer 🙂"
 
-    try:
-        completion = client.chat.completions.create(
-            model=MODEL_ID,
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": question},
-            ],
-            max_tokens=512,
-            temperature=0.7,
-            top_p=0.95,
-        )
+    last_error = None
 
-        # Avec la nouvelle lib openai, le contenu est dans .message.content
-        return completion.choices[0].message.content
+    for model_id in MODEL_IDS:
+        try:
+            completion = client.chat.completions.create(
+                model=model_id,
+                messages=[
+                    {"role": "system", "content": SYSTEM_PROMPT},
+                    {"role": "user", "content": question},
+                ],
+                max_tokens=512,
+                temperature=0.7,
+                top_p=0.95,
+            )
+            # Avec la nouvelle lib openai, le contenu est dans .message.content
+            return completion.choices[0].message.content
+        except Exception as e:
+            # On garde l'erreur de côté et on essaie le modèle suivant de la liste.
+            last_error = e
+            continue
 
-    except Exception as e:
-        # On affiche une erreur lisible au lieu du gros JSON
-        return f"Erreur lors de l'appel au modèle :\n\n{e}"
+    # Si tous les modèles de la liste ont échoué, on affiche une erreur lisible
+    # au lieu du gros JSON brut.
+    return (
+        "Erreur lors de l'appel au modèle (tous les modèles de secours ont "
+        f"échoué) :\n\n{last_error}"
+    )
 
 demo = gr.Interface(
     fn=answer,