Update app.py

app.py

@@ -1,8 +1,17 @@
+import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from pptx import Presentation
 from pptx.util import Inches, Pt
 import os
+import io
+
+@st.cache_resource
+def load_model():
+    model_name = "meta-llama/Llama-2-7b-chat-hf"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+    return tokenizer, model
 
 def generate_presentation_content(topic, model, tokenizer):
     prompt = f"Crea una presentación de PowerPoint sobre el tema: {topic}. Incluye 5 diapositivas con títulos y contenido."
@@ -14,7 +23,7 @@ def generate_presentation_content(topic, model, tokenizer):
     slides = generated_text.split("\n\n")
     return slides[:5]  # Limitar a 5 diapositivas
 
-def create_powerpoint(slides, output_file):
+def create_powerpoint(slides):
     prs = Presentation()
     
     for slide_content in slides:
@@ -30,24 +39,40 @@ def create_powerpoint(slides, output_file):
         title_shape.text = title
         content_shape.text = content
     
-    prs.save(output_file)
+    # Guardar la presentación en un buffer de bytes
+    pptx_buffer = io.BytesIO()
+    prs.save(pptx_buffer)
+    pptx_buffer.seek(0)
+    
+    return pptx_buffer
 
 def main():
-    print("Bienvenido al generador de presentaciones PowerPoint con IA")
-    topic = input("Por favor, ingrese el tema de la presentación: ")
-    
-    model_name = "meta-llama/Llama-2-7b-chat-hf"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
-    
-    print("Generando contenido de la presentación...")
-    slides = generate_presentation_content(topic, model, tokenizer)
+    st.title("Generador de presentaciones PowerPoint con IA")
     
-    output_file = f"{topic.replace(' ', '_')}_presentacion.pptx"
-    print(f"Creando archivo PowerPoint: {output_file}")
-    create_powerpoint(slides, output_file)
+    topic = st.text_input("Por favor, ingrese el tema de la presentación:")
     
-    print(f"Presentación generada y guardada como {output_file}")
+    if st.button("Generar Presentación"):
+        if topic:
+            with st.spinner("Cargando modelo..."):
+                tokenizer, model = load_model()
+            
+            with st.spinner("Generando contenido de la presentación..."):
+                slides = generate_presentation_content(topic, model, tokenizer)
+            
+            with st.spinner("Creando archivo PowerPoint..."):
+                pptx_buffer = create_powerpoint(slides)
+            
+            st.success("Presentación generada con éxito!")
+            
+            # Ofrecer la descarga del archivo
+            st.download_button(
+                label="Descargar Presentación",
+                data=pptx_buffer,
+                file_name=f"{topic.replace(' ', '_')}_presentacion.pptx",
+                mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
+            )
+        else:
+            st.warning("Por favor, ingrese un tema para la presentación.")
 
 if __name__ == "__main__":
     main()