Update app.py

app.py

@@ -4,14 +4,14 @@ from pptx.util import Inches
 import requests
 from PIL import Image
 from io import BytesIO
+import io
 import os
 import json
 import re
 import google.generativeai as genai
 
-
 # Configurar Gemini
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Asegúrate de que esté bien escrito
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)
 
 # Obtener la clave API de Pixabay desde los secretos de Streamlit
@@ -19,7 +19,7 @@ PIXABAY_API_KEY = st.secrets["pixabay"]
 
 @st.cache_resource
 def get_gemini_model():
-    return genai.GenerativeModel("gemini-1.5-flash")  # Puedes cambiar al modelo que d
+    return genai.GenerativeModel("gemini-1.5-flash")
 
 def extract_and_clean_json(text):
     json_match = re.search(r'\{.*\}', text, re.DOTALL)
@@ -54,27 +54,22 @@ def generate_presentation_content(topic, client, max_retries=3):
     """
     
     for attempt in range(max_retries):
-        response = client.text_generation(prompt, max_new_tokens=2000, temperature=0.7)
-        
         try:
+            response = client.generate_content(prompt).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:  # Último intento
+            if attempt == max_retries - 1:
                 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)
                 return None
             else:
                 st.warning(f"Intento {attempt + 1} fallido. Reintentando...")
-    
     return None
 
 def buscar_imagen_pixabay(query):
@@ -107,7 +102,7 @@ def create_powerpoint(slides, template_path):
 
         # Buscar e insertar imagen
         try:
-            img = buscar_imagen_pixabay(slide_data['title'])  # usar el título como consulta
+            img = buscar_imagen_pixabay(slide_data['title'])
             if img:
                 img_path = "/tmp/temp_img.jpg"
                 img.save(img_path)
@@ -125,38 +120,37 @@ def create_powerpoint(slides, template_path):
     return pptx_buffer
 
 def main():
-    st.title("PowerPoint Mágico con el Buho")
-    
-    client = get_inference_client()
-    
+    st.title("PowerPoint Mágico con el Búho")
+
+    client = 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)
-                
+
                 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..."):
                         pptx_buffer = create_powerpoint(slides, template_path)
-                    
-                    st.success("Presentación generada con éxito!")
-                    
+
+                    st.success("¡Presentación generada con éxito!")
+
                     st.download_button(
                         label="Descargar Presentación",
                         data=pptx_buffer,