Update app.py

app.py

@@ -1,6 +1,5 @@
 #https://huggingface.co/spaces/MisterAI/GenDoc_05
-#143
-
+#app.py_143
 
 
 import os
@@ -15,6 +14,7 @@ import time
 
 # Configuration des modèles et PREPROMPT
 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"
@@ -163,8 +163,8 @@ class PresentationGenerator:
 
         return prs
 
-def generate_presentation_with_progress(text, text_model_name, temperature, max_tokens):
-    """Fonction principale de génération avec suivi de progression"""
+def generate_skeleton(text, text_model_name, temperature, max_tokens):
+    """Génère le squelette de la présentation"""
     try:
         start_time = time.time()
         generator = PresentationGenerator()
@@ -178,8 +178,23 @@ def generate_presentation_with_progress(text, text_model_name, temperature, max_
         full_prompt = PREPROMPT + "\n\n" + text
         generated_content = generator.generate_text(full_prompt, temperature, max_tokens)
 
+        execution_time = time.time() - start_time
+        status = f"Squelette généré avec succès en {execution_time:.2f} secondes!"
+
+        # Retourne le statut et le contenu généré
+        return status, generated_content, None
+
+    except Exception as e:
+        print(f"Erreur lors de la génération: {str(e)}")
+        return f"Erreur: {str(e)}", None, None
+
+def create_presentation_file(generated_content):
+    """Crée le fichier PowerPoint à partir du contenu généré"""
+    try:
+        generator = PresentationGenerator()
+
         # Création de la présentation
-        yield "Création de la présentation PowerPoint...", generated_content, None
+        yield "Création de la présentation PowerPoint...", None
         slides = generator.parse_presentation_content(generated_content)
         prs = generator.create_presentation(slides)
 
@@ -191,17 +206,13 @@ def generate_presentation_with_progress(text, text_model_name, temperature, max_
         if not os.path.exists(output_path):
             raise FileNotFoundError(f"Le fichier {output_path} n'a pas été créé correctement")
 
-        execution_time = time.time() - start_time
-        status = f"Présentation générée avec succès en {execution_time:.2f} secondes!"
-
-        # Retourne le statut, le contenu généré et le fichier
-        return status, generated_content, output_path
+        return output_path
 
     except Exception as e:
-        print(f"Erreur lors de la génération: {str(e)}")
-        return f"Erreur: {str(e)}", None, None
+        print(f"Erreur lors de la création du fichier: {str(e)}")
+        return None
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Glass()) as demo:
     gr.Markdown(
         """
         # Générateur de Présentations PowerPoint IA
@@ -241,7 +252,7 @@ with gr.Blocks() as demo:
             )
 
     with gr.Row():
-        generate_btn = gr.Button("Générer la présentation", variant="primary")
+        generate_skeleton_btn = gr.Button("Générer le Squelette de la Présentation", variant="primary")
 
     with gr.Row():
         with gr.Column():
@@ -254,12 +265,14 @@ with gr.Blocks() as demo:
                 lines=10,
                 show_copy_button=True
             )
+            create_presentation_btn = gr.Button("Créer Présentation")
             output_file = gr.File(
                 label="Présentation PowerPoint"
             )
+            progress_bar = gr.Progress(track_tqdm=True)
 
-    generate_btn.click(
-        fn=generate_presentation_with_progress,
+    generate_skeleton_btn.click(
+        fn=generate_skeleton,
         inputs=[
             input_text,
             text_model_choice,
@@ -269,7 +282,16 @@ with gr.Blocks() as demo:
         outputs=[
             status_output,
             generated_content,
-            output_file
+            create_presentation_btn
+        ]
+    )
+
+    create_presentation_btn.click(
+        fn=create_presentation_file,
+        inputs=generated_content,
+        outputs=[
+            output_file,
+            progress_bar
         ]
     )