Update app.py

app.py

@@ -1,81 +1,129 @@
+from dotenv import load_dotenv
 import streamlit as st
-import google.generativeai as genai
 import os
-from dotenv import load_dotenv
-import textwrap
+import google.generativeai as genai
+from PIL import Image
 
-# Cargar variables de entorno
 load_dotenv()
 
-# Configurar la API de Google Gemini
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para formatear texto en Markdown
-def to_markdown(text):
-    return textwrap.indent(text, '> ', predicate=lambda _: True)
+# Function to get response from Gemini model
+def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
+    model = genai.GenerativeModel('gemini-1.5-flash')
+    # If both image and text are provided
+    if image_data and input_prompt:
+        full_prompt = f"""
+        You are a famous creative writer. Look at the image provided and also consider the following prompt: "{input_prompt}". 
+        Create a {length} {genre} in {language}. The {genre} should be {mood}, based on both the image and the prompt, and contain realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt, image_data[0]])
 
-# Función para generar el párrafo de apertura
-def generate_opening_paragraph(target_audience, product, temperature, text_type, model_name):
-    # Crear la configuración del modelo
-    generation_config = {
-        "temperature": temperature,
-        "top_p": 0.95,
-        "top_k": 62,
-        "max_output_tokens": 2048,
-        "response_mime_type": "text/plain",
-    }
+    # If only image is provided
+    elif image_data:
+        full_prompt = f"""
+        You are a famous creative writer. Look at the image provided and create a {length} {genre} in {language}. 
+        The {genre} should be {mood}. The {genre} should be based on the image and contain realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt, image_data[0]])
 
-    model = genai.GenerativeModel(
-        model_name=model_name,
-        generation_config=generation_config,
-        system_instruction=f"Eres un especialista en copywriting altamente capacitado, enfocado en la creación de textos persuasivos que logran captar la atención del lector, mantener su interés y llevarlo a la acción. Tienes una profunda comprensión de la psicología del consumidor, y utilizas la persuasión de manera efectiva en cada palabra que escribes, conectando con las emociones y necesidades de tu audiencia. Dominas técnicas avanzadas de redacción, como el uso de encabezados poderosos, aperturas intrigantes, llamados a la acción efectivos y storytelling cautivador. Tu habilidad para ajustar el tono y estilo al contexto asegura que cada mensaje resuene profundamente. Tienes una amplia experiencia demostrada en generar resultados de ventas tanto para negocios en línea como offline. Las respuestas deben presentarse en formato de texto, en párrafos, sin mencionar directamente el producto o servicio, ni referirse al público objetivo de manera explícita. Al responder escribe un encabezado que diga: 'Este es tu párrafo de apertura para cautivar a {target_audience}'"
-    )
-
-    chat_session = model.start_chat(
-        history=[
+    # If only text is provided
+    elif input_prompt:
+        full_prompt = f"""
+        You are a creative writer. Create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the following prompt:
+        
+        "{input_prompt}"
+        
+        Make sure it contains realistic and emotional elements.
+        """
+        response = model.generate_content([full_prompt])
+    
+    # If neither image nor text is provided
+    else:
+        raise ValueError("Please provide either an image or a text prompt.")
+    
+    # Check if response is valid and return text
+    if response and response.parts:
+        return response.parts[0].text
+    else:
+        raise ValueError("Sorry, please try a different combination of inputs.")
+        
+# Function to setup uploaded image
+def input_image_setup(uploaded_file):
+    if uploaded_file is not None:
+        bytes_data = uploaded_file.getvalue()
+        image_parts = [
             {
-                "role": "user",
-                "parts": [
-                    f"Crea un párrafo de apertura para una {text_type.lower()} que haga conciente a {target_audience} que tiene un problema explicándolos con situaciones reales de su vida, utiliza un lenguaje natural o conversacional. El objetivo de este párrafo es que tengan el deseo de seguir leyendo este párrafo y saber de qué se trata {product}. Utiliza la persuasión de manera efectiva en cada palabra, dominando técnicas avanzadas como el uso de encabezados poderosos y aperturas intrigantes."
-                    "Para escribir este párrafo de apertura utiliza una o algunas de estas técnicas de Apertura para Cartas de Ventas, escoge las más adecuadas:"
-                    "1. Si.. Entonces"
-                    "2. Si / Entonces + Autoridad"
-                    "3. Honestidad"
-                    "4. Sensacionalista"
-                    "5. Haz una pregunta"
-                    "6. 'Micro aperturas'"
-                ],
-            },
+                "mime_type": uploaded_file.type,
+                "data": bytes_data
+            }
         ]
-    )
+        return image_parts
+    else:
+        return None
 
-    response = chat_session.send_message("Genera el párrafo de apertura")
-    return to_markdown(response.text)
+# Initialize Streamlit app
+st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wide")  # Set layout to wide
 
-# Título de la aplicación
-st.title("Generador de Párrafos de Apertura")
+# Main UI components
+st.markdown("<h1 style='text-align: center;'>VisionTales</h1>", unsafe_allow_html=True)
+st.markdown("<h3 style='text-align: center;'>Convierte tus ideas en historias inolvidables.</h3>", unsafe_allow_html=True)
 
-# Crear un formulario para la entrada del usuario
-with st.form(key='my_form'):
-    target_audience = st.text_input("Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
-    product = st.text_input("Producto", placeholder="Ejemplo: Curso de Inglés")
-    temperature = st.slider("Creatividad", 0.0, 1.0, 0.5, 0.1)
-    text_type = st.selectbox("Tipo de Texto", ["Página de Ventas", "Correo", "Historia"])
-    model_selector = st.selectbox("Selecciona el modelo", ["gemini-1.5-flash", "gemini-1.5-pro"])
-    
-    # Botón para enviar el formulario
-    submit_btn = st.form_submit_button("Generar Párrafo de Apertura")
-    
-    # Botón para limpiar el formulario (reiniciar)
-    if st.button("Empezar de nuevo"):
-        target_audience = ""
-        product = ""
-        temperature = 0.5
-        text_type = "Página de Ventas"
-        model_selector = "gemini-1.5-flash"
+# Add custom CSS for the button
+st.markdown("""
+    <style>
+    div.stButton > button {
+        background-color: #FFCC00; /* Color llamativo */
+        color: black;  /* Texto en negro */
+        width: 90%;
+        height: 60px;
+        font-weight: bold;
+        font-size: 22px; /* Tamaño más grande */
+        text-transform: uppercase; /* Texto en mayúsculas */
+        border: 1px solid #000000; /* Borde negro de 1px */
+        border-radius: 8px;
+        display: block;
+        margin: 0 auto;  /* Centramos el botón */
+    }
+    div.stButton > button:hover {
+        background-color: #FFD700; /* Color al pasar el mouse */
+        color: black;  /* Texto sigue en negro */
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+# Create two columns for layout (40% and 60%)
+col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 parts total
+
+with col1:
+    # Subir imagen primero
+    uploaded_file = st.file_uploader("Elegir una imagen...", type=["jpg", "jpeg", "png"])
+
+    # Agrupar opciones en un desplegable con un nuevo título
+    with st.expander("Clic para personalizar tu historia", expanded=False):  # El desplegable está cerrado por defecto
+        input_prompt = st.text_input("Escribe de qué quieres que trate tu historia (opcional):", placeholder="Escribe tu mensaje aquí...")
+        genre = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
+        length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
+        language = st.selectbox("Idioma del texto:", ["Inglés", "Español"])
+        mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
+
+    # Generate button con el nuevo estilo, centrado y ajustado
+    submit = st.button("Escribir mi historia")
+
+# Display uploaded image
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    col1.image(image, caption="Imagen subida", use_column_width=True)
 
-# Mostrar el resultado
-if submit_btn:
-    # Llamar a la función para generar el párrafo
-    output_text = generate_opening_paragraph(target_audience, product, temperature, text_type, model_selector)
-    st.markdown(output_text)
+# Handling the button click
+if submit:
+    if uploaded_file or input_prompt:
+        try:
+            image_data = input_image_setup(uploaded_file)
+            response = get_gemini_response(input_prompt, image_data, genre, length, language, mood)
+            col2.subheader("Contenido generado:")
+            col2.write(response)
+        except ValueError as e:
+            col2.error(f"Error: {str(e)}")
+    else:
+        col2.error("Por favor, sube una imagen o proporciona un mensaje.")