import streamlit as st
from PIL import Image


# Load images
logo_image = Image.open("img/fireworksai_logo.png")
parameters_1_image = Image.open("img/parameters_1.png")
parameters_2_image = Image.open("img/parameters_2.png")
parameters_3_image = Image.open("img/parameters_3.png")
parameters_4_image = Image.open("img/parameters_4.png")
card_with_message_1_image = Image.open("img/card_with_message_1.png")
card_with_message_2_image = Image.open("img/card_with_message_2.png")



# Fireworks Logo at the top
st.image(logo_image)

# Title of the app
st.title("🎨 Flux Holiday Magic: Custom Card Creator")


# Description using Streamlit text and markdown
st.markdown("""
Welcome to the ultimate holiday card design experience! 🎄❄️ 
            
\n Powered by **Flux models** through **Fireworks API**, this app lets you effortlessly create stunning, personalized holiday cards with just a few clicks. Here’s what you can do:
""")

# Features list with emojis
st.markdown("""
- **Upload Your Image**: Choose any photo as the base for your holiday card.
- **Crop & Preview**: Fine-tune the area of your image to fit perfectly within the card.
- **Holiday Borders**: Choose from festive prompts like snowflakes, ornaments, and cozy fireplace scenes to frame your photo.
- **Add Custom Messages**: Personalize your card with a message, choosing from a variety of fonts and adjustable text sizes.
- **Advanced Customization**: Take control of parameters like ControlNet conditioning, guidance scale, inference steps, and more for tailored results.
- **Save & Share**: Download your final creations or save them for later in a handy ZIP file.
            
And be sure to share with us on social!
            
![Holiday Vibes](https://i.giphy.com/media/v1.Y2lkPTc5MGI3NjExMXo5Y3RhOGNzaTkzZHFmandsZmF0MW9jNHN4M3VjMjdpOGhyYnRxMSZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9Zw/637gfK5GJdv552H9ab/giphy.gif)
""")

# Call to action
st.markdown("""
Get started now and add a magical touch to your holiday greetings! 🎅🌟
""")

st.divider()
st.subheader("Experiment with Flux to figure out the perfect look for your holiday card")
st.markdown(""" ![For science!](https://i.giphy.com/media/v1.Y2lkPTc5MGI3NjExbGwwdGh4czc3bnU3czNzaWRhaXl4YXYzdm53Nmw1OXhpd2E4emFuYSZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9Zw/RdJIM4Uesg37eN1ZNL/giphy.gif)""")
col1, col2 = st.columns(2)

with col1:
    st.image(parameters_1_image)
    st.image(parameters_3_image)

with col2:
    st.image(parameters_2_image)
    st.image(parameters_4_image)



st.divider()
st.subheader("Customize your holiday card with a special message")
st.markdown(""" ![Special message, so special](https://i.giphy.com/media/v1.Y2lkPTc5MGI3NjExenVkdmZzZTdxNjIxcDk2b2NrdXUxbDBneTNqdHUyamFxNTQzd2dleiZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9Zw/qDJSK8oxEkBV0XJE7B/giphy.gif)""")
col3, col4 = st.columns(2)
with col3:
    st.image(card_with_message_1_image)
with col4:
    st.image(card_with_message_2_image)