Update app.py

app.py

@@ -50,11 +50,44 @@ 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)
+        
+        # Se asume que la API de Gemini tiene un endpoint para generar imágenes (ajústalo según lo que soporte Gemini)
+        model = "gemini-2.0-flash-thinking-exp-01-21"  # Cambiar según el modelo de imagen
+        contents = [
+            types.Content(
+                role="user",
+                parts=[types.Part.from_text(text=description)],
+            ),
+        ]
+        generate_content_config = types.GenerateContentConfig(
+            temperature=0.7,
+            top_p=0.95,
+            top_k=64,
+            max_output_tokens=65536,
+            response_mime_type="image/png",  # Suponiendo que la respuesta es en formato de 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  # Suponiendo que la API devuelve una URL de la imagen generada
+
+        return response_image_url
+    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 o PDF
-option = st.radio("Elige una opción:", ("Hablar con el chat", "Subir y preguntar sobre un PDF"))
+# 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"))
 
 # Opción 1: Hablar con el chat
 if option == "Hablar con el chat":
@@ -102,3 +135,21 @@ 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.")