Update app.py

app.py

@@ -1,19 +1,20 @@
 import streamlit as st
 from pptx import Presentation
 from pptx.util import Inches
-import io
+import requests
+from PIL import Image
+from io import BytesIO
+import os
 import json
 import re
-import os
-import google.generativeai as genai
+from huggingface_hub import InferenceClient
 
-# Configurar Gemini
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Asegúrate de que esté bien escrito
-genai.configure(api_key=GEMINI_API_KEY)
+# Obtener la clave API de Pixabay desde los secretos de Streamlit
+PIXABAY_API_KEY = st.secrets["pixabay"]["api_key"]
 
 @st.cache_resource
-def get_gemini_model():
-    return genai.GenerativeModel("gemini-1.5-flash")  # Puedes cambiar al modelo que desees
+def get_inference_client():
+    return InferenceClient("gemini-2.0-flash")
 
 def extract_and_clean_json(text):
     json_match = re.search(r'\{.*\}', text, re.DOTALL)
@@ -25,101 +26,142 @@ def extract_and_clean_json(text):
         return json_str
     return None
 
-def generate_presentation_content(topic, model, max_retries=3):
+def generate_presentation_content(topic, client, 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.
-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"}}
-  ]
-}}"""
-
+    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.
+    """
+    
     for attempt in range(max_retries):
-        response = model.generate_content(prompt)
-
+        response = client.text_generation(prompt, max_new_tokens=2000, temperature=0.7)
+        
         try:
-            json_str = extract_and_clean_json(response.text)
+            json_str = extract_and_clean_json(response)
             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:
-                st.error(f"Error después de {max_retries} intentos: {str(e)}")
+            if attempt == max_retries - 1:  # Último intento
+                st.error(f"Error al generar el contenido después de {max_retries} intentos: {str(e)}")
                 st.text("Última respuesta del modelo:")
-                st.code(response.text)
+                st.code(response)
                 return None
             else:
                 st.warning(f"Intento {attempt + 1} fallido. Reintentando...")
+    
+    return None
+
+def buscar_imagen_pixabay(query):
+    url = f"https://pixabay.com/api/?key={PIXABAY_API_KEY}&q={requests.utils.quote(query)}&image_type=photo&per_page=10"
+    response = requests.get(url)
+    data = response.json()
 
+    if data.get('hits'):
+        image_url = data['hits'][0]['webformatURL']
+        image_response = requests.get(image_url)
+        return Image.open(BytesIO(image_response.content))
     return None
 
 def create_powerpoint(slides, template_path):
     prs = Presentation(template_path)
+    
     for slide_data in slides:
         slide = prs.slides.add_slide(prs.slide_layouts[1])
-        if slide.shapes.title:
-            slide.shapes.title.text = slide_data['title']
-        if len(slide.placeholders) > 1:
-            slide.placeholders[1].text = slide_data['content']
+        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']
         else:
-            textbox = slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(9), Inches(5))
-            textbox.text_frame.text = slide_data['content']
+            txBox = slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(9), Inches(5))
+            tf = txBox.text_frame
+            tf.text = slide_data['content']
 
+        # Buscar e insertar imagen
+        try:
+            img = buscar_imagen_pixabay(slide_data['title'])  # usar el título como consulta
+            if img:
+                img_path = "/tmp/temp_img.jpg"
+                img.save(img_path)
+                slide.shapes.add_picture(img_path, Inches(5), Inches(1.5), width=Inches(4))
+        except Exception as e:
+            print(f"No se pudo insertar imagen: {e}")
+    
+    # Agregar diapositiva final
     final_slide = prs.slides.add_slide(prs.slide_layouts[1])
-    if final_slide.shapes.title:
-        final_slide.shapes.title.text = "Gracias"
-
+    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 GEMINI")
-
-    model = get_gemini_model()
+    st.title("PowerPoint Mágico con el Buho")
+    
+    client = get_inference_client()
+    
     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..."):
-                    slides = generate_presentation_content(topic, model)
-
+                with st.spinner("Generando contenido de la presentación..."):
+                    slides = generate_presentation_content(topic, client)
+                
                 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 PowerPoint..."):
+                    
+                    with st.spinner("Creando archivo PowerPoint..."):
                         pptx_buffer = create_powerpoint(slides, template_path)
-
-                    st.success("¡Presentación generada!")
+                    
+                    st.success("Presentación generada con éxito!")
+                    
                     st.download_button(
                         label="Descargar Presentación",
                         data=pptx_buffer,
-                        file_name=f"{topic.replace(' ', '_')}_{selected_template}.pptx",
+                        file_name=f"{topic.replace(' ', '_')}_{selected_template}_presentacion.pptx",
                         mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
                     )
             except Exception as e:
-                st.error(f"Ocurrió un error: {str(e)}")
+                st.error(f"Ocurrió un error al generar la presentación: {str(e)}")
         else:
-            st.warning("Por favor, ingrese un tema.")
+            st.warning("Por favor, ingrese un tema para la presentación.")
 
 if __name__ == "__main__":
     main()