app.py

import streamlit as st
import google.generativeai as genai
import os
import time
from dotenv import load_dotenv
from styles import get_custom_css, get_response_html_wrapper
from formulas import offer_formulas
import PyPDF2
import docx
from PIL import Image
import io

# Set page to wide mode to use full width
st.set_page_config(layout="wide")

# Load environment variables
load_dotenv()

# Configure Google Gemini API
genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
model = genai.GenerativeModel('gemini-2.0-flash')

# Initialize session state variables if they don't exist
if 'submitted' not in st.session_state:
    st.session_state.submitted = False
if 'offer_result' not in st.session_state:
    st.session_state.offer_result = ""
if 'generated' not in st.session_state:
    st.session_state.generated = False

# Hide Streamlit menu and footer
st.markdown("""
<style>
#MainMenu {visibility: hidden;}
footer {visibility: hidden;}
header {visibility: hidden;}
</style>
""", unsafe_allow_html=True)

# Custom CSS
st.markdown(get_custom_css(), unsafe_allow_html=True)

# App title and description
st.markdown('<h1 style="text-align: center;">Great Offer Generator</h1>', unsafe_allow_html=True)
st.markdown('<h3 style="text-align: center;">Transform your skills into compelling offers!</h3>', unsafe_allow_html=True)

# Create two columns for layout - left column 40%, right column 60%
col1, col2 = st.columns([4, 6])

# Main input section in left column
with col1:
    # Keep only the manual input tab
    with st.container():
        # Input fields for user skills/product/service
        st.markdown("👍 Tus Habilidades")
        skills = st.text_area("", key="skills", height=70)
        
        # Input fields for target audience
        st.markdown("🎯 Producto/Servicio")
        target_audience = st.text_area("", key="target_audience", height=70)
        
        # Formula type selection
        st.markdown("📝 Tipo de Fórmula")
        formula_type = st.selectbox("", list(offer_formulas.keys()), key="formula_type")
        
        # Creativity level slider
        st.markdown("🌡️ Nivel de Creatividad")
        creativity = st.slider("", min_value=0.0, max_value=2.0, value=1.0, step=0.01, key="creativity")
        
        # Generate button - MOVED UP before the advanced configuration
        if st.button("Generar Oferta 🔥"):
            # Your existing button logic here
            pass
        
        # Advanced configuration - NOW AFTER the button
        with st.expander("⚙️ Configuración Avanzada"):
            # Your existing advanced configuration options here
            pass

# Results column
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
            
            Then, create a compelling and persuasive offer using the {st.session_state.formula_type} formula that speaks directly to this avatar.
            """
            
            if st.session_state.input_type == "manual":
                prompt = base_prompt + f"""
                Based on the following information:
                Skills: {st.session_state.skills}
                Product/Service: {st.session_state.product_service}
                """
            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.
                """
            elif st.session_state.input_type == "both":
                prompt = base_prompt + f"""
                Based on the following combined 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.
                """
            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}
                
                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)
                else:
                    response = model.generate_content(prompt, generation_config=generation_config)
                    
                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)}')
                st.session_state.submitted = False
    
    # Display results if we have an offer result
    if st.session_state.generated:
        # With this line that uses the wrapper:
        st.markdown(get_response_html_wrapper(st.session_state.offer_result), unsafe_allow_html=True)
        
        # Add a small space
        st.markdown('<div style="height: 15px;"></div>', unsafe_allow_html=True)
        
        col_download, col_empty = st.columns([8, 2])
        with col_download:
            st.download_button(
                label="Descargar Oferta",
                data=st.session_state.offer_result,
                file_name="oferta_generada.txt",
                mime="text/plain"
            )

# Footer
st.markdown('---')
st.markdown('Made with ❤️ by Jesús Cabrera')