Update app.py

app.py

@@ -4,8 +4,13 @@ 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 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()
@@ -139,26 +144,24 @@ SUCCESSFUL EXAMPLES OF THE {selected_angle} ANGLE:
 
     # 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.\n\n"
+        email_instruction += f"IMPORTANT: Each email MUST follow the {selected_angle} angle clearly and consistently."
 
-    email_instruction += (
-        f"Avoid obvious mentions of {product} and focus on generating genuine interest"
-    )
-
-    if angle_instructions:
-        email_instruction += f" using the selected angle"
-
-    email_instruction += ".\n\n"
+    email_instruction += "\n\n"
 
-    # Agregar ejemplos de la fórmula
-    examples_to_use = selected_formula['examples'][:min(2, len(selected_formula['examples']))]  # Reduced to only 2 examples
+    # 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 STUDY (ONLY FOR STRUCTURE, NOT CONTENT):\n"
-    for i, example in enumerate(examples_to_use, 1):
+    email_instruction += "FORMULA EXAMPLES TO FOLLOW:\n"
+    for i, example in enumerate(random_examples, 1):
         email_instruction += f"{i}. {example}\n"
-        
-    # Añadir advertencia crítica con emoji para llamar la atención
-    email_instruction += "\n⚠️ WARNING: DO NOT COPY THESE EXAMPLES. Create completely original content with different stories, scenarios and language. The examples are ONLY for understanding the structure.\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"
@@ -166,59 +169,73 @@ SUCCESSFUL EXAMPLES OF THE {selected_angle} ANGLE:
     email_instruction += f"FORMULA TO FOLLOW:\n{selected_formula['description']}\n\n"
 
     # Consolidar las instrucciones finales en un solo bloque
-    final_instructions = [
-        "Follow the logical structure but not the exact wording",
-        "Be creative and original with your content",
-        "Maintain the sequence flow between emails"
-    ]
+    final_reminder = ["Follow the structure of the selected formula"]
     
-    # Añadir instrucciones específicas para el ángulo si es necesario
+    # Add angle-specific instructions only if an angle is selected
     if selected_angle != "NINGUNO":
-        final_instructions.extend([
+        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, instruction in enumerate(final_instructions, 1):
-        email_instruction += f"{i}. {instruction}\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"
     
-    # Simplify the final instruction to clearly specify 5 emails
-    email_instruction += f"\nGENERATE NOW:\nCreate 5 CREATIVE emails that follow the logical structure but with original content.\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_text = "Generate the emails in Spanish following only the STRUCTURE of the examples, but with completely original content, drawing inspiration from the provided image."
-    else:
-        instruction_text = "Generate the emails in Spanish following only the STRUCTURE of the examples, but with completely original content and stories."
-    
-    # Simplificar las instrucciones finales para evitar redundancia
-    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."
+        instruction_components.append("drawing inspiration from the provided image.")
     
-    # Create the chat session with the message parts
-    chat_session = model.start_chat(
-        history=[
-            {
-                "role": "user",
-                "parts": message_parts,
-            },
-        ]
-    )
+    # Add final instruction
+    instruction_components.append("Do not include explanations, only the emails.")
     
-    # Enviar el mensaje con las instrucciones
-    response = chat_session.send_message(instruction_text)
+    # Join all instruction components with proper spacing
+    instruction_text = " ".join(instruction_components)
     
-    return response.text
+    # 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"""
-    import re
     # Replace multiple newlines with just two
     text = re.sub(r'\n{3,}', '\n\n', text)
     # Remove leading/trailing whitespace
@@ -243,18 +260,24 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # 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)
+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
-with open("styles/main.css", "r") as f:
-    css = f.read()
-    
-# Apply the CSS
-st.markdown(f"<style>{css}</style>", unsafe_allow_html=True)
+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"<style>{css}</style>", unsafe_allow_html=True)
+else:
+    st.warning("CSS file not found.")
 
 # Centrar el título y el subtítulo
 st.markdown("<h1 style='text-align: center;'>Generador de Emails</h1>", unsafe_allow_html=True)
@@ -263,6 +286,11 @@ st.markdown("<h4 style='text-align: center;'>Transforma tu marketing con emails
 # 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")
@@ -369,26 +397,43 @@ with col1:
     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() != ""
+# 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' in locals() and emotion.strip() != ""
-    has_action = 'desired_action' in locals() and desired_action.strip() != ""
+    has_emotion = emotion.strip() != "" if emotion else False
+    has_action = desired_action.strip() != "" if desired_action else False
     
-    # 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)
+    # 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 valid_inputs and selected_formula:
+    if is_valid and selected_formula:
         try:
             # Use spinner within col2 context
             with col2:
@@ -400,44 +445,43 @@ if submit:
                         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,
+                        file_content,
+                        image_parts,
+                        is_image,
                         emotion,
                         desired_action,
                         creative_idea  # Add the creative idea parameter
                     )
                     
-                    # Clean the response text to remove extra spaces
-                    generated_emails = clean_response_text(generated_emails)
-                
-                # Display the generated emails in col2 (still within col2 context)
-                st.markdown(f"""
-                    <div class="results-container">
-                        <h4>Tus emails persuasivos:</h4>
-                        <p>{generated_emails}</p>
-                    </div>
-                """, unsafe_allow_html=True)
-                
-                # Add download button (styling is in styles/main.css)
-                import datetime
-                
-                # Get current timestamp for the filename
-                timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-                
-                # Download button
-                st.download_button(
-                    label="DESCARGAR EMAILS",
-                    data=generated_emails,
-                    file_name=f"emails_persuasivos_{timestamp}.txt",
-                    mime="text/plain"
-                )
-        except ValueError as e:
+                    # 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)
+                    
+                        # Display the generated emails in col2 (still within col2 context)
+                        st.markdown(f"""
+                            <div class="results-container">
+                                <h4>Tus emails persuasivos:</h4>
+                                <p>{generated_emails}</p>
+                            </div>
+                        """, unsafe_allow_html=True)
+                        
+                        # Get current timestamp for the filename
+                        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+                        
+                        # Download button
+                        st.download_button(
+                            label="DESCARGAR EMAILS",
+                            data=generated_emails,
+                            file_name=f"emails_persuasivos_{timestamp}.txt",
+                            mime="text/plain"
+                        )
+        except Exception 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.")
+            col2.error(error_message)