Update app.py

app.py

@@ -1,10 +1,9 @@
-import os
 import gradio as gr
 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)
+    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)
 
 max_images = 8
 MAX_SEED = 2**32-1
@@ -16,20 +15,7 @@ css = """
 .gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
 """
 
-def set_token(token):
-    os.environ["HF_TOKEN"] = token  # Сохранение токена в переменной окружения
-    return f"✅ Токен сохранен!"
-
-# Основной интерфейс Gradio
 with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
-    with gr.Tab("Token Input"):  # Новая вкладка для ввода токена
-        gr.Markdown("# 🔑 Введите ваш HF_TOKEN")
-        with gr.Row():
-            token_input = gr.Textbox(label="HF_TOKEN", type="password", placeholder="Введите ваш токен...")
-            save_button = gr.Button("Сохранить")
-        output_text = gr.Textbox(label="Статус", interactive=False)
-        save_button.click(set_token, inputs=[token_input], outputs=[output_text])
-
     with gr.Tab("Image Generator"):
         with gr.Row():
             with gr.Column(scale=10): 
@@ -67,11 +53,11 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
                 with gr.Group():
                     with gr.Row():
                         output = [gr.Image(label='', elem_classes="output", type="filepath", format="png",
-                                           show_download_button=True, show_share_button=False, show_label=False,
-                                           interactive=False, min_width=80, visible=True, width=112, height=112) for _ in range(max_images)]
+                                show_download_button=True, show_share_button=False, show_label=False,
+                                interactive=False, min_width=80, visible=True, width=112, height=112) for _ in range(max_images)]
                 with gr.Group():
                     results = gr.Gallery(label="Gallery", elem_classes="gallery", interactive=False, show_download_button=True, show_share_button=False,
-                                         container=True, format="png", object_fit="cover", columns=2, rows=2)
+                                        container=True, format="png", object_fit="cover", columns=2, rows=2)
                     image_files = gr.Files(label="Download", interactive=False)
                     clear_results = gr.Button("Clear Gallery / Download 🗑️", variant="secondary")
         with gr.Column():
@@ -120,35 +106,37 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
 
     gr.on(triggers=[run_button.click, prompt.submit, random_button.click], fn=lambda: gr.update(interactive=True), inputs=None, outputs=stop_button, show_api=False)
     model_name.change(change_model, [model_name], [model_info], queue=True, show_api=True)\
-        .success(warm_model, [model_name], None, queue=True, show_api=True)
-    
+    .success(warm_model, [model_name], None, queue=True, show_api=True)
     for i, o in enumerate(output):
         img_i = gr.Number(i, visible=False)
-        image_num.change(lambda i, n: gr.update(visible=(i < n)), [img_i, image_num], o, show_api=True)
+        image_num.change(lambda i, n: gr.update(visible = (i < n)), [img_i, image_num], o, show_api=True)
         gen_event = gr.on(triggers=[run_button.click, prompt.submit],
-                          fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4: infer_fn(m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4) if (i < n) else None,
-                          inputs=[img_i, image_num, model_name, prompt, neg_prompt, height, width, steps, cfg, seed,
-                                  positive_prefix, positive_suffix, negative_prefix, negative_suffix],
-                          outputs=[o], queue=True, show_api=False)
+         fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4: infer_fn(m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4) if (i < n) else None,
+         inputs=[img_i, image_num, model_name, prompt, neg_prompt, height, width, steps, cfg, seed,
+                  positive_prefix, positive_suffix, negative_prefix, negative_suffix],
+         outputs=[o], queue=True, show_api=False) # Be sure to delete ", queue=False" when activating the stop button
         gen_event2 = gr.on(triggers=[random_button.click],
-                           fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4: infer_rand_fn(m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4) if (i < n) else None,
-                           inputs=[img_i, image_num, model_name, prompt, neg_prompt, height, width, steps, cfg, seed,
-                                   positive_prefix, positive_suffix, negative_prefix, negative_suffix],
-                           outputs=[o], queue=True, show_api=False)
+         fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4: infer_rand_fn(m, t1, t2, n1, n2, n3, n4, n5, l1, l2, l3, l4) if (i < n) else None,
+         inputs=[img_i, image_num, model_name, prompt, neg_prompt, height, width, steps, cfg, seed,
+                  positive_prefix, positive_suffix, negative_prefix, negative_suffix],
+         outputs=[o], queue=True, show_api=False) # Be sure to delete ", queue=False" when activating the stop button
         o.change(save_gallery, [o, results], [results, image_files], show_api=False)
         stop_button.click(lambda: gr.update(interactive=False), None, stop_button, cancels=[gen_event, gen_event2], show_api=False)
 
     clear_prompt.click(lambda: (None, None), None, [prompt, neg_prompt], queue=True, show_api=True)
     clear_results.click(lambda: (None, None), None, [results, image_files], queue=True, show_api=True)
     recom_prompt_preset.change(set_recom_prompt_preset, [recom_prompt_preset],
-                                [positive_prefix, positive_suffix, negative_prefix, negative_suffix], queue=True, show_api=True)
+     [positive_prefix, positive_suffix, negative_prefix, negative_suffix], queue=True, show_api=True)
     seed_rand.click(randomize_seed, None, [seed], queue=True, show_api=True)
     trans_prompt.click(translate_to_en, [prompt], [prompt], queue=True, show_api=True)\
-        .then(translate_to_en, [neg_prompt], [neg_prompt], queue=True, show_api=True)
+    .then(translate_to_en, [neg_prompt], [neg_prompt], queue=True, show_api=True)
+
+
+
 
 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()
+demo.launch()