init test

.gitignore

@@ -0,0 +1 @@
+.idea/

app.py

@@ -0,0 +1,83 @@
+import gradio as gr
+
+import numpy as np
+from interface_modules.X2Painting.template_config import style_example
+from interface_modules.X2Painting.client_process import send_to_server
+
+
+def on_select(evt: gr.SelectData):
+    index = evt.index
+    style_name = list(style_example.values())[index]
+    return gr.update(value=style_name)
+
+
+css = """
+        .gradio-container {background-color: #F0F5FF; width: 95% !important}
+        """
+title = r"""
+    </br>
+    <h1 align="center" style="font-size: 42px;">Character2Painting</h1>
+    <h2 align="center"> <span style="color: red;">Character</span> <--- <span style="color: black;">Zoom out⭐️⭐⭐⭐⭐Zoom in ---> </span> <span style="color: red;">Painting</span></h2>
+    </br>
+
+    </br>   
+    <img src="file/docs/resource/X2Painting/pics/xword_intro.png" style="display: block; margin: 0 auto; max-height: 384px;">
+    </br>
+    <h2 style="text-align: center;">===================🤪🥳 Have a Try 🤩😄===================</h2>
+    """
+
+with gr.Blocks(css=css) as demo:
+    # description
+    gr.HTML(title)
+    with gr.Row():
+        with gr.Column():
+            # with gr.Row(scale=1):
+            #     with gr.Column():
+            gr.Markdown('''
+                     ⭐️<b> User Tips </b>
+                    - <b>step1：</b>Input a Character. Recommended Chinese Character.
+                    - <b>step2：</b>Select a style in the Gallery
+                    - <b>step3：</b>Click Run,Waiting for about 1 Min. Enjoy
+                ''')
+            word = gr.Textbox(
+                label="Input Character",
+                info="please type Character, such as 李. (输入中文字，例如，李)",
+                value="李",
+                elem_id="InputCha"
+            )
+
+            submit = gr.ClearButton(value="Run",elem_id="RunBtn")
+            style_name = gr.Textbox(
+                label="style_name",
+                info="style_name",
+                value="", visible=False,
+            )
+
+        with gr.Column(scale=6):
+            gr.Markdown('''
+            <h1 align="center">Style Gallery</h1>
+            ''')
+            example_gallery = gr.Gallery(label="style_type", show_label=True, elem_id="example_gallery",
+                                         value=list(style_example.keys()), columns=10
+                                         )
+    # vis result gallery
+    gr.Markdown('''
+    <h1 align="center">Result Gallery</h1>
+    ''')
+    final_gallery = gr.Gallery(
+        label="最终生成图(paintings)，第一行：refine后生成结果，第二行：初始生成结果，第三行：refine前的文字增强处理图",
+        show_label=False,
+        elem_classes="final_gallery",
+    ).style(columns=[4], rows=[2])
+
+    # server_port = gr.Number(value=args.server_port, visible=False)
+    submit.add([final_gallery])
+    submit.click(send_to_server,
+                 inputs=[word, style_name],
+                 outputs=[final_gallery])
+    example_gallery.select(on_select, None,
+                           [style_name])
+
+demo.queue()
+demo.launch(share=True, server_name="0.0.0.0", server_port=12404)
+# demo.launch(share=True)

interface_modules/X2Painting/client_process.py

@@ -0,0 +1,54 @@
+import gradio
+
+import interface_modules.X2Painting.server_config as server_config
+import json
+import requests
+from utils.data_convert import base64_to_image
+from interface_modules.WordArt.template_config import font_style_dict
+
+
+def send_to_server(word, style_name):
+    send_data = server_config.send_data.copy()
+    # fill send_data
+    send_data["gen_num"] = 4  #
+
+    # task specific
+    send_data['extra_input_infor'] = {
+        'WordArt':
+            {
+                "style": style_name,
+                "word": word,
+            }
+    }
+
+    # prepare to send
+    post_addr = server_config.post_addr
+    json_data = json.dumps(send_data)
+    response = requests.post(post_addr, data=json_data, headers=server_config.headers)
+
+    # process feed back images
+    if response.status_code != 200:
+        print("post error")
+        return [None]
+
+    print("HF XPainting post sucess, process image")
+    result_dict = json.loads(response.text)
+    if result_dict.get("error_code") != 0:
+        print("error", "-" * 30)
+        print(result_dict["infor"])
+        gradio.Error(result_dict["infor"])
+        return [None]
+
+    result_img_list = result_dict["image"]
+    enhance_output = []
+    loop1_output = []
+    for i, img_obj in enumerate(result_img_list):
+        final_img = base64_to_image(img_obj["image"])  # 最终结果图片 PIL格式
+        enhance_output.append((final_img, "{}_{}_enhance.jpg".format(word, i)))
+
+        if "loop1_images" in img_obj['extra_image']:
+            loop1_images = img_obj['extra_image']["loop1_images"]
+            img = base64_to_image(loop1_images)
+            loop1_output.append((img, "{}_{}_loop1_images.jpg".format(word, i)))
+
+    return enhance_output + loop1_output

interface_modules/X2Painting/server_config.py

@@ -0,0 +1,7 @@
+post_addr = ""
+
+send_data = {}
+send_data["task_type"] = "gen_WordArt"
+send_data["visual_id"] = "-1"
+
+headers = {'Content-Type': 'application/json'}

interface_modules/X2Painting/template_config.py

@@ -0,0 +1,21 @@
+style_example = {
+    "./docs/resource/X2Painting/gradio_examples/01_te.jpg": "butterflygirl_black",
+    "./docs/resource/X2Painting/gradio_examples/02_si.jpg": "male_black",
+    "./docs/resource/X2Painting/gradio_examples/03_lian.jpg": "tree_black",
+    "./docs/resource/X2Painting/gradio_examples/04_wei.jpg": "winter_black",
+    "./docs/resource/X2Painting/gradio_examples/05_lai.jpg": "dunhuanggirl_yellow",
+    "./docs/resource/X2Painting/gradio_examples/06_cheng.jpg": "shuimocartoon_green",
+    "./docs/resource/X2Painting/gradio_examples/07_shi.jpg": "cartoongirl3_green",
+    "./docs/resource/X2Painting/gradio_examples/08_shi.jpg": "phoenix_white",
+    "./docs/resource/X2Painting/gradio_examples/09_yan.jpg": "cartoongirl2_while",
+    "./docs/resource/X2Painting/gradio_examples/10_shi.jpg": "gufenggirl_white",
+    "./docs/resource/X2Painting/gradio_examples/11_ren.jpg": "shuimocartoonfox_black",
+    "./docs/resource/X2Painting/gradio_examples/12_gong.jpg": "shuimocartoongirl2_green",
+    "./docs/resource/X2Painting/gradio_examples/13_zhi.jpg": "shuimowood2_black",
+    "./docs/resource/X2Painting/gradio_examples/14_neng.jpg": "shuimowood_black",
+    "./docs/resource/X2Painting/gradio_examples/15_chan.jpg": "tiankongzhicheng_black",
+    "./docs/resource/X2Painting/gradio_examples/16_pin.jpg": "shuimocartoongirl_black",
+    "./docs/resource/X2Painting/gradio_examples/17_ce.jpg": "shuimoflower_black",
+    "./docs/resource/X2Painting/gradio_examples/18_shi.jpg": "cartoongirl_black",
+    "./docs/resource/X2Painting/gradio_examples/19_ban.jpg": "shuimodesigner_black",
+}

requirements.txt

@@ -0,0 +1,3 @@
+gradio==3.35.2
+requests
+Pillow==9.5.0