# import streamlit as st
# from PIL import Image
# from transformers import pipeline
# import random

# # --- Use st.cache_resource to cache the models ---
# @st.cache_resource
# def load_caption_generator():
#     return pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")

# @st.cache_resource
# def load_poem_generator():
#     return pipeline("text2text-generation", model="google/flan-t5-base")

# # Load the models once and store them in memory
# caption_generator = load_caption_generator()
# poem_generator = load_poem_generator()

# # --- Your original app code follows ---
# st.set_page_config(...)
# # ... rest of your code ...



import streamlit as st
from PIL import Image
from transformers import pipeline
import random

# -------------------- PAGE CONFIG --------------------
st.set_page_config(
    page_title="Whispers of Images",
    page_icon="https://img.icons8.com/?size=100&id=UeTjKUNmOxts&format=png&color=000000",
    layout="centered",
    initial_sidebar_state="expanded"
)

# -------------------- CUSTOM DARK THEME CSS --------------------
st.markdown(
    """
    <style>
    /* Background Gradient */
    [data-testid="stAppViewContainer"] {
        background: linear-gradient(135deg, #0f2027 0%, #203a43 50%, #2c5364 100%);
        color: #f5f5f5;
    }



    /* Main Title */
    .title {
        font-size: 44px;
        color: #ffffff;
        text-align: center;
        font-family: 'Georgia', serif;
        margin-bottom: 5px;
        text-shadow: 1px 1px 5px rgba(0,0,0,0.5);
    }

    /* Subtitle */
    .subtitle {
        font-size: 18px;
        text-align: center;
        color: #d1d1d1;
        margin-bottom: 20px;
        font-style: italic;
    }

    /* Poem Card Styling */
    .poem-card {
        background-color: rgba(255, 255, 255, 0.05);
        padding: 20px;
        border-radius: 15px;
        box-shadow: 0 4px 20px rgba(0, 0, 0, 0.5);
        margin-top: 20px;
        border: 1px solid rgba(255, 255, 255, 0.2);
    }
    .poem-title {
        color: #ffda79;
        font-family: Georgia, serif;
        font-size: 24px;
        margin-bottom: 10px;
    }
    .poem-text {
        font-size: 18px;
        font-style: italic;
        color: #f1f1f1;
        white-space: pre-line;
    }

    /* Info Box Override */
    [data-testid="stInfo"] {
        background-color: rgba(255, 255, 255, 0.05);
        color: #f1f1f1;
        border: 1px solid rgba(255, 255, 255, 0.2);
    }

    /* Footer */
    .footer {
        position: fixed;
        left: 0;
        bottom: 0;
        width: 100%;
        background-color: #101820;
        color: #fff;
        text-align: center;
        padding: 10px;
        font-size: 14px;
        font-family: 'Trebuchet MS', sans-serif;
        box-shadow: 0 -2px 5px rgba(0,0,0,0.3);
    }

    /* Surprise Button Styling */
    div.stButton > button {
        background-color: #1a1f2e;
        color: white;
        font-size: 16px;
        border-radius: 8px;
        padding: 10px 20px;
        border: none;
        box-shadow: 0px 4px 10px rgba(0,0,0,0.3);
        transition: 0.3s;
    }
    div.stButton > button:hover {
        background-color: #ff92e3;
        box-shadow: 0px 6px 14px rgba(0,0,0,0.5);
    }
    </style>
    """,
    unsafe_allow_html=True
)

# -------------------- APP TITLE --------------------
if "title" not in st.session_state:
    st.session_state.title = "Image to Poetry Generator"

st.markdown(f"<h1 class='title'>{st.session_state.title}</h1>", unsafe_allow_html=True)
st.markdown(
    "<p class='subtitle'>Turn your pictures into soulful, rhyming verses ✍️✨</p>",
    unsafe_allow_html=True
)

# -------------------- SURPRISE ME BUTTON --------------------
if st.button("✨ Surprise Me ✨"):
    st.session_state.title = random.choice([
        "Pixels to Poetry 🌿",
        "When Images Whisper 🎭",
        "Canvas of Words ✍️",
        "Visual Rhymes 🎨",
        "Poetry in Pixels 🪄"
    ])
    st.experimental_rerun()

# -------------------- FILE UPLOAD --------------------
uploaded_file = st.file_uploader(
    "Choose an image...",
    type=["jpg", "jpeg", "png"]
)

# -------------------- POEM GENERATION --------------------
if uploaded_file is not None:
    image = Image.open(uploaded_file)
    st.image(image, caption="Uploaded Image", use_column_width=True)
    st.markdown("---")

    with st.spinner("Generating poetry... ✍️✨"):
        # Image captioning
        caption_generator = pipeline(
            "image-to-text",
            model="Salesforce/blip-image-captioning-base"
        )
        caption_result = caption_generator(image)
        image_caption = caption_result[0]['generated_text']

        # Poem generation
        poem_prompt = f"Describe this scene in the form of a short, rhyming, emotional poem. Be artistic and creative, avoid repeating words.:\n\n{image_caption}\n\nPoem:"

        poem_generator = pipeline(
            "text2text-generation",
            model="google/flan-t5-base"
        )
        poem_result = poem_generator(
            poem_prompt,
            max_length=150,
            num_return_sequences=1
        )
        generated_poem = poem_result[0]['generated_text']
        final_poem = generated_poem.replace(poem_prompt, "").strip()

    # -------------------- DISPLAY RESULTS --------------------
    st.subheader("Image Description")
    st.info(image_caption)

    st.markdown(
        f"""
        <div class="poem-card">
            <h3 class="poem-title">Generated Poem 🪄</h3>
            <p class="poem-text">{final_poem}</p>
        </div>
        """,
        unsafe_allow_html=True
    )
    st.balloons()

# -------------------- FOOTER --------------------
st.markdown(
    """
    <div class="footer">
        <img src="https://img.icons8.com/?size=100&id=blwqK79LWfZz&format=png&color=000000" style="vertical-align: middle; margin-right:5px;">
        <span>Turning pixels into poetry, beautifully</span>
      
    </div>
    """,
    unsafe_allow_html=True
)