Update app.py

app.py

@@ -2,7 +2,6 @@
 #app.py_144
 #Uniquement Granite 3b instruct
 
-
 import os
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -11,14 +10,6 @@ from pptx.util import Inches, Pt
 import torch
 import time
 
-#TEXT_MODELS = {
-#    "Utter-Project_EuroLLM-1.7B": "utter-project/EuroLLM-1.7B",
-#    "Mistral Nemo 2407 (GGUF)": "MisterAI/Bartowski_MistralAI_Mistral-Nemo-Instruct-2407-IQ4_XS.gguf",
-##    "Mixtral 8x7B": "mistralai/Mixtral-8x7B-v0.1",
-#    "Lucie 7B": "OpenLLM-France/Lucie-7B"
-#}
-
-
 # Configuration du modèle unique
 MODEL_PATH = "ibm-granite/granite-3.1-3b-a800m-Instruct"
 
@@ -79,16 +70,14 @@ class PresentationGenerator:
         self.tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
         self.model = AutoModelForCausalLM.from_pretrained(
             MODEL_PATH,
-            torch_dtype=torch.float32,  # Utilisation de float32 pour plus de stabilité
-            device_map="auto"  # Laisse le modèle choisir le meilleur device
+            torch_dtype=torch.float32,
+            device_map="auto"
         )
-        self.model.eval()  # Mode évaluation
+        self.model.eval()
         print("Modèle initialisé avec succès!")
 
     def generate_text(self, prompt, temperature=0.7, max_tokens=2048):
-        """Génère le texte de la présentation selon le format Granite"""
         try:
-            # Formatage du chat selon les instructions Granite
             chat = [{"role": "user", "content": prompt}]
             formatted_prompt = self.tokenizer.apply_chat_template(
                 chat,
@@ -96,7 +85,6 @@ class PresentationGenerator:
                 add_generation_prompt=True
             )
 
-            # Tokenisation et génération
             inputs = self.tokenizer(
                 formatted_prompt,
                 return_tensors="pt",
@@ -120,7 +108,6 @@ class PresentationGenerator:
             raise
 
     def parse_presentation_content(self, content):
-        """Parse le contenu généré en sections pour les diapositives"""
         slides = []
         current_slide = None
         
@@ -143,14 +130,11 @@ class PresentationGenerator:
         return slides
 
     def create_presentation(self, slides):
-        """Crée la présentation PowerPoint"""
         prs = Presentation()
         
-        # Première diapo (titre)
         title_slide = prs.slides.add_slide(prs.slide_layouts[0])
         title_slide.shapes.title.text = slides[0]['title']
         
-        # Autres diapos
         for slide in slides[1:]:
             content_slide = prs.slides.add_slide(prs.slide_layouts[1])
             content_slide.shapes.title.text = slide['title']
@@ -168,21 +152,19 @@ class PresentationGenerator:
 # Timer global pour le suivi du temps
 timer = ExecutionTimer()
 
-def update_status():
-    return timer.get_status()
-
-def generate_skeleton(text, temperature, max_tokens, status_output):
+def generate_skeleton(text, temperature, max_tokens):
     """Génère le squelette de la présentation"""
     try:
         timer.start()
         generator = PresentationGenerator()
         
-        # Génération du contenu
         full_prompt = PREPROMPT + "\n\n" + text
         generated_content = generator.generate_text(full_prompt, temperature, max_tokens)
         
+        status = timer.get_status()
         timer.stop()
-        return timer.get_status(), generated_content, gr.update(visible=True)
+        
+        return status, generated_content, gr.update(visible=True)
         
     except Exception as e:
         timer.stop()
@@ -196,7 +178,6 @@ def create_presentation_file(generated_content):
         timer.start()
         generator = PresentationGenerator()
         
-        # Création et sauvegarde de la présentation
         slides = generator.parse_presentation_content(generated_content)
         prs = generator.create_presentation(slides)
         
@@ -254,34 +235,25 @@ with gr.Blocks(theme=gr.themes.Glass()) as demo:
             status_output = gr.Textbox(
                 label="Statut",
                 lines=2,
-                every=1  # Met à jour toutes les secondes
+                value="En attente..."
             )
             generated_content = gr.Textbox(
                 label="Contenu généré",
                 lines=10,
                 show_copy_button=True
             )
-            create_presentation_btn = gr.Button("Créer La Présentation", visible=False)
+            create_presentation_btn = gr.Button("Créer Présentation", visible=False)
             output_file = gr.File(
                 label="Présentation PowerPoint",
-                type="filepath"  # Utilise filepath au lieu de binary
+                type="filepath"
             )
     
-    # Mise à jour du statut en temps réel
-    status_output.change(
-        fn=update_status,
-        inputs=[],
-        outputs=[status_output],
-        every=1
-    )
-    
     generate_skeleton_btn.click(
         fn=generate_skeleton,
         inputs=[
             input_text,
             temperature,
-            max_tokens,
-            status_output
+            max_tokens
         ],
         outputs=[
             status_output,
@@ -304,4 +276,3 @@ if __name__ == "__main__":
 
 
 
-