from dotenv import load_dotenv import streamlit as st import os import google.generativeai as genai import random from streamlit import session_state as state from formulas import email_formulas from angles import angles import re import datetime import PyPDF2 import docx from PIL import Image # Cargar las variables de entorno load_dotenv() # Configurar la API de Google genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) # Fórmulas con ejemplos y explicaciones # email_formulas dictionary has been moved to formulas/email_formulas.py # Cambiar el nombre de la función def generate_emails(target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False, emotion="", desired_action="", creative_idea="", brand_tone=""): # Crear la configuración del modelo generation_config = { "temperature": temperature, "top_p": 0.65, "top_k": 360, "max_output_tokens": 8196, } model = genai.GenerativeModel( model_name="gemini-2.0-flash", generation_config=generation_config, ) # Definir las instrucciones de formato una sola vez format_instructions = """ FORMAT RULES: - Each email must start with "Correo 1", "Correo 2", etc. - Each email must have a clear and attractive subject line - The email body must be persuasive and emotional - Include a clear call to action - Add a professional signature - Separate each email clearly - Do not include greetings like 'Hello [Name]' - Make sure that the postscripts (P.D.) are smaller and more discrete than the main body - IMPORTANT: The first email must be an introduction without referencing any previous communications - Emails 2-5 should build on the sequence in a logical progression """ # Incluir las instrucciones del sistema en el prompt principal system_prompt = f"""You are a world-class direct response copywriter trained by Gary Halbert, Gary Bencivenga, and David Ogilvy. You understand how real people interact with emails in their busy inboxes: - They quickly scan and delete anything that looks like generic marketing - They only open emails that feel personal or spark genuine curiosity - They respond to messages that seem written by a real person, not a corporation - They engage with content that hooks them from the first line and maintains interest Your task is to create a 5-email sequence that makes {target_audience} feel {'' if emotion == 'NINGUNO' else emotion + ' '}about {product} and convince them to {desired_action}. AUDIENCE-SPECIFIC LANGUAGE ADAPTATION: - Analyze the language patterns typical of {target_audience} and mirror them in your writing - Adjust formality level based on audience demographics (more formal for professionals/executives, more casual for younger audiences) - Use vocabulary, references, and examples that will resonate specifically with {target_audience} - Match the communication style preferences of this audience (direct vs indirect, detailed vs concise) - Include cultural touchpoints or shared experiences relevant to this specific audience """ # Add brand tone instructions only if a specific tone is selected if brand_tone != "NINGUNO": system_prompt += f""" BRAND TONE: {brand_tone} - Write all emails using a consistent {brand_tone} tone of voice - Ensure language, expressions, and style reflect this tone throughout - Maintain this tone while still being persuasive and effective """ system_prompt += f""" Each email must follow these four essential actions: 1. ATTRACT with a subject line that feels personal and creates curiosity (never screaming "Buy me!") 2. CONNECT through rhythm, emotion, and storytelling that maintains interest until the end 3. CONVINCE with authentic language that feels believable and relevant 4. GUIDE to action naturally, making readers click almost without realizing it WRITING STYLE: - Write as if speaking to one person, not an audience - Vary sentence length deliberately (mix short punchy sentences with occasional longer ones) - Create natural paragraph rhythm (1-3 sentences per paragraph) - Use conversational transitions between ideas - Include specific details that make scenarios feel real {format_instructions} IMPORTANT: - Each email must be unique and memorable - Avoid clichés and generalities - Maintain a persuasive but credible tone - Adapt language to the target audience - Focus on transformative benefits""" # Iniciar el prompt con las instrucciones del sistema email_instruction = f"{system_prompt}\n\n" # Add instructions for using the enhanced formula structure if hasattr(selected_formula, 'get') and selected_formula.get('subject_line_types') and selected_formula.get('email_structure_template'): email_instruction += """ ENHANCED FORMULA STRUCTURE: Use the following structure elements to create more effective emails: SUBJECT LINE TYPES: """ # Add subject line types with descriptions for subject_type, description in selected_formula.get('subject_line_types', {}).items(): email_instruction += f"- {subject_type}: {description}\n" # Add technique types (either infotainment or nurturing) technique_key = next((k for k in selected_formula.keys() if k.endswith('_techniques')), None) if technique_key: email_instruction += f"\n{technique_key.upper()}:\n" for technique, description in selected_formula.get(technique_key, {}).items(): email_instruction += f"- {technique}: {description}\n" # Add email structure template email_instruction += "\nEMAIL STRUCTURE TEMPLATE:\n" for email_num, structure in selected_formula.get('email_structure_template', {}).items(): email_instruction += f"\n{email_num.upper()}:\n" email_instruction += f"- Purpose: {structure.get('purpose', '')}\n" email_instruction += f"- Emotional Focus: {structure.get('emotional_focus', '')}\n" # Add recommended subject types if structure.get('recommended_subject_types'): email_instruction += "- Recommended Subject Types: " + ", ".join(structure.get('recommended_subject_types', [])) + "\n" # Add recommended techniques if structure.get('recommended_techniques'): email_instruction += "- Recommended Techniques: " + ", ".join(structure.get('recommended_techniques', [])) + "\n" # Add key elements if structure.get('key_elements'): email_instruction += "- Key Elements:\n" for element in structure.get('key_elements', []): email_instruction += f" * {element}\n" # Añadir instrucciones para la idea creativa si existe if creative_idea: email_instruction += f""" CREATIVE CONCEPT: Use the following creative concept as the central theme for all emails in the sequence: "{creative_idea}" CREATIVE CONCEPT INSTRUCTIONS: 1. This concept should be the unifying theme across all emails 2. Use it as a metaphor or analogy throughout the sequence 3. Develop different aspects of this concept in each email 4. Make sure the concept naturally connects to the product benefits 5. The concept should make the emails more memorable and engaging """ # Añadir contenido del archivo si existe if file_content: email_instruction += f""" REFERENCE CONTENT: Carefully analyze the following content as a reference for generating emails: {file_content[:3000]} ANALYSIS INSTRUCTIONS: 1. Extract key information about the product or service mentioned 2. Identify the tone, style, and language used 3. Detect any data about the target audience or customer avatar 4. Look for benefits, features, or pain points mentioned 5. Use relevant terms, phrases, or concepts from the content 6. Maintain consistency with the brand identity or main message 7. Adapt the emails to resonate with the provided content IMPORTANT COMBINATIONS: """ # Updated conditions for specific input combinations if product and not target_audience: email_instruction += f"""- FILE + PRODUCT: You have a reference document and product ({product}). Create emails that highlight this specific product's benefits and features using insights from the document. Extract audience information from the document to better target the emails. """ elif target_audience and not product: email_instruction += f"""- FILE + TARGET AUDIENCE: You have a reference document and target audience ({target_audience}). Create emails tailored to this specific audience using language and concepts from the document. Identify products or services from the document that would appeal to this audience. """ elif product and target_audience: email_instruction += f"""- PRODUCT + TARGET AUDIENCE: You have both product ({product}) and target audience ({target_audience}). Create emails that connect this specific product with this specific audience, using insights from the document to strengthen the connection. """ email_instruction += """ IMPORTANT: Naturally integrate the elements found in the content with the selected formula and angle. """ # Preparar instrucciones específicas del ángulo una sola vez angle_instructions = "" if selected_angle != "NINGUNO": angle_instructions = f""" MAIN ANGLE: {selected_angle} SPECIFIC ANGLE INSTRUCTIONS: {angles[selected_angle]["instruction"]} IMPORTANT: The angle {selected_angle} must be applied as a "style layer" over the formula structure: 1. Keep the base structure of the formula intact 2. Apply the tone and style of the {selected_angle} angle 3. Ensure each element of the formula reflects the angle 4. The angle affects "how" it's said, not "what" is said SUCCESSFUL EXAMPLES OF THE {selected_angle} ANGLE: """ for example in angles[selected_angle]["examples"]: angle_instructions += f"- {example}\n" # Añadir las instrucciones del ángulo al prompt principal email_instruction += angle_instructions # Dentro de la función, actualizar el prompt para incluir emoción y acción deseada email_instruction += ( f"\nYour task is to create 5 persuasive emails for {target_audience} " f"that evoke {emotion} and convince them to {desired_action} about {product}. " ) # Usar la variable angle_instructions para determinar si hay un ángulo seleccionado if angle_instructions: email_instruction += f"IMPORTANT: Each email MUST follow the {selected_angle} angle clearly and consistently." email_instruction += "\n\n" # Take the first 5 examples (or all if less than 5) in order sequential_examples = selected_formula['examples'][:min(5, len(selected_formula['examples']))] email_instruction += "FORMULA EXAMPLES TO FOLLOW:\n" for i, example in enumerate(sequential_examples, 1): email_instruction += f"{i}. {example}\n" # Añadir instrucciones específicas sobre cómo seguir los ejemplos email_instruction += "\nSPECIFIC INSTRUCTIONS:\n" email_instruction += "1. Maintain the same structure and length as the previous examples\n" email_instruction += "2. Use the same tone and writing style\n" email_instruction += "3. Replicate the phrase construction patterns\n" email_instruction += "4. Preserve the level of specificity and detail\n" email_instruction += f"5. Adapt the content for {target_audience} while maintaining the essence of the examples\n\n" # Añadir instrucciones específicas sobre la temperatura creativa email_instruction += f"\nCREATIVITY LEVEL: {temperature}. Higher values mean more creative and original content.\n\n" email_instruction += f"FORMULA TO FOLLOW:\n{selected_formula['description']}\n\n" # Consolidar las instrucciones finales en un solo bloque final_reminder = ["Follow the structure of the selected formula"] # Add angle-specific instructions only if an angle is selected if selected_angle != "NINGUNO": final_reminder.extend([ "Apply the angle as a 'style layer'", "Maintain coherence between formula and angle", "Ensure each email reflects both elements" ]) email_instruction += "\nFINAL REMINDER:\n" for i, reminder in enumerate(final_reminder, 1): email_instruction += f"{i}. {reminder}\n" email_instruction += "\nGENERATE NOW:\nCreate 5 emails that faithfully follow the style and structure of the examples shown.\n" # Modificar la forma de enviar el mensaje según si hay imagen o no message_parts = [email_instruction] # Build instruction text more clearly with conditional components instruction_components = ["Generate the emails in Spanish following exactly the style and structure of the examples shown."] # Add the image to the message parts if it exists if is_image and image_parts: message_parts.append(image_parts) instruction_components.append("drawing inspiration from the provided image.") # Add final instruction instruction_components.append("Do not include explanations, only the emails.") # Join all instruction components with proper spacing instruction_text = " ".join(instruction_components) # Create the chat session with the message parts try: chat_session = model.start_chat( history=[ { "role": "user", "parts": message_parts, }, ] ) # Enviar el mensaje con las instrucciones response = chat_session.send_message(instruction_text) return response.text except genai.types.generation_types.StopCandidateException as e: # Handle content filtering/safety issues error_message = f"La generación se detuvo debido a restricciones de contenido: {str(e)}" st.error(error_message) return f"Error: {error_message}" except genai.types.generation_types.BlockedPromptException as e: # Handle blocked prompt issues error_message = f"El prompt fue bloqueado por políticas de contenido: {str(e)}" st.error(error_message) return f"Error: {error_message}" except Exception as e: # Handle general API errors error_message = f"Error al comunicarse con la API de Google: {str(e)}" st.error(error_message) return f"Error: {error_message}" # Define the clean_response_text function before it's used def clean_response_text(text): """Remove extra spaces and normalize whitespace in the response text""" # Replace multiple newlines with just two text = re.sub(r'\n{3,}', '\n\n', text) # Remove leading/trailing whitespace text = text.strip() return text # Configurar la interfaz de usuario con Streamlit st.set_page_config( page_title="Email Composer", layout="wide", menu_items={}, # Empty dict removes the three-dot menu initial_sidebar_state="expanded" ) # Add custom CSS to reduce margins st.markdown(""" """, unsafe_allow_html=True) # Leer el contenido del archivo manual.md manual_path = "manual.md" if os.path.exists(manual_path): with open(manual_path, "r", encoding="utf-8") as file: manual_content = file.read() # Mostrar el contenido del manual en el sidebar st.sidebar.markdown(manual_content) else: st.sidebar.warning("Manual file not found.") # Load CSS from file css_path = "styles/main.css" if os.path.exists(css_path): with open(css_path, "r") as f: css = f.read() # Apply the CSS st.markdown(f"", unsafe_allow_html=True) else: st.warning("CSS file not found.") # Centrar el título y el subtítulo st.markdown("

Generador de Emails

", unsafe_allow_html=True) st.markdown("

Transforma tu marketing con emails persuasivos que convierten. Esta aplicación es tu arma secreta para crear emails emocionales de respuesta directa que impulsan a la acción.

", unsafe_allow_html=True) # Crear columnas col1, col2 = st.columns([1, 2]) # Initialize these variables at the top level before the UI components file_content = "" is_image = False image_parts = None # Columnas de entrada with col1: target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios") product = st.text_input("¿Qué producto/servicio estás promocionando?", placeholder="Ejemplo: Curso de Inglés") # Move "Acción deseada" outside the accordion desired_action = st.text_input("Acción deseada", placeholder="Ejemplo: Registrarse para una prueba gratuita") # Move formula selection here, right after desired action selected_formula_key = st.selectbox( "Selecciona una fórmula para tus emails", options=list(email_formulas.email_formulas.keys()), key="formula_selectbox" # Add a unique key ) # Only one submit button with a unique key submit = st.button("Generar Emails", key="generate_emails_button") # Crear un único acordeón para fórmula, creatividad y ángulo with st.expander("Personaliza tus emails"): # 1. Idea Creativa al principio con altura de 70px creative_idea = st.text_area("Idea Creativa", placeholder="Ejemplo: Tu curso es como Netflix: ofrece contenido que engancha y soluciones que la gente realmente quiere ver", height=70) # 2. Lo demás (cargador de archivos) uploaded_file = st.file_uploader("📄 Archivo o imagen de referencia", type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png']) file_content = "" is_image = False image_parts = None if uploaded_file is not None: # El código para manejar archivos permanece igual file_type = uploaded_file.name.split('.')[-1].lower() # Manejar archivos de texto if file_type in ['txt', 'pdf', 'docx']: # El código para manejar archivos de texto permanece igual # ... if file_type == 'txt': try: file_content = uploaded_file.read().decode('utf-8') st.success(f"Archivo TXT cargado correctamente: {uploaded_file.name}") except Exception as e: st.error(f"Error al leer el archivo TXT: {str(e)}") file_content = "" elif file_type == 'pdf': try: import PyPDF2 pdf_reader = PyPDF2.PdfReader(uploaded_file) file_content = "" for page in pdf_reader.pages: file_content += page.extract_text() + "\n" st.success(f"Archivo PDF cargado correctamente: {uploaded_file.name}") except Exception as e: st.error(f"Error al leer el archivo PDF: {str(e)}") file_content = "" elif file_type == 'docx': try: import docx doc = docx.Document(uploaded_file) file_content = "\n".join([para.text for para in doc.paragraphs]) st.success(f"Archivo DOCX cargado correctamente: {uploaded_file.name}") except Exception as e: st.error(f"Error al leer el archivo DOCX: {str(e)}") file_content = "" # Manejar archivos de imagen elif file_type in ['jpg', 'jpeg', 'png']: try: from PIL import Image image = Image.open(uploaded_file) image_bytes = uploaded_file.getvalue() image_parts = { "mime_type": uploaded_file.type, "data": image_bytes } is_image = True st.image(image, caption="Imagen cargada", use_column_width=True) except Exception as e: st.error(f"Error processing image: {str(e)}") is_image = False # 4. Ángulo angle_keys = ["NINGUNO"] + sorted([key for key in angles.keys() if key != "NINGUNO"]) selected_angle = st.selectbox( "Selecciona un ángulo para tus emails", options=angle_keys, key="angle_selectbox" # Add a unique key ) # 5. Emoción emotion = st.selectbox( "¿Qué emoción quieres evocar?", options=["NINGUNO", "Curiosidad", "Miedo", "Esperanza", "Entusiasmo", "Confianza", "Urgencia"], key="emotion_selectbox" # Add a unique key ) # Añadir selector de tono de marca brand_tone = st.selectbox( "Tono de marca", options=["NINGUNO", "Profesional", "Conversacional", "Entusiasta", "Formal", "Informal", "Humorístico", "Sarcástico", "Serio", "Inspirador", "Educativo"], key="brand_tone_selectbox" ) # 6. Creatividad (slider) temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1) selected_formula = email_formulas.email_formulas[selected_formula_key] # Updated reference # Removed the submit button from here # Define a function to validate inputs def validate_inputs(file_content, product, target_audience, emotion, desired_action): """ Validates input combinations and returns a tuple of (is_valid, error_message) """ has_file = file_content.strip() != "" if file_content else False has_product = product.strip() != "" has_audience = target_audience.strip() != "" has_emotion = emotion.strip() != "" if emotion else False has_action = desired_action.strip() != "" if desired_action else False # Check for valid combinations if has_file and has_product: return True, "" # File + Product is valid elif has_file and has_audience: return True, "" # File + Audience is valid elif has_product and has_audience and has_emotion and has_action: return True, "" # Product + Audience + Emotion + Action is valid # If we get here, no valid combination was found if not (has_emotion and has_action): return False, "Por favor especifica la emoción que quieres evocar y la acción deseada." else: return False, "Por favor proporciona al menos una de estas combinaciones: archivo + producto, archivo + público objetivo, o producto + público objetivo + emoción + acción deseada." # Mostrar los emails generados if submit: # Validate inputs using the new function is_valid, error_message = validate_inputs( file_content, product, target_audience, emotion, desired_action ) if is_valid and selected_formula: try: # Use spinner within col2 context with col2: with st.spinner("Creando los emails..."): # Update the function call to include brand_tone generated_emails = generate_emails( target_audience, product, temperature, selected_formula, selected_angle, file_content, image_parts, is_image, emotion, desired_action, creative_idea, brand_tone # Add the brand tone parameter ) # Check if the response starts with "Error:" if generated_emails.startswith("Error:"): st.error(generated_emails) else: # Clean the response text to remove extra spaces generated_emails = clean_response_text(generated_emails) # Get current timestamp for the filename timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") # Add a download button ABOVE the results st.download_button( label="DESCARGAR EMAILS", data=generated_emails, file_name=f"emails_persuasivos_{timestamp}.txt", mime="text/plain", key="download_top" # Unique key for the top button ) # Display the generated emails in col2 (still within col2 context) st.markdown(f"""

Tus emails persuasivos:

{generated_emails}

""", unsafe_allow_html=True) # Download button at the bottom (same as before but with a different key) st.download_button( label="DESCARGAR EMAILS", data=generated_emails, file_name=f"emails_persuasivos_{timestamp}.txt", mime="text/plain", key="download_bottom" # Unique key for the bottom button ) except Exception as e: col2.error(f"Error: {str(e)}") else: if not selected_formula: col2.error("Por favor selecciona una fórmula.") else: col2.error(error_message)