import gradio as gr
import requests
import os

# API ссылка
url = os.getenv("URL")

# Функция для отправки запроса
def render(prompt, negative_prompt, output_format, output_size, style, lighting, background, camera_position, camera_angle):
    data = {
        "prompt": prompt,
        "negative_prompt": negative_prompt,
        "output_format": output_format,
        "output_size": output_size,
        "style": style,
        "lighting": lighting,
        "background": background,
        "camera_position": camera_position,
        "camera_angle": camera_angle,
    }
    response = requests.post(url, json=data)
    if response.status_code == 200:
        return response.content
    else:
        return None

# UI
gr.Interface(render,
             inputs={"prompt": gr.Textbox(placeholder="Введите описание изображения"),
                    "negative_prompt": gr.Textbox(placeholder="Введите отрицательный образ"),
                    "output_format": gr.Dropdown(["png", "jpg", "webp", "gif"]),
                    "output_size": gr.Dropdown(["256x256", "512x512", "1024x1024", "2048x2048"]),
                    "style": gr.Dropdown(["photorealistic", "cartoon", "anime", "sketch", "painting"]),
                    "lighting": gr.Dropdown(["natural", "studio", "night", "random"]),
                    "background": gr.Dropdown(["white", "black", "random"]),
                    "camera_position": gr.Dropdown(["front", "top", "side"]),
                    "camera_angle": gr.Dropdown(["0", "45", "90"])},
             outputs=gr.Output(type="image")).launch()