Update app.py

app.py

@@ -1,10 +1,11 @@
+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)
-from huggingface_hub import login
+                      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
 load_models(models)
@@ -15,21 +16,20 @@ css = """
 .gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
 """
 
-
 def set_token(token):
-    global HF_TOKEN
-    HF_TOKEN = token
-    os.environ["HF_TOKEN"] = token  # Сохраняем в переменной окружения
-    login(HF_TOKEN)
-    return "Токен установлен!"
+    os.environ["HF_TOKEN"] = token  # Сохранение токена в переменной окружения
+    return f"✅ Токен сохранен!"
 
-with gr.Blocks() as demo:
-    hf_token_input = gr.Textbox(label="Введите HF_TOKEN", type="password")
-    submit_button = gr.Button("Сохранить")
-    output_text = gr.Textbox(label="Статус")
-
-    submit_button.click(set_token, inputs=hf_token_input, outputs=output_text)
+# Основной интерфейс 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 +67,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,37 +120,35 @@ 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) # Be sure to delete ", queue=False" when activating the stop button
+                          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)
         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) # Be sure to delete ", queue=False" when activating the stop button
+                           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)
         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()