Update app.py

app.py

@@ -2,7 +2,7 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
-import random  # Nueva importación para la selección aleatoria de mención
+import random
 
 load_dotenv()
 
@@ -11,76 +11,57 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Función para obtener una mención del producto de manera probabilística
 def get_random_product_mention():
     mentions = ["Directa", "Indirecta", "Metafórica"]
-    probabilities = [0.34, 0.33, 0.33]  
+    probabilities = [0.05, 0.10, 0.85]
     return random.choices(mentions, probabilities)[0]
 
 # Crear la instrucción de mención basada en la opción seleccionada
 def get_mention_instruction(product_mention, product):
     if product_mention == "Directa":
         return f"""
-        Directly introduce the product '{product}' as the clear solution to the problem the reader is facing. Ensure that the product is presented in a way that highlights its key benefits and demonstrates how it directly addresses the issue at hand. The mention should feel natural and seamlessly integrated into the narrative.
+        Directly introduce the product '{product}' as the clear solution to the problem the reader is facing. Ensure that the product is presented in a way that highlights its key benefits and demonstrates how it directly addresses the issue at hand.
         """
     elif product_mention == "Indirecta":
         return f"""
-        Subtly reference the product '{product}' as a potential solution to the reader's problem without naming it explicitly. Weave the product's core benefits into the description of how the reader can overcome the issue, creating an implicit connection between the solution and the product. Ensure the mention is subtle but clear enough to guide the reader towards the product.
+        Subtly reference the product '{product}' as a potential solution to the reader's problem without naming it explicitly. Weave the product's core benefits into the description of how the reader can overcome the issue.
         """
     elif product_mention == "Metafórica":
         return f"""
-        Introduce the product '{product}' using a metaphor, connecting it symbolically to the solution the reader needs. The metaphor should relate to the problem being discussed and should creatively suggest how the product offers a resolution without explicitly stating its name. The metaphor should evoke the benefits of the product in a memorable and thought-provoking way.
+        Introduce the product '{product}' using a metaphor, connecting it symbolically to the solution the reader needs. The metaphor should creatively suggest how the product offers a resolution without explicitly stating its name.
         """
     return ""
 
 # Función para obtener una técnica de copywriting de manera aleatoria
 def get_random_copywriting_technique():
     techniques = [
-        "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:'"
+        "If/Then: Create a condition and show how it leads to the desired result.",
+        "Questions: Pose a question to engage the reader and lead into the solution.",
+        "Honesty: Build trust by being transparent about limitations or challenges.",
+        "Sensationalist: Use an unusual or shocking statement to hook the reader.",
+        "Micro Openings: Use a very short and impactful opening to capture attention.",
+        "Authority: Refer to a credible figure or data to reinforce your point."
     ]
-    return random.sample(techniques, k=random.randint(1, 3))  # Escoge entre 1 y 3 técnicas al azar
+    return random.sample(techniques, k=random.randint(1, 3))
 
-# Al crear el prompt completo, incluye el tamaño y las técnicas seleccionadas
-def get_gemini_response(target_audience, product, text_type, length, mood, emotionality):
+# Generar respuesta del modelo
+def get_gemini_response(target_audience, product, formula, length, mood, emotionality):
     product_mention = get_random_product_mention()
-    mention_instruction = get_mention_instruction(product_mention, product)  # Agrega esta línea
-    model_choice = "gemini-1.5-flash"  # Modelo predeterminado
-
-    model = genai.GenerativeModel(model_choice)
+    mention_instruction = get_mention_instruction(product_mention, product)
 
     # Obtener técnicas aleatorias
     copywriting_techniques = get_random_copywriting_technique()
     techniques_text = " ".join(copywriting_techniques)
 
-    # Instrucción específica para el tipo de texto
-    format_instruction = ""
-    if text_type == "Historia":
-        format_instruction = f"""
-        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. The paragraph should be approximately {length} words long to maintain engagement and avoid losing the reader's attention.
-        """
-    elif text_type == "Carta de venta":
-        format_instruction = f"""
-        Craft a persuasive lead paragraph for a sales letter that quickly identifies a common pain point your reader is experiencing. The paragraph should be around {length} words long and agitate the problem by emphasizing the negative consequences of not addressing it, and tease the possibility of a solution to spark curiosity.
-        """
-    elif text_type == "Correo":
-        format_instruction = f"""
-        Write a concise but impactful opening paragraph for an email that captures the reader’s attention. The text should be approximately {length} words, addressing a challenge they are likely facing, 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 = f"""
-        Design a compelling first paragraph for a landing page that immediately draws attention to a significant problem your target audience is encountering. The paragraph should be about {length} words long, stirring emotions and enticing them to keep scrolling by subtly suggesting a solution.
-        """
-
-    # Crear el prompt completo basado en los campos del frontend
+    # Crear el prompt completo
     full_prompt = f"""
-    You are a creative writer skilled in the art of persuasion. The tone of the {text_type} should be {mood} and {emotionality} emotionally resonate with a {target_audience}. Evoke feelings of {target_audience} through relatable, real-life situations that reflect their experiences. {mention_instruction} {format_instruction} 
-Use the following copywriting techniques to write this opening paragraph: {techniques_text}.
-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} of {length} words in Spanish that makes {target_audience} aware they have a problem. Highlight this problem by explaining it with relatable situations tailored to their context, using a natural or conversational tone. The goal of this paragraph is to entice them to keep reading and discover what {product} is about. Use persuasion effectively in every word, mastering advanced techniques.
+    You are a creative writer skilled in the art of persuasion. Write an opening paragraph using the {formula} formula to connect with a {target_audience}. 
+    The tone should be {mood}, and the emotionality should convey {emotionality}.
+    {mention_instruction}
+    Use the following copywriting techniques: {techniques_text}.
+    The paragraph should be approximately {length} words long and emotionally resonate with the audience, enticing them to continue reading.
     """
-
+    
+    model_choice = "gemini-1.5-flash"
+    model = genai.GenerativeModel(model_choice)
     response = model.generate_content([full_prompt])
 
     # Comprobar si la respuesta es válida y devolver el texto
@@ -89,74 +70,37 @@ Create an opening paragraph for a {text_type} of {length} words in Spanish that
     else:
         raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
 
-# Inicializar la aplicación Streamlit
+# Configuración de la página de Streamlit
 st.set_page_config(page_title="First Line Alchemy", layout="wide")
 
-# Inicializar el estado de la expansión del acordeón
-if "accordion_expanded" not in st.session_state:
-    st.session_state["accordion_expanded"] = False  # Por defecto está cerrado
-
-# Definir una función para manejar el cambio de estado del acordeón
-def toggle_accordion():
-    st.session_state["accordion_expanded"] = not st.session_state["accordion_expanded"]
-
-# Centrar el título y el subtítulo
+# Interfaz de usuario
 st.markdown("<h1 style='text-align: center;'>First Line Alchemy</h1>", unsafe_allow_html=True)
-st.markdown("<h4 style='text-align: center;'>Experimenta la magia de transformar conceptos ordinarios en líneas de apertura extraordinarias que cautivan la imaginación e inspiran acción.</h4>", 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)
+st.markdown("<h4 style='text-align: center;'>Transforma conceptos en líneas que cautivan e inspiran acción.</h4>", unsafe_allow_html=True)
 
-# Create two columns for the layout (40% and 60%)
-col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 parts in total
+col1, col2 = st.columns([2, 3])
 
 with col1:
-    # Create input fields
     target_audience = st.text_input("¿Quién es tu público objetivo?")
     product = st.text_input("¿Qué producto tienes en mente?")
     
-    # Accordion for customizing the opening paragraph, controlled by session_state
-    with st.expander("Personaliza tu párrafo de apertura", expanded=st.session_state["accordion_expanded"]):
-        text_type = st.selectbox("Tipo de texto", ["Historia", "Carta de venta", "Correo", "Landing page"])
+    with st.expander("Personaliza tu párrafo de apertura"):
+        formula = st.selectbox("Selecciona una fórmula de copywriting", ["If/Then", "Questions", "Honesty", "Sensationalist", "Micro Openings", "Authority"])
         length = st.slider("Número de palabras", min_value=50, max_value=100, value=50)
         mood = st.selectbox("Tono de voz", ["Conversacional", "Formal", "Persuasivo"])
-        emotionality = st.selectbox("Emocionalidad del texto", ["Triste", "Feliz", "Horror", "Comedia", "Romántico", "Sorpresa", "Desesperación", "Esperanza", "Ira", "Confianza"])
+        emotionality = st.selectbox("Emocionalidad del texto", ["Triste", "Feliz", "Romántico", "Sorpresa", "Esperanza", "Ira", "Confianza"])
 
-    # Submit button
-    submit = st.button("Escribir mi Texto", on_click=toggle_accordion)
+    submit = st.button("Generar Texto")
 
-# Button to generate text (outside the accordion)
 if submit:
-    if target_audience and product:  # Check if target audience and product are provided
+    if target_audience and product:
         try:
-            # Get the response from the model
-            generated_text = get_gemini_response(target_audience, product, text_type, length, mood, emotionality)
-            col2.markdown("""
-                <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
-                    <h4>Aquí está tu párrafo de apertura:</h4>
-                    <p>{}</p>
+            generated_text = get_gemini_response(target_audience, product, formula, length, mood, emotionality)
+            col2.markdown(f"""
+                <div style="border: 1px solid #000000; padding: 10px; border-radius: 8px; background-color: #ffffff;">
+                    <h4>Tu párrafo de apertura:</h4>
+                    <p>{generated_text}</p>
                 </div>
-            """.format(generated_text), unsafe_allow_html=True)  # Update with generated text
+            """, unsafe_allow_html=True)
         except ValueError as e:
             col2.error(f"Error: {str(e)}")
     else: