Update app.py

app.py

@@ -201,64 +201,73 @@ with col2:
             else:
                 product_name = "Producto/Servicio"
                 
-            # Get the instruction using the formula
-            instruction = create_offer_instruction(
-                avatar_description=avatar_description,
-                product_name=product_name,
-                selected_formula_name=st.session_state.formula_type
-            )
-            
-            # Removing the creative hook instruction for Oferta Dorada
-            
-            # Add instruction for generating benefit bullets based on the promise
-            bullet_content = None
-            if hasattr(st.session_state, 'file_content') and st.session_state.input_type in ["file", "both"]:
-                bullet_content = st.session_state.file_content
+                # Get the instruction using the formula
+                instruction = create_offer_instruction(
+                    avatar_description=avatar_description,
+                    product_name=product_name,
+                    selected_formula_name=st.session_state.formula_type
+                )
                 
-            instruction += create_bullet_instruction(
-                avatar_description=avatar_description,
-                product_name=product_name,
-                uploaded_content=bullet_content
-            )
-            
-            # Add instruction for generating bonuses that complement the offer
-            instruction += create_bonus_instruction(
-                avatar_description=avatar_description,
-                product_name=product_name,
-                selected_formula_name=st.session_state.formula_type
-            )
-            
-            # Add additional context based on input type
-            if st.session_state.input_type == "manual":
-                additional_context = f"""
-                Additional information:
-                Skills: {st.session_state.skills}
-                """
-                instruction += additional_context
-            elif st.session_state.input_type == "file":
-                additional_context = f"""
-                Additional information from file:
-                {st.session_state.file_content}
-                """
-                instruction += additional_context
-            elif st.session_state.input_type == "both":
-                additional_context = f"""
-                Additional information:
-                Skills: {st.session_state.skills}
-                File content: {st.session_state.file_content}
-                """
-                instruction += additional_context
+                # Add instruction for generating benefit bullets based on the promise
+                bullet_content = None
+                if hasattr(st.session_state, 'file_content') and st.session_state.input_type in ["file", "both"]:
+                    bullet_content = st.session_state.file_content
+                
+                instruction += create_bullet_instruction(
+                    avatar_description=avatar_description,
+                    product_name=product_name,
+                    uploaded_content=bullet_content
+                )
                 
-            try:
-                generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
+                # Add instruction for generating bonuses that complement the offer
+                instruction += create_bonus_instruction(
+                    avatar_description=avatar_description,
+                    product_name=product_name,
+                    selected_formula_name=st.session_state.formula_type
+                )
                 
-                if "image" in st.session_state.input_type:
-                    response = model.generate_content([instruction, st.session_state.image_parts[0]], generation_config=generation_config)
-                else:
-                    response = model.generate_content(instruction, generation_config=generation_config)
+                # Add additional context based on input type
+                if st.session_state.input_type == "manual":
+                    additional_context = f"""
+                    Additional information:
+                    Skills: {st.session_state.skills}
+                    """
+                    instruction += additional_context
+                elif st.session_state.input_type == "file":
+                    additional_context = f"""
+                    Additional information from file:
+                    {st.session_state.file_content}
+                    """
+                    instruction += additional_context
+                elif st.session_state.input_type == "both":
+                    additional_context = f"""
+                    Additional information:
+                    Skills: {st.session_state.skills}
+                    File content: {st.session_state.file_content}
+                    """
+                    instruction += additional_context
+                    
+                try:
+                    generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
+                    
+                    if "image" in st.session_state.input_type:
+                        response = model.generate_content([instruction, st.session_state.image_parts[0]], generation_config=generation_config)
+                    else:
+                        response = model.generate_content(instruction, generation_config=generation_config)
+                    
+                    # Get the response text
+                    response_text = response.text
+                    
+                    # Special processing for Oferta Dorada to remove any markdown or formatting that might cause the box
+                    if st.session_state.formula_type == "Oferta Dorada":
+                        # Remove any potential markdown formatting that might be causing the box
+                        response_text = response_text.replace("```", "").replace("`", "")
+                        # Remove any HTML tags that might be in the response
+                        response_text = response_text.replace("<div>", "").replace("</div>", "")
+                        response_text = response_text.replace("<p>", "").replace("</p>", "")
                     
-                st.session_state.offer_result = response.text
-                st.session_state.generated = True  # Mark as generated
+                    st.session_state.offer_result = response_text
+                    st.session_state.generated = True  # Mark as generated
                 
             except Exception as e:
                 st.error(f'Ocurrió un error: {str(e)}')