JeCabrera's picture
Update app.py
f0d2397 verified
raw
history blame
4.05 kB
import streamlit as st
import google.generativeai as genai
import os
from dotenv import load_dotenv
from styles import get_custom_css
# 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 = ""
# 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.title('🚀 Great Offer Generator')
st.markdown('''### Transform your skills into compelling offers!''')
# 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:
skills = st.text_area('💪 Your Skills', height=70,
help='List your key skills and expertise')
product_service = st.text_area('🎯 Product/Service', height=70,
help='Describe your product or service')
# Accordion for additional settings
with st.expander('⚙️ Advanced Settings'):
target_audience = st.text_area('👥 Target Audience', height=70,
help='Describe your ideal customer or client')
temperature = st.slider('🌡️ Creativity Level', min_value=0.0, max_value=2.0, value=0.7,
help='Higher values make the output more creative but less focused')
# Generate button with callback
def generate_offer():
if not skills or not product_service:
st.error('Please fill in both Skills and Product/Service fields')
else:
# Set submitted flag to True
st.session_state.submitted = True
# Store input values in session state
st.session_state.skills = skills
st.session_state.product_service = product_service
st.session_state.target_audience = target_audience
st.session_state.temperature = temperature
st.button('Generate Offer 🎉', on_click=generate_offer)
# Results column
with col2:
# Check if form has been submitted
if st.session_state.submitted:
with st.spinner('Creating your perfect offer...'):
prompt = f"""Based on the following information, create a compelling offer:
Skills: {st.session_state.skills}
Product/Service: {st.session_state.product_service}
Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
Please create a professional and engaging offer that highlights the value proposition
and appeals to the target audience. Include a clear call to action."""
try:
# Create generation config with temperature
generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
# Pass the generation config to generate_content
response = model.generate_content(prompt, generation_config=generation_config)
st.session_state.offer_result = response.text
st.success('✨ Your offer is ready!')
# Display result
st.markdown('### 📝 Generated Offer')
st.markdown(st.session_state.offer_result)
except Exception as e:
st.error(f'An error occurred: {str(e)}')
st.session_state.submitted = False
# Footer
st.markdown('---')
st.markdown('Made with ❤️ by Jesús Cabrera')