import streamlit as st
from together import Together

# Укажите ваш API-ключ здесь
API_KEY = "72f5f1157eb672d254c3dfff5eab724e216d04aa1ee4e905f1f1c8c98c75c249"

client = Together(api_key=API_KEY)

st.title("Генератор изображений")

# Поле для ввода текста
prompt = st.text_input("Введите текст для генерации изображения:")

if st.button("Сгенерировать изображение"):
    if prompt:
        response = client.images.generate(
            prompt=prompt,
            model="black-forest-labs/FLUX.1-schnell",
            width=1024,
            height=1024,
            steps=12,
            n=1,
            response_format="b64_json"
        )
        
        # Отображение сгенерированного изображения
        image_data = response.data[0].b64_json
        st.image(f"data:image/png;base64,{image_data}", caption="Сгенерированное изображение", use_column_width=True)
    else:
        st.warning("Пожалуйста, введите текст для генерации изображения.")