import streamlit as st
from model import ImagePromptModel
import tempfile

st.set_page_config(page_title="🧠 Image to Prompt Generator", layout="centered")
st.title("🧠 Image-to-Prompt Model")

st.write("Upload an image to generate a creative caption using BLIP + Transformers.")

uploaded_file = st.file_uploader("📤 Upload Image", type=["jpg", "jpeg", "png"])

if uploaded_file:
    with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
        tmp_file.write(uploaded_file.read())
        tmp_path = tmp_file.name

    st.image(tmp_path, caption="Uploaded Image", use_column_width=True)

    model = ImagePromptModel()
    with st.spinner("Generating prompt..."):
        result = model.generate_prompt(tmp_path)

    st.success("✅ Prompt Generated!")
    st.write(f"**Prompt:** {result}")