import requests
import streamlit as st

account_id = st.secrets["CLOUDFLARE_ACCOUNT_ID"]
api_token = st.secrets["CLOUDFLARE_API_TOKEN"]

st.set_page_config(
    page_title="Texto a Imagen",
    page_icon="media/icon.png"
)

" ## Texto a Imagen"
"""
---
"""

with st.form("text_to_image"):
    model = st.selectbox(
        "Elige un modelo",
        options=(
            "@cf/lykon/dreamshaper-8-lcm",
            "@cf/bytedance/stable-diffusion-xl-lightning",
            "@cf/stabilityai/stable-diffusion-xl-base-1.0",
        ),
    )
    prompt = st.text_area(label="Describe de manera detallada la imagen que deseas generar")
    submitted = st.form_submit_button("Generar")
    if submitted:
        headers = {
            "Authorization": f"Bearer {api_token}",
        }
        with st.spinner("Generando..."):
            url = f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/{model}"
            response = requests.post(
                url,
                headers=headers,
                json={"prompt": prompt},
            )
            st.image(response.content, caption=prompt)