Spaces:
Sleeping
Sleeping
| 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 # Updated import statement | |
| from angles import angles | |
| # 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(number_of_emails, target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False, emotion="", desired_action=""): | |
| # 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, | |
| ) | |
| # Angle dictionaries have been moved to angles/angle_data.py | |
| # 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 have helped many marketers before me persuade their clients through emotional email sequences. | |
| Your task is to create email sequences that make my [buyer persona] feel [emotion] about my [product/service] and convince them to register/take [desired action]. | |
| FORMAT RULES: | |
| - Each email must have a clear and attractive subject line | |
| - Include personalized greeting | |
| - The email body must be persuasive and emotional | |
| - Include a clear call to action | |
| - Add a professional signature | |
| - Separate each email with a dividing line | |
| IMPORTANT ANGLE INSTRUCTIONS: | |
| - The selected angle MUST be applied to EACH email | |
| - The angle modifies HOW the message is expressed, not its structure | |
| - Think of the angle as a "tone overlay" on the content | |
| - The formula provides the structure, the angle provides the style | |
| - Both must work together seamlessly | |
| FORMAT EXAMPLE: | |
| --- | |
| SUBJECT: [Attractive subject line] | |
| [Email body with persuasive and emotional content] | |
| [Clear call to action] | |
| [Signature] | |
| --- | |
| 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 | |
| - Follow the selected angle style while maintaining the structure""" | |
| # Iniciar el prompt con las instrucciones del sistema | |
| email_instruction = f"{system_prompt}\n\n" | |
| # 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. | |
| """ | |
| # Añadir instrucciones de ángulo solo si no es "NINGUNO" | |
| if selected_angle != "NINGUNO": | |
| email_instruction += 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"]: | |
| email_instruction += f"- {example}\n" | |
| # Dentro de la función, actualizar el prompt para incluir emoción y acción deseada | |
| email_instruction += ( | |
| f"\nYour task is to create {number_of_emails} persuasive emails for {target_audience} " | |
| f"that evoke {emotion} and convince them to {desired_action} about {product}. " | |
| ) | |
| if selected_angle != "NINGUNO": | |
| email_instruction += f"IMPORTANT: Each email MUST follow the {selected_angle} angle clearly and consistently.\n\n" | |
| email_instruction += ( | |
| f"Avoid obvious mentions of {product} and focus on generating genuine interest" | |
| ) | |
| if selected_angle != "NINGUNO": | |
| email_instruction += f" using the selected angle" | |
| email_instruction += ".\n\n" | |
| email_instruction += ( | |
| f"IMPORTANT: Carefully study these examples of the selected formula. " | |
| f"Each example represents the style and structure to follow" | |
| ) | |
| if selected_angle != "NINGUNO": | |
| email_instruction += f", adapted to the {selected_angle} angle" | |
| email_instruction += ":\n\n" | |
| # Agregar 5 ejemplos aleatorios de la fórmula | |
| random_examples = random.sample(selected_formula['examples'], min(5, len(selected_formula['examples']))) | |
| email_instruction += "FORMULA EXAMPLES TO FOLLOW:\n" | |
| for i, example in enumerate(random_examples, 1): | |
| email_instruction += f"{i}. {example}\n" | |
| 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" | |
| email_instruction += f"FORMULA TO FOLLOW:\n{selected_formula['description']}\n\n" | |
| # CORRECTO (con indentación): | |
| if selected_angle != "NINGUNO": | |
| email_instruction += f""" | |
| FINAL REMINDER: | |
| 1. Follow the structure of the selected formula | |
| 2. Apply the angle as a "style layer" | |
| 3. Maintain coherence between formula and angle | |
| 4. Ensure each email reflects both elements | |
| GENERATE NOW: | |
| Create {number_of_emails} emails that faithfully follow the style and structure of the examples shown. | |
| """ | |
| else: | |
| email_instruction += f""" | |
| GENERATE NOW: | |
| Create {number_of_emails} emails that faithfully follow the style and structure of the examples shown. | |
| """ | |
| # Modificar la forma de enviar el mensaje según si hay imagen o no | |
| message_parts = [email_instruction] | |
| # Add the image to the message parts if it exists | |
| if is_image and image_parts: | |
| message_parts.append(image_parts) | |
| instruction_text = "Generate the emails in Spanish following exactly the style of the examples shown, drawing inspiration from the provided image." | |
| else: | |
| instruction_text = "Generate the emails in Spanish following exactly the style of the examples shown." | |
| # Common instruction for both cases | |
| instruction_text += " Do not include explanations, only the emails. IMPORTANT: Do not include greetings like 'Hello [Name]' and make sure that the postscripts (P.D.) are smaller and more discrete than the main body of the email, using a lighter format." | |
| # Create the chat session with the message parts | |
| 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 | |
| # Configurar la interfaz de usuario con Streamlit | |
| st.set_page_config(page_title="Email Composer", layout="wide") | |
| # Leer el contenido del archivo manual.md | |
| with open("manual.md", "r", encoding="utf-8") as file: | |
| manual_content = file.read() | |
| # Mostrar el contenido del manual en el sidebar | |
| st.sidebar.markdown(manual_content) | |
| # Load CSS from file | |
| with open("styles/main.css", "r") as f: | |
| css = f.read() | |
| # Apply the CSS | |
| st.markdown(f"<style>{css}</style>", unsafe_allow_html=True) | |
| # Centrar el título y el subtítulo | |
| st.markdown("<h1 style='text-align: center;'>Generador de Emails</h1>", unsafe_allow_html=True) | |
| st.markdown("<h4 style='text-align: center;'>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.</h4>", unsafe_allow_html=True) | |
| # Crear columnas | |
| col1, col2 = st.columns([1, 2]) | |
| # 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") | |
| number_of_emails = st.selectbox("Número de Emails", options=[1, 2, 3, 4, 5], index=2) | |
| # Moved the submit button here, right after number_of_emails | |
| submit = st.button("Generar Emails") | |
| # Crear un único acordeón para fórmula, creatividad y ángulo | |
| with st.expander("Personaliza tus emails"): | |
| temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1) | |
| emotion = st.selectbox( | |
| "¿Qué emoción quieres evocar?", | |
| options=["Curiosidad", "Miedo", "Esperanza", "Entusiasmo", "Confianza", "Urgencia"] | |
| ) | |
| desired_action = st.text_input("Acción deseada", placeholder="Ejemplo: Registrarse para una prueba gratuita") | |
| selected_formula_key = st.selectbox( | |
| "Selecciona una fórmula para tus emails", | |
| options=list(email_formulas.email_formulas.keys()) # Updated reference | |
| ) | |
| # Automatically use the keys from the angles dictionary | |
| # Make sure "NINGUNO" appears first, then the rest alphabetically | |
| 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 | |
| ) | |
| # Añadir cargador de archivos dentro del acordeón | |
| 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: | |
| file_type = uploaded_file.name.split('.')[-1].lower() | |
| # Manejar archivos de texto | |
| if file_type in ['txt', 'pdf', 'docx']: | |
| 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 | |
| selected_formula = email_formulas.email_formulas[selected_formula_key] # Updated reference | |
| # Removed the submit button from here | |
| # Mostrar los emails generados | |
| if submit: | |
| # Check if we have a valid combination of inputs | |
| has_file = 'file_content' in locals() and file_content.strip() != "" | |
| has_product = product.strip() != "" | |
| has_audience = target_audience.strip() != "" | |
| has_emotion = 'emotion' in locals() and emotion.strip() != "" | |
| has_action = 'desired_action' in locals() and desired_action.strip() != "" | |
| # Valid combinations: | |
| # 1. File + Product (no audience needed) | |
| # 2. File + Audience (no product needed) | |
| # 3. Product + Audience (traditional way) | |
| valid_inputs = ( | |
| (has_file and has_product) or | |
| (has_file and has_audience) or | |
| (has_product and has_audience and has_emotion and has_action) | |
| ) | |
| if valid_inputs and selected_formula: | |
| try: | |
| # Update the function call to include emotion and desired_action | |
| generated_emails = generate_emails( | |
| number_of_emails, | |
| target_audience, | |
| product, | |
| temperature, | |
| selected_formula, | |
| selected_angle, | |
| file_content if 'file_content' in locals() else "", | |
| image_parts if 'image_parts' in locals() else None, | |
| is_image if 'is_image' in locals() else False, | |
| emotion, | |
| desired_action | |
| ) | |
| # Remove the inline CSS since we've moved it to main.css | |
| col2.markdown(f""" | |
| <div class="results-container"> | |
| <h4>Tus emails persuasivos:</h4> | |
| <p>{generated_emails}</p> | |
| </div> | |
| """, unsafe_allow_html=True) | |
| except ValueError as e: | |
| col2.error(f"Error: {str(e)}") | |
| else: | |
| if not selected_formula: | |
| col2.error("Por favor selecciona una fórmula.") | |
| elif not (has_emotion and has_action): | |
| col2.error("Por favor especifica la emoción que quieres evocar y la acción deseada.") | |
| else: | |
| col2.error("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.") | |