Update app.py

app.py

@@ -20,6 +20,9 @@ load_dotenv()
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
 model = genai.GenerativeModel('gemini-2.0-flash')
 
+# Import the create_offer_instruction function from formulas
+from formulas import create_offer_instruction, offer_formulas
+
 # Initialize session state variables if they don't exist
 if 'submitted' not in st.session_state:
     st.session_state.submitted = False
@@ -50,7 +53,7 @@ col1, col2 = st.columns([4, 6])
 # Main input section in left column
 with col1:
     # Define the generate_offer function first
-    def generate_offer():
+    def handle_generate_button():  # Renamed to avoid conflict
         has_manual_input = bool(skills or product_service)
         has_file_input = bool(uploaded_file is not None and not is_image)
         has_image_input = bool(uploaded_file is not None and is_image)
@@ -101,7 +104,7 @@ with col1:
                                     help='Describe tu producto o servicio')
         
         # Generate button moved here - right after product/service
-        st.button('Generar Oferta 🎉', on_click=generate_offer)
+        st.button('Generar Oferta 🎉', on_click=handle_generate_button)  # Updated function name
     
     # Accordion for additional settings
     with st.expander('⚙️ Configuración Avanzada'):
@@ -183,91 +186,50 @@ with col1:
 with col2:
     if st.session_state.submitted and not st.session_state.generated:
         with st.spinner('Creando tu oferta perfecta...'):
-            base_prompt = f"""You are a professional copywriter specializing in creating irresistible offers. 
-            First, analyze the target audience/avatar based on the information provided:
-            Target Audience: {st.session_state.target_audience if hasattr(st.session_state, 'target_audience') and st.session_state.target_audience else 'General audience'}
-            
-            Consider their:
-            - Pain points and frustrations
-            - Desires and aspirations
-            - Current situation
-            - Emotional state
-            - Objections they might have
+            # Use the create_offer_instruction function to generate the prompt
+            avatar_description = st.session_state.target_audience if hasattr(st.session_state, 'target_audience') and st.session_state.target_audience else 'General audience'
             
-            Then, create a compelling and persuasive offer using the {st.session_state.formula_type} formula that speaks directly to this avatar.
-            """
+            # Determine product name based on input type
+            if hasattr(st.session_state, 'product_service') and st.session_state.product_service:
+                product_name = st.session_state.product_service
+            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
+            )
             
+            # Add additional context based on input type
             if st.session_state.input_type == "manual":
-                prompt = base_prompt + f"""
-                Based on the following information:
+                additional_context = f"""
+                Additional information:
                 Skills: {st.session_state.skills}
-                Product/Service: {st.session_state.product_service}
                 """
+                instruction += additional_context
             elif st.session_state.input_type == "file":
-                prompt = base_prompt + f"""
-                Based on the following information from the uploaded file:
-                File Content: {st.session_state.file_content}
-                """
-            elif st.session_state.input_type == "image":
-                prompt = base_prompt + f"""
-                Based on the image provided, create an offer that highlights the visual elements and appeals to the target audience.
+                additional_context = f"""
+                Additional information from file:
+                {st.session_state.file_content}
                 """
+                instruction += additional_context
             elif st.session_state.input_type == "both":
-                prompt = base_prompt + f"""
-                Based on the following combined information:
+                additional_context = f"""
+                Additional information:
                 Skills: {st.session_state.skills}
-                Product/Service: {st.session_state.product_service}
-                Additional Information from File: {st.session_state.file_content}
-                
-                Please consider both the manual input and file content to create a comprehensive offer.
+                File content: {st.session_state.file_content}
                 """
-            elif st.session_state.input_type == "manual_image":
-                prompt = base_prompt + f"""
-                Based on the following information and the image provided:
-                Skills: {st.session_state.skills}
-                Product/Service: {st.session_state.product_service}
+                instruction += additional_context
                 
-                Please analyze both the text information and the visual elements in the image to create a comprehensive offer.
-                """
-            elif st.session_state.input_type == "file_image":
-                prompt = base_prompt + f"""
-                Based on the following information from the uploaded file and the image provided:
-                File Content: {st.session_state.file_content}
-                
-                Please analyze both the file content and the visual elements in the image to create a comprehensive offer.
-                """
-            else:
-                prompt = base_prompt + f"""
-                Based on all the following information:
-                Skills: {st.session_state.skills}
-                Product/Service: {st.session_state.product_service}
-                Additional Information from File: {st.session_state.file_content}
-                
-                Please analyze the text information, file content, and the visual elements in the image to create the most comprehensive offer.
-                """
-                
-            prompt += f"""
-            Formula Description:
-            {offer_formulas[st.session_state.formula_type]["description"]}
-            
-            Examples of this formula:
-            {chr(10).join(offer_formulas[st.session_state.formula_type]["examples"])}
-            
-            Please create a professional, engaging, and irresistible offer that:
-            1. Highlights the value proposition based on the skills/product/service provided
-            2. Addresses the specific pain points of the analyzed avatar
-            3. Creates urgency and desire
-            4. STRICTLY follows the structure format of the examples above
-            
-            IMPORTANT: Provide ONLY the final offer text. Do not include any explanations, labels, formatting instructions, brackets, or call to action at the end."""
-            
             try:
                 generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
                 
                 if "image" in st.session_state.input_type:
-                    response = model.generate_content([prompt, st.session_state.image_parts[0]], generation_config=generation_config)
+                    response = model.generate_content([instruction, st.session_state.image_parts[0]], generation_config=generation_config)
                 else:
-                    response = model.generate_content(prompt, generation_config=generation_config)
+                    response = model.generate_content(instruction, generation_config=generation_config)
                     
                 st.session_state.offer_result = response.text
                 st.session_state.generated = True  # Mark as generated
@@ -295,4 +257,6 @@ with col2:
 
 # Footer
 st.markdown('---')
-st.markdown('Made with ❤️ by Jesús Cabrera')
+st.markdown('Made with ❤️ by Jesús Cabrera')
+
+# Remove the duplicate functions at the bottom