gen_tab.py

import gradio as gr
from src.utils import add_to_prompt
from src.rep_api import generate_image_replicate,generate_image_control_net, model_dict
import os

# The dictionary data
prompt_dict = {
    "Character": ["Asian girl with black hair", "A man with blond hair", "A Cat girl anime character with purple hair", "A Green Alien with big black eyes"],
    "Clothes": ["Wearing a blue jacket", "Wearing a black business suit", "Wearing a purple jumpsuit", "Wearing shorts and a white T-shirt"],
    "Pose": ["Close up portrait", "Standing doing a peace sign", "Folding arms", "holding a phone"],
    "Style": ["Simple white background", "Fashion runway", "Inside a business conference", "Inside a spaceship"],
}
prompt_dict_kp ={
    "Character": ["黒髪のアジア系女性", "金髪の男性", "紫髪のキャットガールアニメキャラクター", "大きな黒い目を持つ緑の宇宙人"],
    "Clothes": ["青いジャケットを着ている", "黒のビジネススーツを着ている", "紫のジャンプスーツを着ている", "ショーツと白いTシャツを着ている"],
    "Pose": ["クローズアップポートレート", "ピースサインをして立っている", "腕を組んでいる", "携帯電話を持っている"],
    "Scene": ["シンプルな白背景", "ファッションランウェイ", "ビジネス会議の中", "宇宙船の内部"]
}

lora_style_support = [
     ["ps1","Test_images/lora_support_examples/ps1.png"],
     ["cyberpunk","Test_images/lora_support_examples/cyberpunk.png"],
     ["Sioux","Test_images/Walk_4.png"],
     ["RetroAnime1","Test_images/lora_support_examples/appleseed.png"],
     ["RetroAnime2","Test_images/lora_support_examples/Galverse.png"],
      ["half-illustration","Test_images/lora_support_examples/halfillust.png"]
]

prompt_support = [
    ["Test_images/prompt_support_examples/Man_1.png","A full body photo man  standing with blond hair wearing a suit"],
    ["Test_images/prompt_support_examples/Woman_1.png","A full body photo girl  standing with blond hair wearing a suit "],
    ["Test_images/prompt_support_examples/Woman_2.png","a Asian girl as at a fashion show  walking down the cat walk, wearing a red dress"],
    ["Test_images/prompt_support_examples/Man_2.png","A High exposure photo of a Asian man wearing a blue shirt at a business conference, blue shirt, short black hair, hands out welcoming people into a booth, booth with a green and white color scheme "],
    ["Test_images/prompt_support_examples/Shoes_1.png","a product photo of high tech shoes with lights and a wifi symbol on the side, low angle shot, front view of walking down stairs "],
    ["Test_images/prompt_support_examples/Jacket_1.png","a product photo of green checked jacked being on a man from behind , furry collar, large cute character logo of a cat on back of jacket,  simple background,"],
]


def create_gen_tab():
    with gr.TabItem("Image Generator"):
        #Gen how too row 
        with gr.Row():
             
             with gr.Accordion("HowTo",open=False):
                with gr.Column():
                    gr.Markdown("### How to use the Image Generator 画像生成器の使い方 \n"
                                "1. Enter a prompt in the textbox. テキストボックスにプロンプトを入力します。\n"
                                "2. Choose the aspect ratio. アスペクト比を選択します。\n"
                                "3. Options Select a style lore. スタイルロアを選択します。\n"
                                "4. Options in extra control net. Extraでコントロールネットのオプションを選択します。\n"
                                "5. Click 'Generate' to create the image. Generateをクリックして画像を生成します。\n"
                                "6. click button to move image to other tab.ボタンをクリックして画像を他のタブに移動します。 \n")
                    gr.Image(value="HowTo/Gen.png",label=None,interactive=False)
        #Gen generatpor options 
        with gr.Row():
            with gr.Column():
                inp = gr.Textbox(label="Prompt",info="日本語対応は可能")
                aspect_ratio = gr.Dropdown(list(["1:1","16:9","9:16","5:4"]),value="1:1", label="Aspect Ratio", info="Image size Aspect Ratio 画像サイズ アスペクト比")
                
                with gr.Accordion("Control Net", open=False):
                    #Control net 
                    use_control_net=gr.Checkbox(label="Use Control net",value=False)
                    control_net_type=gr.Dropdown(["depth","canny","soft_edge"],value="depth",label="Control net type")
                    control_net_img= gr.Image(label="Control net image",type="pil")
                    control_net_strength = gr.Slider(0,1,value=0.4,label="control net strength")
                    control_net_examples = gr.Examples(
                        examples=["Test_images/Walk_3.png","Test_images/example_outputs/input_2.png","Test_images/example_outputs/input_4.png","Test_images/Sit_1.png","Test_images/controlnet-in-automatic1111-for-character-design-sheets-v0-ir6p0gdwpfia1.webp"],
                        inputs=[control_net_img]
                    )

              
            with gr.Column():
                
                style_mode = gr.Dropdown(list(model_dict.keys()),label="Style lore",value="Base",info="style model スタイルのモデル")
                with gr.Accordion("Style Imgaes",open=False):
                   # with gr.Column(elem_id="prompt_support_column"):
                    for i,lora in enumerate(lora_style_support):
                        if i % 3 == 0:
                            row = gr.Row()  # Start a new row
                        with row:
                            lora_img = gr.Image(value=lora[1], label=lora[0],width=200,height=200, interactive=False)
                            lora_img.select(lambda x:x, inputs=gr.Textbox(value=lora[0],visible=False),outputs=style_mode)
                    api_path = gr.Textbox(label="Other_style",info="Style lora .safetensors route goes here or replicate link goes here　スタイルのローラの.safetensorsのルートはこちら。",value=None)

                style_strength = gr.Slider(0,2,value=1,label="Style Strength")
                

        with gr.Row():
            btn = gr.Button("Generate")

        with gr.Accordion("Prompt Support", open=False):
            for i,item in enumerate(prompt_support):
                if i % 6 ==0:
                    row=gr.Row()
                with row:
                    with gr.Column(elem_id="prompt_support_column"):
                        img_ps=gr.Image(value=item[0],label="",interactive=False)
                        btn_ps=gr.Button(item[1],elem_id="prompt_support_button")
                        img_ps.select(lambda x:x, inputs=gr.Textbox(item[1],visible=False),outputs=inp)
                        btn_ps.click(lambda x:x, inputs=gr.Textbox(item[1],visible=False),outputs=inp)
         

        with gr.Row():
            gen_out = gr.Image(label="Generated Image",type="filepath",elem_id="output_image")
       

        with gr.Row(): #Move gen out row
            move_to_ip =gr.Button("Move to Cloth IP Adapter")     
            move_to_vto =gr.Button("Move to Virtual Try On")    
            move_to_cnmk =gr.Button("Move to Control net mockup ")
            move_to_relight= gr.Button("Move to Relight")
           # move_to_fs =gr.Button("Move to Face Swap")
           # move_to_zest=gr.Button("Move to Zest IP Adapter")


        btn.click(generate_image_control_net, 
                  inputs=[inp,style_mode,api_path,aspect_ratio,style_strength,use_control_net,control_net_type,control_net_img,control_net_strength], 
                  outputs=[gen_out])
        
        return gen_out,move_to_vto,move_to_ip,move_to_cnmk,move_to_relight# move_to_fs,#,move_to_zest