detect korean and change interface

app.py

@@ -71,27 +71,31 @@ def text2img(prompt, negative_prompt, x, y, isRandom, fixedRandom ,num_inference
        seed = int(fixedRandom)
     generator = torch.manual_seed(seed)
 
-    # Translate prompt with negative prompt
-    #allPrompt = (translate(prompt +"/"+ negative_prompt).split("/"))
+    # Check Koran prompts
     allPrompt = ["",""]
-    allPrompt[0] = translate_mBart(prompt)
-    allPrompt[1] = translate_mBart(negative_prompt)
+    if prompt.upper() != prompt.lower():
+       print("prompt is an alphabet")
+       allPrompt[0] = prompt
+    else:
+       print("prompt is not an alphabet")
+       allPrompt[0] = translate_mBart(prompt)
+    
+    if negative_prompt != "":
+        if negative_prompt.upper() != negative_prompt.lower():
+            print("negative prompt is an alphabet")
+            allPrompt[1] = negative_prompt
+        else:
+            print("negative prompt is not an alphabet")
+            allPrompt[1] = translate_mBart(negative_prompt)
+    else:
+        negative_prompt = ""
     
-    print(len(allPrompt))
+    print("prompts length : "+ str(len(allPrompt)))
     print("prompt : " + allPrompt[0])
     print("negative prompt : " + allPrompt[1])
-    _prompt = allPrompt[0]
-    if len(allPrompt) > 1:
-        _negative_prompt = allPrompt[1]
-    else:
-       _negative_prompt = " "
     
-    # Check about it is English
-    if _prompt.upper() != _prompt.lower():
-       print(" it is an alphabet")
-    else:
-       print(" it is not an alphabet")
-       _prompt = "traffic sign of stop says SDXL"
+    _prompt = allPrompt[0]
+    _negative_prompt = allPrompt[1]
 
     #_negative_prompt = translate(negative_prompt)
     image = pipeline(
@@ -99,18 +103,20 @@ def text2img(prompt, negative_prompt, x, y, isRandom, fixedRandom ,num_inference
         ).images[0]
     
     _seed = str(seed)
-    _prompt = re.sub(r"[^\uAC00-\uD7A30-9a-zA-Z\s]", "", _prompt)
+    #_prompt = re.sub(r"[^\uAC00-\uD7A30-9a-zA-Z\s]", "", _prompt)
     timestamp = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
     #image.save( "./output/" + "sdxl_base_"  + "_seed_" + _seed+ "_time_" + timestamp +".png")
     #image.save("sdxl_prompt_" + "_seed_" + _seed + ".png")
     print(seed)
 
+    _info = "prompt : " + _prompt + " / negative prompt : " + _negative_prompt + " / seed : " + _seed + " / refine : " +str(refine) +" / time : " + timestamp
+
     if refine:
        image = img2img(prompt=_prompt, negative_prompt=_negative_prompt, image=image)
 
-       return image
+       return image, _info
     
-    return image 
+    return image, _info 
 
 
 def img2img(prompt, negative_prompt, image):
@@ -125,9 +131,21 @@ def img2img(prompt, negative_prompt, image):
     
 demo = gr.Interface(
    fn=text2img, 
-   inputs=["text", "text", gr.Slider(0,2048), gr.Slider(0,2048), gr.Checkbox(["random"]), "number", "number", "number", gr.Checkbox(["refine"])], 
-   outputs=["image"], 
+   inputs=[gr.Text(label=("prompt"), value=""), 
+           gr.Text(label=("negative prompt"), value=""), 
+           gr.Slider(128, 2048, value=1024, label="width"), 
+           gr.Slider(128, 2048, value=1024, label="height"),
+           gr.Checkbox(label="auto random seed", value=True),
+           gr.Slider(1, 999999, label="fixed random seed"), 
+           gr.Slider(1, 120, value=20, label="inference steps"), 
+           gr.Slider(1, 16, value=7, label="guidance scale"), 
+           gr.Checkbox(["refine"])
+           ],  
+   outputs=[gr.Image(label="generated image"),
+            gr.Textbox(label="properties")], 
    title ="한글로 하는 SDXL",
+   #description="한글로 하는 SDXL",
+   #article="test"
    )
 
 demo.launch()