Update app.py

app.py

@@ -50,52 +50,11 @@ def extract_text_from_pdf(pdf_file):
     except Exception as e:
         return f"Se produjo un error al leer el PDF: {e}"
 
-# Función para generar una imagen a partir de una descripción
-def generate_image(description):
-    try:
-        client = genai.Client(api_key=GEMINI_API_KEY)
-
-        # Suponiendo que la API de Gemini tiene un endpoint para generar imágenes
-        model = "gemini-image-model"  # Suponiendo que este es el modelo correcto
-        contents = [
-            types.Content(
-                role="user",
-                parts=[types.Part.from_text(text=description)],
-            ),
-        ]
-        
-        # Configuración de la generación de imágenes, puedes ajustar estos parámetros
-        generate_content_config = types.GenerateContentConfig(
-            temperature=0.7,
-            top_p=0.95,
-            top_k=64,
-            max_output_tokens=2048,  # Reducir los tokens si es necesario
-            response_mime_type="image/png",  # Ajustar a "image/jpeg" si la API lo requiere
-        )
-        
-        # Suponiendo que la API devuelva una URL de la imagen
-        response_image_url = ""
-        for chunk in client.models.generate_content_stream(
-            model=model,
-            contents=contents,
-            config=generate_content_config,
-        ):
-            response_image_url += chunk.text  # Se supone que aquí obtenemos la URL de la imagen generada
-
-        # Verifica si obtuviste una URL
-        if response_image_url:
-            return response_image_url
-        else:
-            return "No se pudo generar la imagen, la respuesta está vacía."
-        
-    except Exception as e:
-        return f"Se produjo un error al generar la imagen: {e}"
-
 # Aplicación Streamlit
 st.title("Gemini LLM App")
 
-# Selección entre Chat, PDF o Generación de Imagen
-option = st.radio("Elige una opción:", ("Hablar con el chat", "Subir y preguntar sobre un PDF", "Generar una imagen"))
+# Selección entre Chat o PDF
+option = st.radio("Elige una opción:", ("Hablar con el chat", "Subir y preguntar sobre un PDF"))
 
 # Opción 1: Hablar con el chat
 if option == "Hablar con el chat":
@@ -143,21 +102,3 @@ elif option == "Subir y preguntar sobre un PDF":
             st.error("No se pudo extraer texto del archivo PDF. Por favor, intenta con otro archivo.")
     else:
         st.info("Por favor, sube un archivo PDF para empezar.")
-
-# Opción 3: Generar una imagen
-elif option == "Generar una imagen":
-    # Entrada para la descripción de la imagen
-    image_description = st.text_area("Introduce una descripción para generar la imagen:", placeholder="Escribe aquí lo que quieres ver en la imagen...")
-
-    if st.button("Generar Imagen"):
-        if image_description.strip():
-            with st.spinner("Generando imagen..."):
-                image_url = generate_image(image_description)
-            
-            if image_url.startswith("http"):  # Verifica si se obtuvo una URL válida de la imagen
-                st.success("¡Imagen generada!")
-                st.image(image_url, caption="Imagen generada por Gemini", use_column_width=True)
-            else:
-                st.error(f"No se pudo generar la imagen. Error: {image_url}")
-        else:
-            st.error("Por favor, introduce una descripción para generar la imagen.")