Upload 4 files

app.py

@@ -168,7 +168,7 @@ from modutils import (list_uniq, download_private_repo, get_model_id_list, get_t
     set_textual_inversion_prompt, get_model_pipeline, change_interface_mode, get_t2i_model_info,
     get_tupled_model_list, save_gallery_images, set_optimization, set_sampler_settings,
     set_quick_presets, process_style_prompt, optimization_list,
-    preset_styles, preset_quality, preset_sampler_setting)
+    preset_styles, preset_quality, preset_sampler_setting, translate_to_en)
 from env import (hf_token, CIVITAI_API_KEY, HF_LORA_ESSENTIAL_PRIVATE_REPO, HF_VAE_PRIVATE_REPO,
     HF_SDXL_EMBEDS_NEGATIVE_PRIVATE_REPO, HF_SDXL_EMBEDS_POSITIVE_PRIVATE_REPO,
     directory_models, directory_loras, directory_vaes, directory_embeds, directory_embeds_sdxl,
@@ -822,9 +822,10 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                             rating_dbt = gr.Radio(label="Rating", choices=list(V2_RATING_OPTIONS), value="sfw")
                             generate_db_random_button = gr.Button(value="EXTEND PROMPT 🎲")
                         with gr.Row():
-                            set_random_seed = gr.Button(value="Seed 🎲", size="sm")
-                            set_params_gui = gr.Button(value="Params ↙️", size="sm")
-                            clear_prompt_gui = gr.Button(value="Clear 🗑️", size="sm")
+                            translate_prompt_gui = gr.Button(value="Translate Prompt 📝", variant="secondary", size="sm")
+                            set_random_seed = gr.Button(value="Seed 🎲", variant="secondary", size="sm")
+                            set_params_gui = gr.Button(value="Params ↙️", variant="secondary", size="sm")
+                            clear_prompt_gui = gr.Button(value="Clear 🗑️", variant="secondary", size="sm")
 
                     generate_button = gr.Button(value="GENERATE IMAGE", size="lg", variant="primary")
 
@@ -1344,6 +1345,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
             queue=False,
         )
         model_name_gui.change(get_t2i_model_info, [model_name_gui], [model_info_gui], queue=False)
+        translate_prompt_gui.click(translate_to_en, [prompt_gui], [prompt_gui], queue=False)\
+        .then(translate_to_en, [neg_prompt_gui], [neg_prompt_gui], queue=False)
 
         gr.on(
             triggers=[quick_model_type_gui.change, quick_genre_gui.change, quick_speed_gui.change, quick_aspect_gui.change],

modutils.py

@@ -27,6 +27,17 @@ def list_sub(a, b):
     return [e for e in a if e not in b]
 
 
+from translatepy import Translator
+translator = Translator()
+def translate_to_en(input: str):
+    try:
+        output = str(translator.translate(input, 'English'))
+    except Exception as e:
+        output = input
+        print(e)
+    return output
+
+
 def get_local_model_list(dir_path):
     model_list = []
     valid_extensions = ('.ckpt', '.pt', '.pth', '.safetensors', '.bin')

requirements.txt

@@ -8,9 +8,7 @@ transformers
 optimum[onnxruntime]
 dartrs
 huggingface_hub
-httpx==0.13.3
-httpcore
-googletrans==4.0.0rc1
+translatepy
 timm
 rapidfuzz
 sentencepiece

tagger.py

@@ -1,11 +1,8 @@
+import spaces
 from PIL import Image
 import torch
 import gradio as gr
-import spaces
-from transformers import (
-    AutoImageProcessor,
-    AutoModelForImageClassification,
-)
+from transformers import AutoImageProcessor, AutoModelForImageClassification
 from pathlib import Path
 
 
@@ -190,18 +187,16 @@ def convert_danbooru_to_e621_prompt(input_prompt: str = "", prompt_type: str = "
     return output_prompt
 
 
+from translatepy import Translator
+translator = Translator()
 def translate_prompt(prompt: str = ""):
-    def translate_to_english(prompt):
-        import httpcore
-        setattr(httpcore, 'SyncHTTPTransport', 'AsyncHTTPProxy')
-        from googletrans import Translator
-        translator = Translator()
+    def translate_to_english(input: str):
         try:
-            translated_prompt = translator.translate(prompt, src='auto', dest='en').text
-            return translated_prompt
+            output = str(translator.translate(input, 'English'))
         except Exception as e:
+            output = input
             print(e)
-            return prompt
+        return output
 
     def is_japanese(s):
         import unicodedata
@@ -224,17 +219,13 @@ def translate_prompt(prompt: str = ""):
 
 
 def translate_prompt_to_ja(prompt: str = ""):
-    def translate_to_japanese(prompt):
-        import httpcore
-        setattr(httpcore, 'SyncHTTPTransport', 'AsyncHTTPProxy')
-        from googletrans import Translator
-        translator = Translator()
+    def translate_to_japanese(input: str):
         try:
-            translated_prompt = translator.translate(prompt, src='en', dest='ja').text
-            return translated_prompt
+            output = str(translator.translate(input, 'Japanese'))
         except Exception as e:
+            output = input
             print(e)
-            return prompt
+        return output
 
     def is_japanese(s):
         import unicodedata