Update app.py

app.py

@@ -1,113 +1,129 @@
-import gradio as gr
-import google.generativeai as genai
-import os
 from dotenv import load_dotenv
-from gradio import Markdown
-import textwrap
+import streamlit as st
+import os
+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"))
-
-def to_markdown(text):
-    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-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",
-    }
+# 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]])
 
-    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 parrafo de apertura para cautivar a {target_audience}'"
-    )
+    # 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]])
 
-    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 explicandolos 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 que 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 mas adecuadas:"
-"1. Si.. Entonces"
-"Este es uno de los párrafos de apertura más usados. Gary Halbert, uno de los publicistas más importantes del siglo XX, solía usarlos en casi cada una de sus cartas de venta. Es muy simple, claro y poderoso. Hace que la gente sepa qué esperar."
-"Y hay una estructura de creencia muy interesante que se construye con este combo. Si eso, entonces esto. Es como si tuvieran que tener una especie de requisito previo a obtener el beneficio."
-"Un ejemplo de esto sería: 'Si quieres hacer más de 100,000 dólares al año trabajando solo 4 horas al día, entonces esta carta te mostrará cómo.'"
-"Muy simple. Si quieres eso, entonces así es cómo lo haces."
-"2. Si / Entonces + Autoridad"
-"Aún más poderoso que la llave 'Si / Entonces' es la misma, pero agregando la figura de autoridad o credibilidad. Por ejemplo: 'Si quieres saber una manera secreta de volverte casi invencible en una pelea... entonces aquí te mostraré lo que hacían los gladiadores y los samuráis. Y por qué tú también puedes hacer lo mismo a partir de hoy.'"
-"Una manera de notar el poder de estos párrafos de apertura es quitando la figura de autoridad. En este caso: los gladiadores y samuráis."
-"Eso cambiaría el párrafo a: 'Si quieres saber una manera secreta de volverte casi invencible en una pelea... entonces aquí te mostraré por qué tú también puedes hacer lo mismo a partir de hoy.'"
-"Como puedes ver, no tiene tanto impacto como la versión que agrega a los gladiadores y samuráis."
-"3. Honestidad"
-"Ser transparente es una de las mejores formas de obtener más clientes. Y también funciona con los párrafos de apertura."
-"Por ejemplo, uno de los anuncios de Gary Halbert iniciaba de esta manera: 'Mi nombre es Betty Adams, y lo primero que debes saber sobre mí es que... no soy una doctora. Tampoco soy experta en nutrición. No fui a la universidad y ni siquiera tengo un grado académico. De hecho, la única cosa en la que me considero experta es...'"
-"Y luego iba a un subtítulo en el que mencionaba cómo podía hacerte bajar 10 kilos de una manera muy sencilla."
-"En este caso puedes notar todos los negativos que usa: 'no soy doctora', 'no soy experta en nutrición', 'no fui a la escuela', 'no tengo un grado académico'. Son cuatro cosas negativas lo cual fortalece su afirmación sobre lo que sí es. Ahora, esto no siempre va a funcionar para todas las cartas. Dependerá de tu mercado y tu cliente ideal. Si tu mercado no es escéptico, no necesitas usar este tipo de párrafo."
-"4. Sensacionalista"
-"Éste está relacionado con el encabezado sensacionalista. Aquí es donde dirías algo como:'Déjame contarte esta historia extraña sobre un hombre que fue arrestado debido a que tenía herpes.'"
-"Es sensacionalista. Y lo sensacionalista vende. Siempre lo ha hecho y lo hará."
-"Sin duda, el párrafo es extraño y hace pensar a tu lector: '¿Cómo alguien fue arrestado por tener herpes? ¿Qué hizo? ¿Qué sucedió?' Los deja enganchados, adictos a querer saber más. Tienen que saberlo."
-"5. Haz una pregunta"
-"Hacen que las personas se detengan si están interesadas. Uno de los mejores párrafos de apertura que hay es el siguiente: '¿Sabías que el valor a largo plazo de las tarjetas de béisbol es mucho mayor que el de las acciones y monedas raras?'"
-"Así que cuando le hablas a alguien que está en el nicho de finanzas, inversionistas y demás, sobre este tema, estará interesado porque es diferente y lo deja listo para iniciar la aventura."
-"6. 'Micro aperturas'"
-"Este párrafo debe ser corto. Unas cuantas palabras... para que lean el siguiente. Esto es debido a que al ser breve, lo hace fácil de leer."
-"Y si dices algo que ponga a tu lector en la mentalidad adecuada, esto puede ser bastante poderoso. Por ejemplo, si tu producto está dirigido a un mercado escéptico, puedes usar un párrafo como el siguiente: 'Yo era un escéptico total.'"
-"Con esto estás yendo justo a su mente, a sus dudas."
-"Uno que puedes usar con efectividad es: 'Imagina lo siguiente:'"
-                ],
-            },
+                "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
 
-# Configurar la interfaz de usuario con Gradio
-css = """
-<style>
-    body { font-family: Arial, sans-serif; }
-    .gradio-container { max-width: 800px; margin: auto; }
-</style>
-"""
+# 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)
 
-with gr.Blocks(css=css) as iface:
-    gr.HTML("<h1 class='text-3xl font-bold text-center mb-6'>Generador de Párrafos de Apertura</h1>")
-    
-    with gr.Row():
-        with gr.Column():
-            target_audience = gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
-            product = gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés")
-            temperature = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Creatividad")
-            text_type = gr.Dropdown(choices=["Página de Ventas", "Correo", "Historia"], label="Tipo de Texto", value="Página de Ventas")
-            model_selector = gr.Dropdown(choices=["gemini-1.5-flash", "gemini-1.5-pro"], value="gemini-1.5-flash", label="Selecciona el modelo")
-        
-            # Colocar los botones en una misma fila
-            with gr.Row():
-                submit_btn = gr.Button("Generar Párrafo de Apertura", variant="primary")
-                clear_btn = gr.Button("Empezar de nuevo", variant="secondary")
-        
-        with gr.Column():
-            output_text = gr.Markdown(label="Párrafo de Apertura Generado")
-    
-    # Funcionalidad del botón Generar
-    submit_btn.click(fn=generate_opening_paragraph, inputs=[target_audience, product, temperature, text_type, model_selector], outputs=output_text)
+# 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:", ["Español", "Inglés"])
+        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")
 
-    # Funcionalidad del botón Limpiar: restablecer valores a los predeterminados
-    clear_btn.click(
-        fn=lambda: ("", "", 0.5, "Página de Ventas", "gemini-1.5-flash"),
-        inputs=[],
-        outputs=[target_audience, product, temperature, text_type, model_selector]
-    )
+# Display uploaded image
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    col1.image(image, caption="Imagen subida", use_column_width=True)
 
-# Lanzar la interfaz
-iface.launch()
+# 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.")