Update app.py

app.py

@@ -1,51 +1,30 @@
 import streamlit as st
-from gradio_client import Client
-from PIL import Image
-import io
+from together import Together
 
-# Инициализация клиента
-client = Client("Serg4451D/FLUX.1-Dev-Serverless-darn-enhanced-prompt")
+# Укажите ваш API-ключ здесь
+API_KEY = "72f5f1157eb672d254c3dfff5eab724e216d04aa1ee4e905f1f1c8c98c75c249"
 
-def generate_image(prompt, steps):
-    # Вызов API для генерации изображения
-    result = client.predict(
-        prompt=prompt,
-        is_negative="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos",
-        steps=steps,
-        cfg_scale=7,
-        sampler="DDIM",
-        seed=-1,
-        strength=0.7,
-        huggingface_api_key="",
-        use_dev=False,
-        enhance_prompt_style="",
-        enhance_prompt_option=False,
-        nemo_enhance_prompt_style="",
-        use_mistral_nemo=False,
-        api_name="/query"
-    )
-    
-    # Обработка результата
-    if isinstance(result, tuple) and len(result) > 0:
-        image_data = result[0]
-        if isinstance(image_data, (bytes, bytearray)):
-            return Image.open(io.BytesIO(image_data))
-        elif isinstance(image_data, str):
-            # Если результат - путь к файлу
-            return Image.open(image_data)
-    
-    raise ValueError("Неподдерживаемый формат результата от API")
+client = Together(api_key=API_KEY)
 
-# Streamlit UI
 st.title("Генератор изображений")
 
-prompt = st.text_input("Введите промпт:", "cat")
-steps = st.radio("Выберите количество шагов:", [1, 4, 10])
+# Поле для ввода текста
+prompt = st.text_input("Введите текст для генерации изображения:")
 
 if st.button("Сгенерировать изображение"):
-    try:
-        with st.spinner("Генерация изображения..."):
-            generated_image = generate_image(prompt, steps)
-        st.image(generated_image, caption="Сгенерированное изображение")
-    except Exception as e:
-        st.error(f"Произошла ошибка при генерации изображения: {str(e)}")
+    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("Пожалуйста, введите текст для генерации изображения.")