Update app.py

app.py

@@ -16,7 +16,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Fórmulas con ejemplos y explicaciones
 # headline_formulas dictionary has been moved to formulas/headline_formulas.py
 
-def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -71,39 +71,6 @@ IMPORTANT:
 
     # Iniciar el prompt con las instrucciones del sistema
     headlines_instruction = f"{system_prompt}\n\n"
-
-    # Añadir contenido del archivo si existe
-    if file_content:
-        headlines_instruction += f"""
-REFERENCE CONTENT:
-Carefully analyze the following content as a reference for generating headlines:
-{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 headlines to resonate with the provided content
-
-IMPORTANT COMBINATIONS:
-"""
-        # Updated conditions for specific input combinations
-        if product and not target_audience:
-            headlines_instruction += f"""- FILE + PRODUCT: You have a reference document and product ({product}). Create headlines that highlight this specific product's benefits and features using insights from the document. Extract audience information from the document to better target the headlines.
-"""
-        elif target_audience and not product:
-            headlines_instruction += f"""- FILE + TARGET AUDIENCE: You have a reference document and target audience ({target_audience}). Create headlines 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:
-            headlines_instruction += f"""- PRODUCT + TARGET AUDIENCE: You have both product ({product}) and target audience ({target_audience}). Create headlines that connect this specific product with this specific audience, using insights from the document to strengthen the connection.
-"""
-        
-        headlines_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":
@@ -199,6 +166,7 @@ Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura
         )
         response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados, inspirándote en la imagen proporcionada.")
     else:
+        # Enviar el mensaje al modelo
         chat_session = model.start_chat(
             history=[
                 {
@@ -208,8 +176,8 @@ Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura
             ]
         )
         response = chat_session.send_message("Genera los titulares siguiendo exactamente el estilo de los ejemplos mostrados.")
-    
-    return response.text
+        
+        return response.text
 
 # Configurar la interfaz de usuario con Streamlit
 st.set_page_config(page_title="Enchanted Hooks", layout="wide")
@@ -257,65 +225,6 @@ with col1:
             "Selecciona el ángulo para tus titulares",
             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 al procesar la imagen: {str(e)}")
-                    is_image = False
 
     selected_formula = headline_formulas[selected_formula_key]
 
@@ -324,20 +233,12 @@ with col1:
 
 # Mostrar los titulares generados
 if submit:
-    # Check if we have a valid combination of inputs
-    has_file = 'file_content' in locals() and file_content.strip() != ""
+    # Check if we have valid inputs
     has_product = product.strip() != ""
     has_audience = target_audience.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)
-    )
+    # Valid combination: Product + Audience
+    valid_inputs = has_product and has_audience
     
     if valid_inputs and selected_formula:
         try:
@@ -347,10 +248,7 @@ if submit:
                 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
+                selected_angle
             )
             col2.markdown(f"""
                 <div class="results-container">
@@ -364,4 +262,4 @@ if submit:
         if not selected_formula:
             col2.error("Por favor, selecciona una fórmula.")
         else:
-            col2.error("Por favor, proporciona al menos una de estas combinaciones: archivo + producto, archivo + público objetivo, o producto + público objetivo.")
+            col2.error("Por favor, proporciona el público objetivo y el producto.")