app.py

from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai

load_dotenv()

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

# Función para obtener respuesta del modelo Gemini
def get_gemini_response(target_audience, product, product_mention, text_type, length, mood, model_choice):
    model = genai.GenerativeModel(model_choice)
    
    # Crear la instrucción de mención basada en la opción seleccionada
    mention_instruction = ""
    if product_mention == "Directa":
        mention_instruction = f"Mention the product '{product}' directly in the text."
    elif product_mention == "Indirecta":
        mention_instruction = f"Ensure that subtle and realistic references to the product '{product}' are included without explicitly mentioning it."
    elif product_mention == "Metafórica":
        mention_instruction = f"Refer to the product '{product}' through a metaphor without explicitly mentioning it."

# Instrucción específica para el tipo de texto
    format_instruction = ""
    if text_type == "Historia":
        format_instruction = """
        Write an engaging opening paragraph that introduces a relatable problem your audience is facing. Stir emotions by highlighting the impact of the problem on their life, and create curiosity with hints of a potential solution without revealing everything. Ensure that the narrative captivates the reader, leaving them wanting to discover how the story unfolds.
        """
    elif text_type == "Carta de venta":
        format_instruction = """
        Craft a persuasive lead paragraph for a sales letter that quickly identifies a common pain point your reader is experiencing. Agitate the problem by emphasizing the negative consequences of not addressing it, and tease the possibility of a solution to spark curiosity and compel the reader to continue reading for the answer.
        """
    elif text_type == "Correo":
        format_instruction = """
        Write a concise but impactful opening paragraph for an email that immediately captures the reader’s attention by acknowledging a challenge they are likely facing. Agitate the problem by emphasizing how it might be affecting their daily life, and leave them intrigued by hinting at a valuable solution that will be revealed if they continue reading.
        """
    elif text_type == "Landing page":
        format_instruction = """
        Design a compelling first paragraph for a landing page that immediately draws attention to a significant problem your target audience is encountering. Stir emotions by describing the frustrations or difficulties this issue causes and entice them to keep scrolling by subtly suggesting a solution without giving too much away.
        """

    # Crear el prompt completo basado en los campos del frontend
    full_prompt = f"""
    You are a creative writer skilled in the art of persuasion. Write a {length} {text_type} in Spanish. The tone of the {text_type} should be {mood} and carefully crafted to emotionally resonate with a {target_audience}. {mention_instruction} {format_instruction} Use persuasive techniques to guide the reader towards an intuitive understanding of the product's benefits, focusing on creating a strong emotional connection with the audience.
    Create an opening paragraph for a {text_type.lower()} that makes {target_audience} aware they have a problem by explaining it with real-life situations, using a natural or conversational tone. The goal of this paragraph is to make them want to keep reading and find out what {product} is about. Use persuasion effectively in every word, mastering advanced techniques like using powerful headlines and intriguing openings.

    To write this opening paragraph, use one or more of these Copywriting Techniques, choosing the most appropriate ones:

    1. If/Then: This is one of the most commonly used opening paragraphs. Gary Halbert, one of the most important copywriters of the 20th century, used them in almost every one of his sales letters. It’s very simple, clear, and powerful. It lets people know what to expect. There’s also a very interesting belief structure that is built with this combo. If this, then that. It’s as if they need to meet some kind of prerequisite to get the benefit. An example of this would be: 'If you want to make more than $100,000 a year working just 4 hours a day, then this letter will show you how.' Very simple. If you want that, then here’s how you do it.
    2. If / Then + Authority: Even more powerful than the 'If / Then' key is the same concept but adding a figure of authority or credibility. For example: 'If you want to know a secret way to become almost invincible in a fight... then here I’ll show you what the gladiators and samurais used to do. And why you can do it too starting today.' One way to notice the power of these opening paragraphs is by removing the authority figure. In this case: the gladiators and samurais. That would change the paragraph to: 'If you want to know a secret way to become almost invincible in a fight... then here I’ll show you why you can do it too starting today.' As you can see, it doesn’t have as much impact as the version with the gladiators and samurais.
    3. Honesty: Being transparent is one of the best ways to get more customers. It also works with opening paragraphs. For example, one of Gary Halbert’s ads started like this: 'My name is Betty Adams, and the first thing you should know about me is... I’m not a doctor. I’m also not a nutrition expert. I didn’t go to college and I don’t even have a degree. In fact, the only thing I consider myself an expert in is...' And then it went to a subheading where she mentioned how she could help you lose 10 kilos in a very simple way. In this case, you can notice all the negatives she uses: 'I’m not a doctor,' 'I’m not a nutrition expert,' 'I didn’t go to school,' 'I don’t have a degree.' These four negative statements strengthen her claim about what she is. Now, this won’t always work for every sales letter. It will depend on your market and ideal customer. If your market isn’t skeptical, you don’t need to use this kind of paragraph.
    4. Sensationalist: This one is related to the sensational headline. Here’s where you’d say something like: 'Let me tell you this strange story about a man who was arrested because he had herpes.' It’s sensationalist. And sensationalism sells. It always has and always will. Without a doubt, the paragraph is strange and makes your reader think: 'How was someone arrested for having herpes? What did he do? What happened?' It keeps them hooked, addicted to finding out more. They need to know.
    5. Ask a Question: It makes people stop if they’re interested. One of the best opening paragraphs is this: 'Did you know that the long-term value of baseball cards is much higher than that of stocks and rare coins?' So when you talk to someone in the finance, investor, or related niches about this topic, they will be interested because it’s different and gets them ready to start the journey.
    6. Micro Openings: This paragraph should be short. Just a few words... so they’ll read the next one. This is because being brief makes it easy to read. And if you say something that puts your reader in the right mindset, this can be quite powerful. For example, if your product is targeted to a skeptical market, you can use a paragraph like the following: 'I was a total skeptic.' With this, you’re going straight to their mind, to their doubts. One that you can use effectively is: 'Imagine the following:'
    """
    
    response = model.generate_content([full_prompt])

    # Comprobar si la respuesta es válida y devolver el texto
    if response and response.parts:
        return response.parts[0].text
    else:
        raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")

# Inicializar la aplicación Streamlit
st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wide")  # Configurar el diseño en ancho

# Componentes principales de la UI
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)

# Añadir CSS personalizado para el botón
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)

# Crear dos columnas para el diseño (40% y 60%)
col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 partes en total

with col1:
    # Entradas del usuario
    target_audience = st.text_input("Público objetivo:", placeholder="Especifica tu público aquí...")
    product = st.text_input("Producto:", placeholder="Especifica el producto aquí...")
    
    # Agrupar todas las opciones en una acordeón
    with st.expander("Personaliza tu texto"):
        # Variable para cómo se debe mencionar el producto
        product_mention = st.selectbox("Mención del producto:", ["Directa", "Indirecta", "Metafórica"])
        
        # Opciones actualizadas para el tipo de texto
        text_type = st.selectbox("Tipo de texto:", ["Historia", "Carta de venta", "Correo", "Landing page"])
        length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
        mood = st.selectbox("Tono del Texto:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
        
        # Opción para seleccionar el modelo
        model_choice = st.selectbox("Selecciona el modelo:", ["gemini-1.5-flash", "gemini-1.5-pro"], index=0)

    # Botón para generar contenido
    submit = st.button("Escribir mi historia")

# Manejo del clic en el botón
if submit:
    if target_audience and product:  # Verificar que se haya proporcionado el público objetivo y el producto
        try:
            response = get_gemini_response(target_audience, product, product_mention, text_type, length, mood, model_choice)
            col2.subheader("Contenido generado:")
            col2.write(response)
        except ValueError as e:
            col2.error(f"Error: {str(e)}")
    else:
        col2.error("Por favor, proporciona el público objetivo y el producto.")