Update app.py

app.py

@@ -1,52 +1,10 @@
-from PIL import Image
 import gradio as gr
-from random import randint
+from model import models
+from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
+    change_model, warm_model, get_model_info_md, loaded_models,
+    get_positive_prefix, get_positive_suffix, get_negative_prefix, get_negative_suffix,
+    get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
 
-import torch
-import os
-
-os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Отключаем CUDA
-
-
-device = "cpu"  # Используем CPU
-model_repo_id = "stabilityai/sdxl-turbo"
-
-torch_dtype = torch.float32  # Тип данных для CPU
-
-
-
-# Список доступных моделей
-models = [
-    'stabilityai/stable-diffusion-3.5-large-turbo',
-    'black-forest-labs/FLUX.1-dev',
-    'strangerzonehf/Flux-Isometric-3D-LoRA',
-    'ckpt/kivotos-xl-2.0',
-    'stabilityai/stable-diffusion-3.5-large',
-    'yodayo-ai/clandestine-xl-1.0',
-    'yodayo-ai/kivotos-xl-2.0',
-    'yodayo-ai/holodayo-xl-2.1',
-    'prithivMLmods/SD3.5-Large-Anime-LoRA',
-    'FluffyKaeloky/Midnight-Miqu-103B-v1.5',
-    'cagliostrolab/animagine-xl-3.1',
-    'votepurchase/ponyDiffusionV6XL',
-    'eienmojiki/Anything-XL',
-    'eienmojiki/Starry-XL-v5.2',
-    "digiplay/MilkyWonderland_v1",
-    'Sombressoul/Yi-34B-200K-AWQ',
-    'digiplay/majicMIX_realistic_v7',
-    'votepurchase/counterfeitV30_v30',
-    'Linaqruf/animagine-xl-2.0',
-    'KBlueLeaf/Kohaku-XL-Epsilon-rev3',
-    'KBlueLeaf/Kohaku-XL-Zeta',
-    'kayfahaarukku/UrangDiffusion-1.4',
-    'Eugeoter/artiwaifu-diffusion-2.0',
-    'Raelina/Rae-Diffusion-XL-V2',
-    'Raelina/Raemu-XL-V4',
-    'Jonjew/NSFWMaster',
-]
-
-  
-                    
 max_images = 8
 MAX_SEED = 2**32-1
 load_models(models)
@@ -57,10 +15,9 @@ css = """
 .gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
 """
 
-
-# Функция основного интерфейса
-def make_me():
-    with gr.Row():
+with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
+    with gr.Tab("Image Generator"):
+        with gr.Row():
             with gr.Column(scale=10): 
                 with gr.Group():
                     prompt = gr.Text(label="Prompt", lines=2, max_lines=8, placeholder="1girl, solo, ...", show_copy_button=True)
@@ -176,179 +133,10 @@ def make_me():
 
 
 
-    with gr.Row():
-        gr.HTML("")
-
-custom_css = """
-:root {
-    --body-background-fill: #2d3d4f;
-}
-
-body {
-    background-color: var(--body-background-fill) !important;
-    color: #2d3d4f;
-    margin: 0;
-    padding: 0;
-    font-family: Arial, sans-serif;
-    height: 100vh;
-    overflow-y: auto;
-}
-
-.gradio-container {
-    background-color: #2d3d4f;
-    color: #c5c6c7;
-    padding: 20px;
-    border-radius: 8px;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    width: 100%;
-    max-width: 1200px;
-    margin: 20px auto;
-    display: block;
-    min-height: 100vh;
-}
-
-.app_title {
-    background-color: #2d3d4f;
-    color: #c5c6c7;
-    padding: 10px 20px;
-    border-bottom: 1px solid #3b4252;
-    text-align: center;
-    font-size: 24px;
-    font-weight: bold;
-    width: 100%;
-    box-sizing: border-box;
-    margin-bottom: 20px;
-}
-
-.custom_textbox {
-    background-color: #2d343f;
-    border: 1px solid #3b4252;
-    color: #7f8184;
-    padding: 10px;
-    border-radius: 4px;
-    margin-bottom: 10px;
-    width: 100%;
-    box-sizing: border-box;
-}
-
-.custom_gen_button {
-    background-color: #8b38ff;
-    border: 1px solid #ffffff;
-    color: blue;
-    padding: 15px 32px;
-    text-align: center;
-    text-decoration: none;
-    display: inline-block;
-    font-size: 16px;
-    margin: 4px 2px;
-    cursor: pointer;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    transition: transform 0.2s, box-shadow 0.2s;
-    border-radius: 4px;
-}
-.custom_gen_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 10px rgba(0, 0, 0, 0.3);
-}
-.custom_stop_button {
-    background-color: #6200ea;
-    border: 1px solid #ffffff;
-    color: blue;
-    padding: 15px 32px;
-    text-align: center;
-    text-decoration: none;
-    display: inline-block;
-    font-size: 16px;
-    margin: 4px 2px;
-    cursor: pointer;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    transition: transform 0.2s, box-shadow 0.2s;
-    border-radius: 4px;
-}
-.custom_stop_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 10px rgba(0, 0, 0, 0.3);
-}
-
-.custom_image {
-    border: 1px solid #3b4252;
-    background-color: #2d343f;
-    border-radius: 4px;
-    margin: 10px;
-    max-width: 100%;
-    box-sizing: border-box;
-}
-
-.custom_accordion {
-    background-color: #2d3d4f;
-    color: #7f8184;
-    border: 1px solid #3b4252;
-    border-radius: 4px;
-    margin-top: 20px;
-    width: 100%;
-    box-sizing: border-box;
-    transition: margin 0.2s ease;
-}
-
-.custom_accordion .gr-accordion-header {
-    background-color: #2d3d4f;
-    color: #7f8184;
-    padding: 10px 20px;
-    border-bottom: 1px solid #5b6270;
-    cursor: pointer;
-    font-size: 18px;
-    font-weight: bold;
-    height: 40px;
-    display: flex;
-    align-items: center;
-}
-
-.custom_accordion .gr-accordion-header:hover {
-    background-color: #2d3d4f;
-}
-
-.custom_accordion .gr-accordion-content {
-    padding: 10px 20px;
-    background-color: #2d3d4f;
-    border-top: 1px solid #5b6270;
-    max-height: 0;
-    overflow: hidden;
-    transition: max-height 0.2s ease;
-}
-
-.custom_accordion .gr-accordion-content.open {
-    max-height: 500px;
-}
-
-.custom_checkbox_group {
-    background-color: #2d343f;
-    border: 1px solid #3b4252;
-    color: #7f8184;
-    border-radius: 4px;
-    padding: 10px;
-    width: 100%;
-    box-sizing: border-box;
-}
-
-@media (max-width: 768px) {
-   .gradio-container {
-        width: 100%;
-        margin: 0;
-        padding: 10px;
-    }
-   .custom_textbox,.custom_image,.custom_checkbox_group {
-        width: 100%;
-        box-sizing: border-box;
-    }
-}
-"""
-
-with gr.Blocks(css=custom_css) as demo: 
-    make_me()
-# Очередь и запуск интерфейса с параметрами
-demo.queue(default_concurrency_limit=340, max_size=400)
-demo.launch(max_threads=400, ssr_mode=False)
 
+demo.queue(default_concurrency_limit=240, max_size=240)
+demo.launch(max_threads=400, ssr_mode=True)
+# https://github.com/gradio-app/gradio/issues/6339
 
 demo.queue(concurrency_count=50)
 demo.launch()