Create app.py

app.py

@@ -0,0 +1,93 @@
+import gradio as gr
+import time
+import random
+
+# Список моделей с отображаемыми названиями и внутренними именами
+models = {
+    "FLUX.1-dev": "black-forest-labs/FLUX.1-dev",
+    "flux-RealismLora": "XLabs-AI/flux-RealismLora",
+    "Midjourney": "Jovie/Midjourney",
+    "Stable Diffusion v1-5": "stable-diffusion-v1-5/stable-diffusion-v1-5",
+    "Pixel Art XL": "nerijs/pixel-art-xl",
+    "Knitted Character Flux LoRA": "prithivMLmods/Knitted-Character-Flux-LoRA",
+    "Flux Ghibsky Illustration": "aleksa-codes/flux-ghibsky-illustration",
+    "Flux Super Realism LoRA": "strangerzonehf/Flux-Super-Realism-LoRA",
+    "Flux Game Assets LoRA v2": "gokaygokay/Flux-Game-Assets-LoRA-v2",
+    "Flux Animex v2 LoRA": "strangerzonehf/Flux-Animex-v2-LoRA",
+    "Flux Qwen Capybara": "cfahlgren1/flux-qwen-capybara",
+    "Softserve Anime": "alvdansen/softserve_anime",
+    "FLUX Aesthetic Anime": "dataautogpt3/FLUX-AestheticAnime",
+    "FLUX Animeo v1 LoRA": "strangerzonehf/Flux-Animeo-v1-LoRA",
+    "Fractured Line Flare": "prithivMLmods/Fractured-Line-Flare",
+    "Castor 3D Portrait Flux LoRA": "prithivMLmods/Castor-3D-Portrait-Flux-LoRA"
+}
+
+# Очередь на генерации
+queue = []
+
+def generate_image(model_name, width, height, seed, steps, guidance_scale):
+    # Проверка очереди
+    if len(queue) >= 150:
+        return "Очередь переполнена. Пожалуйста, подождите."
+    
+    # Добавляем задачу в очередь
+    task_id = random.randint(100000, 999999)
+    queue.append(task_id)
+    
+    # Имитация генерации изображения
+    time.sleep(random.uniform(0.1, 1.0))  # Таймаут до 1 секунды
+    
+    # Удаляем задачу из очереди
+    queue.remove(task_id)
+    
+    # Имитация изображения
+    image = f"generated_images/{task_id}.png"
+    return image
+
+# Создание интерфейса
+with gr.Blocks() as demo:
+    gr.Markdown("# Генератор изображений")
+    
+    with gr.Row():
+        model_dropdown = gr.Dropdown(list(models.keys()), label="Модель")
+    
+    with gr.Row():
+        with gr.Column():
+            width_slider = gr.Slider(128, 1024, value=512, step=64, label="Ширина")
+        with gr.Column():
+            height_slider = gr.Slider(128, 1024, value=512, step=64, label="Высота")
+    
+    with gr.Row():
+        seed_slider = gr.Slider(0, 1000000, value=random.randint(0, 1000000), step=1, label="Сид")
+    
+    with gr.Row():
+        steps_slider = gr.Slider(1, 100, value=50, step=1, label="Шаги")
+        guidance_scale_slider = gr.Slider(0.1, 20.0, value=7.5, step=0.1, label="Гуидансе скала")
+    
+    generate_button = gr.Button("Сгенерировать изображение")
+    output_image = gr.Image(label="Сгенерированное изображение")
+    
+    generate_button.click(
+        fn=generate_image,
+        inputs=[model_dropdown, width_slider, height_slider, seed_slider, steps_slider, guidance_scale_slider],
+        outputs=output_image,
+        api_name="/generate"
+    )
+    
+    demo.queue(max_size=150, api_open=False)
+    demo.launch(share=False, server_port=7860)
+
+# Скрытие футера
+hide_footer = """
+<style>
+    .gradio-container .footer {
+        display: none !important;
+    }
+</style>
+"""
+demo = demo.add_css(hide_footer)
+
+# Установка таймаута
+demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
+
+demo.launch()