fix more

app.py

@@ -1,49 +1,122 @@
+import random
+
 import gradio as gr
 from datasets import load_dataset
 from PIL import Image
+from trans_google import google_translator
 
-import re
-import os
-import requests
-
-from share_btn import community_icon_html, loading_icon_html, share_js
+from i18n import i18nTranslator
 
 word_list_dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 word_list = word_list_dataset["train"]['Prompt']
 
-from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler, DDIMScheduler, KDPM2AncestralDiscreteScheduler, \
+    UniPCMultistepScheduler, DPMSolverSinglestepScheduler, DEISMultistepScheduler, PNDMScheduler, \
+    DPMSolverMultistepScheduler, HeunDiscreteScheduler, EulerAncestralDiscreteScheduler, DDPMScheduler, \
+    LMSDiscreteScheduler, KDPM2DiscreteScheduler
 import torch
+import base64
+from io import BytesIO
 
+#
 model_id = "stabilityai/stable-diffusion-2-1-base"
 
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
     scheduler=scheduler,
-    safety_checker=None,
+    # safety_checker=None,
+    revision="fp16",
     torch_dtype=torch.float16)
 
+if torch.cuda.is_available():
+    pipe = pipe.to('cuda')
 is_gpu_busy = False
 
+translator = i18nTranslator()
+translator.init(path='locales')
+samplers = [
+    "EulerDiscrete",
+    "EulerAncestralDiscrete",
+    "UniPCMultistep",
+    "DPMSolverSinglestep",
+    "DPMSolverMultistep",
+    "KDPM2Discrete",
+    "KDPM2AncestralDiscrete",
+    "DEISMultistep",
+    "HeunDiscrete",
+    "PNDM",
+    "DDPM",
+    "DDIM",
+    "LMSDiscrete",
+]
+
+rand = random.Random()
+translator = google_translator()
 
-def infer(prompt, negative, scale):
+
+def infer(prompt: str, negative: str, width: int, height: int, sampler: str, steps: int, seed: int, scale):
     global is_gpu_busy
-    for filter in word_list:
-        if re.search(rf"\b{filter}\b", prompt):
-            raise gr.Error("Unsafe content found. Please try again with different prompts.")
+
+    if seed == 0:
+        seed = rand.randint(0, 10000)
+    else:
+        seed = int(seed)
 
     images = []
-    generator = torch.manual_seed(68781279)
-    payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
-    image = pipe(prompt=prompt,
-                 negative_prompt=negative,
+    device = "cpu"
+    if torch.cuda.is_available():
+        device = "cuda"
+    generator = torch.Generator(device=device).manual_seed(seed)
+    if sampler == "EulerDiscrete":
+        pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "EulerAncestralDiscrete":
+        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "KDPM2Discrete":
+        pipe.scheduler = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "KDPM2AncestralDiscrete":
+        pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "UniPCMultistep":
+        pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "DPMSolverSinglestep":
+        pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "DPMSolverMultistep":
+        pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "HeunDiscrete":
+        pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "DEISMultistep":
+        pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "PNDM":
+        pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "DDPM":
+        pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "DDIM":
+        pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+    elif sampler == "LMSDiscrete":
+        pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
+
+    try:
+        translate_prompt = translator.translate(prompt, lang_tgt='en')
+        translate_negative = translator.translate(negative, lang_tgt='en')
+    except Exception as ex:
+        print(ex)
+        translate_prompt = prompt
+        translate_negative = negative
+
+    image = pipe(prompt=translate_prompt,
+                 negative_prompt=translate_negative,
                  guidance_scale=scale,
-                 num_inference_steps=15,
+                 num_inference_steps=steps,
                  generator=generator,
-                 height=512,
-                 width=768).images[0]
+                 height=height,
+                 width=width).images[0]
+
+    buffered = BytesIO()
+    image.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue())
+    img_base64 = bytes("data:image/jpeg;base64,", encoding='utf-8') + img_str
 
-    images.append(image)
+    images.append(img_base64)
 
     return images
 
@@ -170,37 +243,76 @@ css = """
         #prompt-text-input, #negative-prompt-text-input{padding: .45rem 0.625rem}
         #component-16{border-top-width: 1px!important;margin-top: 1em}
         .image_duplication{position: absolute; width: 100px; left: 50px}
+        .generate-container {display: flex; justify-content: flex-end;} 
+        #generate-btn {background: linear-gradient(to bottom right, #ffedd5, #fdba74)}
 """
 
 block = gr.Blocks(css=css)
 
-examples = [
-    [
-        'A high tech solarpunk utopia in the Amazon rainforest',
-        'low quality',
-        9
-    ],
-    [
-        'A pikachu fine dining with a view to the Eiffel Tower',
-        'low quality',
-        9
-    ],
-    [
-        'A mecha robot in a favela in expressionist style',
-        'low quality, 3d, photorealistic',
-        9
-    ],
-    [
-        'an insect robot preparing a delicious meal',
-        'low quality, illustration',
-        9
-    ],
-    [
-        "A small cabin on top of a snowy mountain in the style of Disney, artstation",
-        'low quality, ugly',
-        9
-    ],
-]
+# text, negative, width, height, sampler, steps, seed, guidance_scale
+# examples = [
+#     [
+#         'A high tech solarpunk utopia in the Amazon rainforest',
+#         'low quality',
+#         512,
+#         512,
+#         'ddim',
+#         30,
+#         0,
+#         9
+#     ],
+#     [
+#         'A pikachu fine dining with a view to the Eiffel Tower',
+#         'low quality',
+#         512,
+#         512,
+#         'ddim',
+#         30,
+#         0,
+#         9
+#     ],
+#     [
+#         'A mecha robot in a favela in expressionist style',
+#         'low quality, 3d, photorealistic',
+#         512,
+#         512,
+#         'ddim',
+#         30,
+#         0,
+#         9
+#     ],
+#     [
+#         'an insect robot preparing a delicious meal',
+#         'low quality, illustration',
+#         512,
+#         512,
+#         'ddim',
+#         30,
+#         0,
+#         9
+#     ],
+#     [
+#         "A small cabin on top of a snowy mountain in the style of Disney, artstation",
+#         'low quality, ugly',
+#         512,
+#         512,
+#         'ddim',
+#         30,
+#         0,
+#         9
+#     ],
+# ]
+
+examples = list(map(lambda x: [
+    x,
+    'low quality',
+    512,
+    512,
+    'ddim',
+    30,
+    0,
+    9
+], word_list))[:500]
 
 with block:
     gr.HTML(
@@ -252,14 +364,9 @@ with block:
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
-                Stable Diffusion 2.1 is the latest text-to-image model from StabilityAI. <a style="text-decoration: underline;" href="https://huggingface.co/spaces/stabilityai/stable-diffusion-1">Access Stable Diffusion 1 Space here</a><br>For faster generation and API
-                access you can try
-                <a
-                  href="http://beta.dreamstudio.ai/"
-                  style="text-decoration: underline;"
-                  target="_blank"
-                  >DreamStudio Beta</a
-                >.</a>
+                Stable Diffusion 2.1 Demo App.
+                Click "Generate image" Button to generate image.
+                Also Change params to have a try
               </p>
             </div>
         """
@@ -290,60 +397,65 @@ with block:
                         rounded=(True, False, False, True),
                         container=False,
                     )
-                btn = gr.Button("Generate image").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                    full_width=False,
-                )
+                    with gr.Row(elem_id="txt2img_size", scale=4):
+                        width = gr.Slider(minimum=64, maximum=2048, step=8, label="Width", value=512,
+                                          elem_id="txt2img_width")
+                        height = gr.Slider(minimum=64, maximum=2048, step=8, label="Height", value=512,
+                                           elem_id="txt2img_height")
 
-        gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+                    with gr.Row(elem_id="txt2img_sampler", scale=4):
+                        seed = gr.Number(value=0, label="Seed", elem_id="txt2img_seed")
+                        sampler = gr.Dropdown(
+                            samplers, value="",
+                            multiselect=False,
+                            label="Sampler",
+                            info="sampler select"
+                        )
+                        steps = gr.Slider(minimum=1, maximum=80, step=1, elem_id=f"steps", label="Sampling steps",
+                                          value=20)
 
-        with gr.Group(elem_id="container-advanced-btns"):
-            # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
-            with gr.Group(elem_id="share-btn-container"):
-                community_icon = gr.HTML(community_icon_html)
-                loading_icon = gr.HTML(loading_icon_html)
-                share_button = gr.Button("Share to community", elem_id="share-btn")
+                    with gr.Accordion("Advanced settings", open=False):
+                        #    gr.Markdown("Advanced settings are temporarily unavailable")
+                        #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
+                        #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
+                        guidance_scale = gr.Slider(
+                            label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1
+                        )
 
-        with gr.Accordion("Advanced settings", open=False):
-            #    gr.Markdown("Advanced settings are temporarily unavailable")
-            #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
-            #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
-            guidance_scale = gr.Slider(
-                label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1
-            )
-        #    seed = gr.Slider(
-        #        label="Seed",
-        #        minimum=0,
-        #        maximum=2147483647,
-        #        step=1,
-        #        randomize=True,
-        #    )
+                    with gr.Row(elem_id="generate-container", elem_classes="generate-container"):
+                        btn = gr.Button("Generate image", elem_id="generate-btn", elem_classes="generate-btn").style(
+                            margin=False,
+                            rounded=(False, True, True, False),
+                            full_width=False,
+                        )
 
-        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, guidance_scale],
-                         outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False)
-        ex.dataset.headers = [""]
-        negative.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
-        text.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
-        btn.click(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
+                gallery = gr.Gallery(
+                    label="Generated images", show_label=False, elem_id="gallery"
+                ).style(grid=[2], height="auto")
+
+        # with gr.Group(elem_id="container-advanced-btns"):
+        #     # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
+        #     with gr.Group(elem_id="share-btn-container"):
+        #         community_icon = gr.HTML(community_icon_html)
+        #         loading_icon = gr.HTML(loading_icon_html)
+        #         share_button = gr.Button("Share to community", elem_id="share-btn")
 
-        # advanced_button.click(
-        #    None,
-        #    [],
-        #    text,
-        #    _js="""
-        #    () => {
-        #        const options = document.querySelector("body > gradio-app").querySelector("#advanced-options");
-        #        options.style.display = ["none", ""].includes(options.style.display) ? "flex" : "none";
-        #    }""",
-        # )
-        share_button.click(
-            None,
-            [],
-            [],
-            _js=share_js,
-        )
+        ex = gr.Examples(examples=examples, fn=infer,
+                         inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
+                         outputs=[gallery],
+                         examples_per_page=10,
+                         cache_examples=False)
+        ex.dataset.headers = [""]
+        negative.submit(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
+                        outputs=[gallery], postprocess=False)
+        text.submit(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
+                    outputs=[gallery], postprocess=False)
+        btn.click(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
+                  outputs=[gallery], postprocess=False)
 
-block.queue(concurrency_count=80, max_size=100).launch(max_threads=150)
+block.queue(concurrency_count=80,
+            max_size=100).launch(
+    max_threads=150,
+    server_port=6006,
+    share=True,
+)

i18n.py

@@ -0,0 +1,28 @@
+import os
+import json
+
+class i18nTranslator:
+    def __init__(self):
+        self.translations = {}
+
+    def init(self, path):
+        for filename in os.listdir(path):
+            if filename.endswith('.json'):
+                language = filename.split('.')[0]
+                with open(os.path.join(path, filename), 'r') as file:
+                    self.translations[language] = json.load(file)
+
+    def trans(self, key, lng):
+        if lng in self.translations and key in self.translations[lng]:
+            return self.translations[lng][key]
+        elif 'en' in self.translations and key in self.translations['en']:
+            return self.translations['en'][key]
+        else:
+            return key
+
+# Usage example
+
+if __name__ == '__main__':
+    translator = i18nTranslator()
+    translator.init(path='locales')
+    print(translator.trans(key="Enter your prompt", lng="zh-CN")) # Output: "输入你的提示词(你希望生成的图片的描述)"

locales/zh-CN.json

@@ -0,0 +1,3 @@
+{
+  "introduce": "Stable Diffusion 2.1 Demo App.\n                Click \"Generate image\" Button to generate image.\n                Also Change params to have a try",
+}

model_test.py

@@ -0,0 +1,20 @@
+
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+import torch
+
+model_id = "stabilityai/stable-diffusion-2-1-base"
+
+scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id,
+    scheduler=scheduler,
+    # safety_checker=None,
+    revision="fp16",
+    torch_dtype=torch.float16)
+
+pipe = pipe.to("cuda")
+
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]
+
+image.save("astronaut_rides_horse.png")

requirements.txt

@@ -3,4 +3,5 @@ diffusers
 transformers
 accelerate
 scipy
-safetensors
+safetensors
+python-i18n

trans_constant.py

@@ -0,0 +1,179 @@
+LANGUAGES = {
+    'af': 'afrikaans',
+    'sq': 'albanian',
+    'am': 'amharic',
+    'ar': 'arabic',
+    'hy': 'armenian',
+    'az': 'azerbaijani',
+    'eu': 'basque',
+    'be': 'belarusian',
+    'bn': 'bengali',
+    'bs': 'bosnian',
+    'bg': 'bulgarian',
+    'ca': 'catalan',
+    'ceb': 'cebuano',
+    'ny': 'chichewa',
+    'zh-cn': 'chinese (simplified)',
+    'zh-tw': 'chinese (traditional)',
+    'co': 'corsican',
+    'hr': 'croatian',
+    'cs': 'czech',
+    'da': 'danish',
+    'nl': 'dutch',
+    'en': 'english',
+    'eo': 'esperanto',
+    'et': 'estonian',
+    'tl': 'filipino',
+    'fi': 'finnish',
+    'fr': 'french',
+    'fy': 'frisian',
+    'gl': 'galician',
+    'ka': 'georgian',
+    'de': 'german',
+    'el': 'greek',
+    'gu': 'gujarati',
+    'ht': 'haitian creole',
+    'ha': 'hausa',
+    'haw': 'hawaiian',
+    'iw': 'hebrew',
+    'he': 'hebrew',
+    'hi': 'hindi',
+    'hmn': 'hmong',
+    'hu': 'hungarian',
+    'is': 'icelandic',
+    'ig': 'igbo',
+    'id': 'indonesian',
+    'ga': 'irish',
+    'it': 'italian',
+    'ja': 'japanese',
+    'jw': 'javanese',
+    'kn': 'kannada',
+    'kk': 'kazakh',
+    'km': 'khmer',
+    'ko': 'korean',
+    'ku': 'kurdish (kurmanji)',
+    'ky': 'kyrgyz',
+    'lo': 'lao',
+    'la': 'latin',
+    'lv': 'latvian',
+    'lt': 'lithuanian',
+    'lb': 'luxembourgish',
+    'mk': 'macedonian',
+    'mg': 'malagasy',
+    'ms': 'malay',
+    'ml': 'malayalam',
+    'mt': 'maltese',
+    'mi': 'maori',
+    'mr': 'marathi',
+    'mn': 'mongolian',
+    'my': 'myanmar (burmese)',
+    'ne': 'nepali',
+    'no': 'norwegian',
+    'or': 'odia',
+    'ps': 'pashto',
+    'fa': 'persian',
+    'pl': 'polish',
+    'pt': 'portuguese',
+    'pa': 'punjabi',
+    'ro': 'romanian',
+    'ru': 'russian',
+    'sm': 'samoan',
+    'gd': 'scots gaelic',
+    'sr': 'serbian',
+    'st': 'sesotho',
+    'sn': 'shona',
+    'sd': 'sindhi',
+    'si': 'sinhala',
+    'sk': 'slovak',
+    'sl': 'slovenian',
+    'so': 'somali',
+    'es': 'spanish',
+    'su': 'sundanese',
+    'sw': 'swahili',
+    'sv': 'swedish',
+    'tg': 'tajik',
+    'ta': 'tamil',
+    'tt': 'tatar',
+    'te': 'telugu',
+    'th': 'thai',
+    'tr': 'turkish',
+    'tk': 'turkmen',
+    'uk': 'ukrainian',
+    'ur': 'urdu',
+    'ug': 'uyghur',
+    'uz': 'uzbek',
+    'vi': 'vietnamese',
+    'cy': 'welsh',
+    'xh': 'xhosa',
+    'yi': 'yiddish',
+    'yo': 'yoruba',
+    'zu': 'zulu',
+}
+
+DEFAULT_SERVICE_URLS = ('translate.google.ac','translate.google.ad','translate.google.ae',
+                        'translate.google.al','translate.google.am','translate.google.as',
+                        'translate.google.at','translate.google.az','translate.google.ba',
+                        'translate.google.be','translate.google.bf','translate.google.bg',
+                        'translate.google.bi','translate.google.bj','translate.google.bs',
+                        'translate.google.bt','translate.google.by','translate.google.ca',
+                        'translate.google.cat','translate.google.cc','translate.google.cd',
+                        'translate.google.cf','translate.google.cg','translate.google.ch',
+                        'translate.google.ci','translate.google.cl','translate.google.cm',
+                        'translate.google.cn','translate.google.co.ao','translate.google.co.bw',
+                        'translate.google.co.ck','translate.google.co.cr','translate.google.co.id',
+                        'translate.google.co.il','translate.google.co.in','translate.google.co.jp',
+                        'translate.google.co.ke','translate.google.co.kr','translate.google.co.ls',
+                        'translate.google.co.ma','translate.google.co.mz','translate.google.co.nz',
+                        'translate.google.co.th','translate.google.co.tz','translate.google.co.ug',
+                        'translate.google.co.uk','translate.google.co.uz','translate.google.co.ve',
+                        'translate.google.co.vi','translate.google.co.za','translate.google.co.zm',
+                        'translate.google.co.zw','translate.google.co','translate.google.com.af',
+                        'translate.google.com.ag','translate.google.com.ai','translate.google.com.ar',
+                        'translate.google.com.au','translate.google.com.bd','translate.google.com.bh',
+                        'translate.google.com.bn','translate.google.com.bo','translate.google.com.br',
+                        'translate.google.com.bz','translate.google.com.co','translate.google.com.cu',
+                        'translate.google.com.cy','translate.google.com.do','translate.google.com.ec',
+                        'translate.google.com.eg','translate.google.com.et','translate.google.com.fj',
+                        'translate.google.com.gh','translate.google.com.gi','translate.google.com.gt',
+                        'translate.google.com.hk','translate.google.com.jm','translate.google.com.kh',
+                        'translate.google.com.kw','translate.google.com.lb','translate.google.com.lc',
+                        'translate.google.com.ly','translate.google.com.mm','translate.google.com.mt',
+                        'translate.google.com.mx','translate.google.com.my','translate.google.com.na',
+                        'translate.google.com.ng','translate.google.com.ni','translate.google.com.np',
+                        'translate.google.com.om','translate.google.com.pa','translate.google.com.pe',
+                        'translate.google.com.pg','translate.google.com.ph','translate.google.com.pk',
+                        'translate.google.com.pr','translate.google.com.py','translate.google.com.qa',
+                        'translate.google.com.sa','translate.google.com.sb','translate.google.com.sg',
+                        'translate.google.com.sl','translate.google.com.sv','translate.google.com.tj',
+                        'translate.google.com.tr','translate.google.com.tw','translate.google.com.ua',
+                        'translate.google.com.uy','translate.google.com.vc','translate.google.com.vn',
+                        'translate.google.com','translate.google.cv','translate.google.cx',
+                        'translate.google.cz','translate.google.de','translate.google.dj',
+                        'translate.google.dk','translate.google.dm','translate.google.dz',
+                        'translate.google.ee','translate.google.es','translate.google.eu',
+                        'translate.google.fi','translate.google.fm','translate.google.fr',
+                        'translate.google.ga','translate.google.ge','translate.google.gf',
+                        'translate.google.gg','translate.google.gl','translate.google.gm',
+                        'translate.google.gp','translate.google.gr','translate.google.gy',
+                        'translate.google.hn','translate.google.hr','translate.google.ht',
+                        'translate.google.hu','translate.google.ie','translate.google.im',
+                        'translate.google.io','translate.google.iq','translate.google.is',
+                        'translate.google.it','translate.google.je','translate.google.jo',
+                        'translate.google.kg','translate.google.ki','translate.google.kz',
+                        'translate.google.la','translate.google.li','translate.google.lk',
+                        'translate.google.lt','translate.google.lu','translate.google.lv',
+                        'translate.google.md','translate.google.me','translate.google.mg',
+                        'translate.google.mk','translate.google.ml','translate.google.mn',
+                        'translate.google.ms','translate.google.mu','translate.google.mv',
+                        'translate.google.mw','translate.google.ne','translate.google.nf',
+                        'translate.google.nl','translate.google.no','translate.google.nr',
+                        'translate.google.nu','translate.google.pl','translate.google.pn',
+                        'translate.google.ps','translate.google.pt','translate.google.ro',
+                        'translate.google.rs','translate.google.ru','translate.google.rw',
+                        'translate.google.sc','translate.google.se','translate.google.sh',
+                        'translate.google.si','translate.google.sk','translate.google.sm',
+                        'translate.google.sn','translate.google.so','translate.google.sr',
+                        'translate.google.st','translate.google.td','translate.google.tg',
+                        'translate.google.tk','translate.google.tl','translate.google.tm',
+                        'translate.google.tn','translate.google.to','translate.google.tt',
+                        'translate.google.us','translate.google.vg','translate.google.vu','translate.google.ws')

trans_google.py

@@ -0,0 +1,251 @@
+# coding:utf-8
+# author LuShan
+# version : 1.1.9
+import json, requests, random, re
+from urllib.parse import quote
+import urllib3
+import logging
+from trans_constant import LANGUAGES, DEFAULT_SERVICE_URLS
+
+log = logging.getLogger(__name__)
+log.addHandler(logging.NullHandler())
+
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+URLS_SUFFIX = [re.search('translate.google.(.*)', url.strip()).group(1) for url in DEFAULT_SERVICE_URLS]
+URL_SUFFIX_DEFAULT = 'com'
+
+
+class google_new_transError(Exception):
+    """Exception that uses context to present a meaningful error message"""
+
+    def __init__(self, msg=None, **kwargs):
+        self.tts = kwargs.pop('tts', None)
+        self.rsp = kwargs.pop('response', None)
+        if msg:
+            self.msg = msg
+        elif self.tts is not None:
+            self.msg = self.infer_msg(self.tts, self.rsp)
+        else:
+            self.msg = None
+        super(google_new_transError, self).__init__(self.msg)
+
+    def infer_msg(self, tts, rsp=None):
+        cause = "Unknown"
+
+        if rsp is None:
+            premise = "Failed to connect"
+
+            return "{}. Probable cause: {}".format(premise, "timeout")
+            # if tts.tld != 'com':
+            #     host = _translate_url(tld=tts.tld)
+            #     cause = "Host '{}' is not reachable".format(host)
+
+        else:
+            status = rsp.status_code
+            reason = rsp.reason
+
+            premise = "{:d} ({}) from TTS API".format(status, reason)
+
+            if status == 403:
+                cause = "Bad token or upstream API changes"
+            elif status == 200 and not tts.lang_check:
+                cause = "No audio stream in response. Unsupported language '%s'" % self.tts.lang
+            elif status >= 500:
+                cause = "Uptream API error. Try again later."
+
+        return "{}. Probable cause: {}".format(premise, cause)
+
+
+class google_translator:
+    '''
+    You can use 108 language in target and source,details view LANGUAGES.
+    Target language: like 'en'、'zh'、'th'...
+
+    :param url_suffix: The source text(s) to be translated. Batch translation is supported via sequence input.
+                       The value should be one of the url_suffix listed in : `DEFAULT_SERVICE_URLS`
+    :type url_suffix: UTF-8 :class:`str`; :class:`unicode`; string sequence (list, tuple, iterator, generator)
+
+    :param text: The source text(s) to be translated.
+    :type text: UTF-8 :class:`str`; :class:`unicode`;
+
+    :param lang_tgt: The language to translate the source text into.
+                     The value should be one of the language codes listed in : `LANGUAGES`
+    :type lang_tgt: :class:`str`; :class:`unicode`
+
+    :param lang_src: The language of the source text.
+                    The value should be one of the language codes listed in :const:`googletrans.LANGUAGES`
+                    If a language is not specified,
+                    the system will attempt to identify the source language automatically.
+    :type lang_src: :class:`str`; :class:`unicode`
+
+    :param timeout: Timeout Will be used for every request.
+    :type timeout: number or a double of numbers
+
+    :param proxies: proxies Will be used for every request.
+    :type proxies: class : dict; like: {'http': 'http:171.112.169.47:19934/', 'https': 'https:171.112.169.47:19934/'}
+
+    '''
+
+    def __init__(self, url_suffix="com", timeout=5, proxies=None):
+        self.proxies = proxies
+        if url_suffix not in URLS_SUFFIX:
+            self.url_suffix = URL_SUFFIX_DEFAULT
+        else:
+            self.url_suffix = url_suffix
+        url_base = "https://translate.google.{}".format(self.url_suffix)
+        self.url = url_base + "/_/TranslateWebserverUi/data/batchexecute"
+        self.timeout = timeout
+
+    def _package_rpc(self, text, lang_src='auto', lang_tgt='auto'):
+        GOOGLE_TTS_RPC = ["MkEWBc"]
+        parameter = [[text.strip(), lang_src, lang_tgt, True], [1]]
+        escaped_parameter = json.dumps(parameter, separators=(',', ':'))
+        rpc = [[[random.choice(GOOGLE_TTS_RPC), escaped_parameter, None, "generic"]]]
+        espaced_rpc = json.dumps(rpc, separators=(',', ':'))
+        # text_urldecode = quote(text.strip())
+        freq_initial = "f.req={}&".format(quote(espaced_rpc))
+        freq = freq_initial
+        return freq
+
+    def translate(self, text, lang_tgt='auto', lang_src='auto', pronounce=False):
+        try:
+            lang = LANGUAGES[lang_src]
+        except:
+            lang_src = 'auto'
+        try:
+            lang = LANGUAGES[lang_tgt]
+        except:
+            lang_src = 'auto'
+        text = str(text)
+        if len(text) >= 5000:
+            return "Warning: Can only detect less than 5000 characters"
+        if len(text) == 0:
+            return ""
+        headers = {
+            "Referer": "http://translate.google.{}/".format(self.url_suffix),
+            "User-Agent":
+                "Mozilla/5.0 (Windows NT 10.0; WOW64) "
+                "AppleWebKit/537.36 (KHTML, like Gecko) "
+                "Chrome/47.0.2526.106 Safari/537.36",
+            "Content-Type": "application/x-www-form-urlencoded;charset=utf-8"
+        }
+        freq = self._package_rpc(text, lang_src, lang_tgt)
+        response = requests.Request(method='POST',
+                                    url=self.url,
+                                    data=freq,
+                                    headers=headers,
+                                    )
+        try:
+            if self.proxies == None or type(self.proxies) != dict:
+                self.proxies = {}
+            with requests.Session() as s:
+                s.proxies = self.proxies
+                r = s.send(request=response.prepare(),
+                           verify=False,
+                           timeout=self.timeout)
+            for line in r.iter_lines(chunk_size=1024):
+                decoded_line = line.decode('utf-8')
+                if "MkEWBc" in decoded_line:
+                    try:
+                        response = decoded_line
+                        response = json.loads(response)
+                        response = list(response)
+                        response = json.loads(response[0][2])
+                        response_ = list(response)
+                        response = response_[1][0]
+                        if len(response) == 1:
+                            if len(response[0]) > 5:
+                                sentences = response[0][5]
+                            else: ## only url
+                                sentences = response[0][0]
+                                if pronounce == False:
+                                    return sentences
+                                elif pronounce == True:
+                                    return [sentences,None,None]
+                            translate_text = ""
+                            for sentence in sentences:
+                                sentence = sentence[0]
+                                translate_text += sentence.strip() + ' '
+                            translate_text = translate_text
+                            if pronounce == False:
+                                return translate_text
+                            elif pronounce == True:
+                                pronounce_src = (response_[0][0])
+                                pronounce_tgt = (response_[1][0][0][1])
+                                return [translate_text, pronounce_src, pronounce_tgt]
+                        elif len(response) == 2:
+                            sentences = []
+                            for i in response:
+                                sentences.append(i[0])
+                            if pronounce == False:
+                                return sentences
+                            elif pronounce == True:
+                                pronounce_src = (response_[0][0])
+                                pronounce_tgt = (response_[1][0][0][1])
+                                return [sentences, pronounce_src, pronounce_tgt]
+                    except Exception as e:
+                        raise e
+            r.raise_for_status()
+        except requests.exceptions.ConnectTimeout as e:
+            raise e
+        except requests.exceptions.HTTPError as e:
+            # Request successful, bad response
+            raise google_new_transError(tts=self, response=r)
+        except requests.exceptions.RequestException as e:
+            # Request failed
+            raise google_new_transError(tts=self)
+
+    def detect(self, text):
+        text = str(text)
+        if len(text) >= 5000:
+            return log.debug("Warning: Can only detect less than 5000 characters")
+        if len(text) == 0:
+            return ""
+        headers = {
+            "Referer": "http://translate.google.{}/".format(self.url_suffix),
+            "User-Agent":
+                "Mozilla/5.0 (Windows NT 10.0; WOW64) "
+                "AppleWebKit/537.36 (KHTML, like Gecko) "
+                "Chrome/47.0.2526.106 Safari/537.36",
+            "Content-Type": "application/x-www-form-urlencoded;charset=utf-8"
+        }
+        freq = self._package_rpc(text)
+        response = requests.Request(method='POST',
+                                    url=self.url,
+                                    data=freq,
+                                    headers=headers)
+        try:
+            if self.proxies == None or type(self.proxies) != dict:
+                self.proxies = {}
+            with requests.Session() as s:
+                s.proxies = self.proxies
+                r = s.send(request=response.prepare(),
+                           verify=False,
+                           timeout=self.timeout)
+
+            for line in r.iter_lines(chunk_size=1024):
+                decoded_line = line.decode('utf-8')
+                if "MkEWBc" in decoded_line:
+                    # regex_str = r"\[\[\"wrb.fr\",\"MkEWBc\",\"\[\[(.*).*?,\[\[\["
+                    try:
+                        # data_got = re.search(regex_str,decoded_line).group(1)
+                        response = (decoded_line + ']')
+                        response = json.loads(response)
+                        response = list(response)
+                        response = json.loads(response[0][2])
+                        response = list(response)
+                        detect_lang = response[0][2]
+                    except Exception:
+                        raise Exception
+                    # data_got = data_got.split('\\\"]')[0]
+                    return [detect_lang, LANGUAGES[detect_lang.lower()]]
+            r.raise_for_status()
+        except requests.exceptions.HTTPError as e:
+            # Request successful, bad response
+            log.debug(str(e))
+            raise google_new_transError(tts=self, response=r)
+        except requests.exceptions.RequestException as e:
+            # Request failed
+            log.debug(str(e))
+            raise google_new_transError(tts=self)

trans_test.py

@@ -0,0 +1,5 @@
+from trans_google import google_translator
+
+translator = google_translator()
+translate_text = translator.translate('Hello World', lang_tgt='zh')
+print(translate_text)