Rooni's picture
Update app.py
dd7838a
raw
history blame
1.48 kB
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=[
gr.inputs.Textbox(placeholder="Введите описание изображения"),
gr.inputs.Textbox(placeholder="Введите отрицательный образ"),
gr.inputs.Dropdown(["png", "jpg", "webp", "gif"]),
gr.inputs.Dropdown(["256x256", "512x512", "1024x1024", "2048x2048"]),
gr.inputs.Dropdown(["photorealistic", "cartoon", "anime", "sketch", "painting"]),
gr.inputs.Dropdown(["natural", "studio", "night", "random"]),
gr.inputs.Dropdown(["white", "black", "random"]),
gr.inputs.Dropdown(["front", "top", "side"]),
gr.inputs.Dropdown(["0", "45", "90"]),
],
outputs=gr.Image()
).launch()