Update app.py

app.py

@@ -1,18 +1,19 @@
 import streamlit as st
 from pptx import Presentation
-from pptx.util import Inches, Pt
+from pptx.util import Inches
 import io
-from huggingface_hub import InferenceClient
 import json
 import re
 import os
+import google.generativeai as genai
 
-
-gemini_token=os.getenv("GEMINI_TOKEN")
+# Configurar Gemini
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Asegúrate de que esté bien escrito
+genai.configure(api_key=GEMINI_API_KEY)
 
 @st.cache_resource
-def get_inference_client():
-    return InferenceClient("gemini-2.0-flash")
+def get_gemini_model():
+    return genai.GenerativeModel("gemini-1.5-flash")  # Puedes cambiar al modelo que desees
 
 def extract_and_clean_json(text):
     json_match = re.search(r'\{.*\}', text, re.DOTALL)
@@ -24,146 +25,101 @@ def extract_and_clean_json(text):
         return json_str
     return None
 
-def generate_presentation_content(topic, client, max_retries=3):
+def generate_presentation_content(topic, model, max_retries=3):
     prompt = f"""Genera una presentación de PowerPoint sobre el tema: "{topic}".
-    Debes crear exactamente 9 diapositivas. Cada diapositiva debe tener un título y contenido.
-    Es CRUCIAL que tu respuesta sea ÚNICAMENTE un JSON válido con la siguiente estructura exacta, sin texto adicional antes o después:
-
-    {{
-        "slides": [
-            {{"title": "Título de la diapositiva 1", "content": "Contenido de la diapositiva 1"}},
-            {{"title": "Título de la diapositiva 2", "content": "Contenido de la diapositiva 2"}},
-            {{"title": "Título de la diapositiva 3", "content": "Contenido de la diapositiva 3"}},
-            {{"title": "Título de la diapositiva 4", "content": "Contenido de la diapositiva 4"}},
-            {{"title": "Título de la diapositiva 5", "content": "Contenido de la diapositiva 5"}},
-            {{"title": "Título de la diapositiva 6", "content": "Contenido de la diapositiva 6"}},
-            {{"title": "Título de la diapositiva 7", "content": "Contenido de la diapositiva 7"}},
-            {{"title": "Título de la diapositiva 8", "content": "Contenido de la diapositiva 8"}},
-            {{"title": "Título de la diapositiva 9", "content": "Contenido de la diapositiva 9"}}
-        ]
-    }}
-
-    No incluyas ningún otro texto, explicación o saludo. Solo el JSON.
-    """
-    
+Debes crear exactamente 9 diapositivas. Cada diapositiva debe tener un título y contenido.
+La respuesta debe ser SOLO un JSON válido con la estructura:
+
+{{
+  "slides": [
+    {{"title": "Título 1", "content": "Contenido 1"}},
+    ...
+    {{"title": "Título 9", "content": "Contenido 9"}}
+  ]
+}}"""
+
     for attempt in range(max_retries):
-        response = client.text_generation(prompt, max_new_tokens=2000, temperature=0.7)
-        
+        response = model.generate_content(prompt)
+
         try:
-            json_str = extract_and_clean_json(response)
+            json_str = extract_and_clean_json(response.text)
             if json_str:
                 slides_data = json.loads(json_str)
                 if 'slides' in slides_data and len(slides_data['slides']) == 9:
                     return slides_data['slides']
-            
-            # Si llegamos aquí, el JSON no era válido o no tenía la estructura correcta
             raise ValueError("JSON inválido o estructura incorrecta")
-        
         except (json.JSONDecodeError, ValueError) as e:
-            if attempt == max_retries - 1:  # Último intento
-                st.error(f"Error al generar el contenido después de {max_retries} intentos: {str(e)}")
+            if attempt == max_retries - 1:
+                st.error(f"Error después de {max_retries} intentos: {str(e)}")
                 st.text("Última respuesta del modelo:")
-                st.code(response)
+                st.code(response.text)
                 return None
             else:
                 st.warning(f"Intento {attempt + 1} fallido. Reintentando...")
-    
+
     return None
 
 def create_powerpoint(slides, template_path):
     prs = Presentation(template_path)
-    
-    # Agregar diapositivas generadas
     for slide_data in slides:
         slide = prs.slides.add_slide(prs.slide_layouts[1])
-        
-        title_shape = slide.shapes.title
-        content_shape = slide.placeholders[1] if len(slide.placeholders) > 1 else None
-        
-        if title_shape:
-            title_shape.text = slide_data['title']
-        
-        if content_shape:
-            content_shape.text = slide_data['content']
+        if slide.shapes.title:
+            slide.shapes.title.text = slide_data['title']
+        if len(slide.placeholders) > 1:
+            slide.placeholders[1].text = slide_data['content']
         else:
-            # Si no hay un placeholder para el contenido, añadimos un cuadro de texto
-            left = Inches(0.5)
-            top = Inches(1.5)
-            width = Inches(9)
-            height = Inches(5)
-            txBox = slide.shapes.add_textbox(left, top, width, height)
-            tf = txBox.text_frame
-            tf.text = slide_data['content']
-    
-    # Agregar una diapositiva final con "Gracias"
-    slide = prs.slides.add_slide(prs.slide_layouts[1])
-    title_shape = slide.shapes.title
-    content_shape = slide.placeholders[1] if len(slide.placeholders) > 1 else None
-    
-    if title_shape:
-        title_shape.text = "Gracias"
-    
-    if content_shape:
-        content_shape.text = ""
-    else:
-        # Si no hay un placeholder para el contenido, añadimos un cuadro de texto
-        left = Inches(0.5)
-        top = Inches(1.5)
-        width = Inches(9)
-        height = Inches(5)
-        txBox = slide.shapes.add_textbox(left, top, width, height)
-        tf = txBox.text_frame
-        tf.text = ""
-    
+            textbox = slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(9), Inches(5))
+            textbox.text_frame.text = slide_data['content']
+
+    final_slide = prs.slides.add_slide(prs.slide_layouts[1])
+    if final_slide.shapes.title:
+        final_slide.shapes.title.text = "Gracias"
+
     pptx_buffer = io.BytesIO()
     prs.save(pptx_buffer)
     pptx_buffer.seek(0)
-    
     return pptx_buffer
 
 def main():
-    st.title("PowerPoint Mágico con el Buho")
-    
-    client = get_inference_client()
-    
+    st.title("PowerPoint Mágico con GEMINI")
+
+    model = get_gemini_model()
     topic = st.text_input("Por favor, ingrese el tema de la presentación:")
-    
-    # Opciones de plantillas
+
     template_options = {
         "Simple": "PLANTILLAS/Simple.pptx",
         "Corporativo": "PLANTILLAS/Corporativo.pptx",
         "Moderno": "PLANTILLAS/Moderno.pptx"
     }
-    
+
     selected_template = st.selectbox("Seleccione una plantilla", list(template_options.keys()))
-    
+
     if st.button("Generar Presentación"):
         if topic:
             try:
-                with st.spinner("Generando contenido de la presentación..."):
-                    slides = generate_presentation_content(topic, client)
-                
+                with st.spinner("Generando contenido..."):
+                    slides = generate_presentation_content(topic, model)
+
                 if slides:
                     template_path = template_options[selected_template]
                     if not os.path.exists(template_path):
                         st.error(f"No se encontró la plantilla: {template_path}")
                         return
-                    
-                    with st.spinner("Creando archivo PowerPoint..."):
+
+                    with st.spinner("Creando PowerPoint..."):
                         pptx_buffer = create_powerpoint(slides, template_path)
-                    
-                    st.success("Presentación generada con éxito!")
-                    
+
+                    st.success("¡Presentación generada!")
                     st.download_button(
                         label="Descargar Presentación",
                         data=pptx_buffer,
-                        file_name=f"{topic.replace(' ', '_')}_{selected_template}_presentacion.pptx",
+                        file_name=f"{topic.replace(' ', '_')}_{selected_template}.pptx",
                         mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
                     )
             except Exception as e:
-                st.error(f"Ocurrió un error al generar la presentación: {str(e)}")
+                st.error(f"Ocurrió un error: {str(e)}")
         else:
-            st.warning("Por favor, ingrese un tema para la presentación.")
+            st.warning("Por favor, ingrese un tema.")
 
 if __name__ == "__main__":
     main()