Fix: add text parameter to image-text-to-text pipeline

app.py

@@ -6,10 +6,8 @@ import time
 print("📥 Chargement du modèle léger (CPU optimisé)...")
 start = time.time()
 
-# Modèle BLIP-base (990M paramètres) - bon compromis qualité/vitesse
-# Alternative plus rapide : "nlpconnect/vit-gpt2-image-captioning"
 captioner = pipeline(
-    "image-text-to-text",  # Changement ici : image-to-text → image-text-to-text
+    "image-text-to-text",
     model="Salesforce/blip-image-captioning-base",
     device=-1
 )
@@ -19,25 +17,37 @@ print(f"✅ Modèle chargé en {time.time() - start:.1f} secondes")
 def generate_caption(image, prompt_text=""):
     """Génère une légende pour l'image"""
     try:
-        # Convertir l'image PIL
         if hasattr(image, 'size'):
             pil_image = image
         else:
             pil_image = Image.fromarray(image.astype('uint8'))
         
-        # Générer la légende (max 50 tokens pour rester rapide)
-        result = captioner(pil_image, max_new_tokens=50)
-        caption = result[0]['generated_text']
-        
-        # Ajouter le prompt personnalisé si fourni
+        # Texte par défaut si aucun prompt fourni
         if prompt_text and prompt_text.strip():
-            return f"{prompt_text} {caption}"
+            user_prompt = prompt_text
+        else:
+            user_prompt = "Describe this image in a short sentence:"
+        
+        # Appel avec image ET texte
+        result = captioner(
+            pil_image,
+            text=user_prompt,
+            max_new_tokens=50
+        )
+        
+        # Extraire la réponse
+        if isinstance(result, list) and len(result) > 0:
+            if 'generated_text' in result[0]:
+                caption = result[0]['generated_text']
+            else:
+                caption = str(result[0])
+        else:
+            caption = str(result)
         
-        return caption
+        return caption.strip()
     except Exception as e:
         return f"Erreur : {str(e)}"
 
-# Interface simple et efficace
 with gr.Blocks(title="Générateur de titres d'images (CPU optimisé)") as demo:
     gr.Markdown("""
     # 🖼️ Générateur de titres d'images
@@ -51,7 +61,7 @@ with gr.Blocks(title="Générateur de titres d'images (CPU optimisé)") as demo:
             image_input = gr.Image(label="📸 Uploadez votre image", type="pil")
             prompt = gr.Textbox(
                 label="Style de titre (optionnel)",
-                placeholder="Exemple: 'Titre court :' ou 'Description :'",
+                placeholder="Exemple: 'Titre court :' ou 'Description détaillée :'",
                 lines=1
             )
             btn = gr.Button("🚀 Générer le titre", variant="primary")