Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from huggingface_hub import hf_hub_download, login
-from transformers import AutoModelForCausalLM, AutoTokenizer, AutoPipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from pptx import Presentation
 from pptx.util import Inches, Pt
 from pptx.enum.text import PP_ALIGN
@@ -36,7 +36,10 @@ Points:
 - Point 1
 - Point 2
 - Point 3
-Image: [Description détaillée de l'image souhaitée pour cette diapo]
+Image: [Description détaillée de l'image souhaitée pour cette diapo. Soyez très précis dans la description pour permettre 
+une génération d'image de qualité. Par exemple : "Une illustration professionnelle montrant un concept clé de cybersécurité 
+avec des éléments visuels modernes, un style épuré et des couleurs corporate (fond noir, couleurs bleu electrique, rouge, gris, blanc). 
+L'image doit être claire, minimaliste et adaptée à une présentation professionnelle."]
 
 DIAPO 2:
 Titre: [Titre de la diapo]
@@ -89,8 +92,9 @@ class PresentationGenerator:
     def load_image_model(self, model_name):
         """Charge le modèle de génération d'images"""
         model_id = IMAGE_MODELS[model_name]
-        self.image_pipeline = AutoPipeline.from_pretrained(
-            model_id,
+        self.image_pipeline = pipeline(
+            "text-to-image",
+            model=model_id,
             token=self.token
         )
 
@@ -124,7 +128,7 @@ class PresentationGenerator:
                 prompt=prompt,
                 negative_prompt=negative_prompt,
                 num_inference_steps=num_inference_steps
-            ).images[0]
+            )[0]  # Pipeline retourne une liste d'images, on prend la première
             return image
         except Exception as e:
             print(f"Erreur lors de la génération de l'image: {str(e)}")
@@ -228,7 +232,7 @@ def generate_presentation_with_progress(text, text_model_name, image_model_name,
     except Exception as e:
         return f"Erreur: {str(e)}", None, None
 
-# Interface Gradio améliorée
+# CSS personnalisé pour un thème sombre amélioré
 css = """
 /* Thème sombre personnalisé */
 .gradio-container {
@@ -375,4 +379,4 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-
+