Upload 14 files

app.py

@@ -7,16 +7,62 @@ from ipadapter_tab import create_ipadaptor_tab
 from relighting_tab import gen_relighting_tab
 from controlnetmockup_tab import create_cnmu_tab
 #from mail_tab import create_mail_tab
-
+from PIL import Image
 from pattern_ip_adapter import sam_zest_tab
 from dotenv import load_dotenv, find_dotenv
 import os 
 from src.utils import convert_to_pil
+from datetime import datetime
+
 gallery_list=[]
+# Ensure the base folder for storing images exists
+base_dir = "gallery_images"
+os.makedirs(base_dir, exist_ok=True)
+
+
+
+# Function to convert and save the image to a dated folder, and update the gallery
+def update_gallery_local(img):
+    print(type(img), len(gallery_list))
+
+    try:
+        # Convert the image to PIL if needed
+        img = convert_to_pil(img)
+    except:
+        print("Error with converting to PIL")
+        print(type(img),img)
+        return gallery_list
+        
+
+    # Get the current date and create a folder for it
+    current_date = datetime.now().strftime("%Y-%m-%d")
+    date_folder = os.path.join(base_dir, current_date)
+    os.makedirs(date_folder, exist_ok=True)
+
+    # Generate a unique filename for the image
+    img_filename = f"generated_image_{len(gallery_list)}.png"
+    img_path = os.path.join(date_folder, img_filename)
+    
+    # Save the image to the date folder
+    img.save(img_path)
+
+    # Add the file path to the gallery list
+    gallery_list.append(img_path)
+
+    # Return the updated gallery list with image paths
+    return gallery_list
 
 def update_gallery(img):
-    img=convert_to_pil(img)
-    gallery_list.append(img)
+
+    try:
+        print(type(img),print(len(gallery_list)))
+        img=convert_to_pil(img)
+        gallery_list.append(img)
+    except:
+        print("Error with saving image to gallery")
+        #img=Image.open(img)
+        #gallery_list.append(img)
+        
     return gallery_list
 
 
@@ -34,15 +80,22 @@ css  = """
         border-radius: 10px;
     }
 
+
     .gradio-container p {
         margin: 0;
         font-size: 16px;
         font-weight: bold;
     }
 
+    .gen-btn {
+        background-color: green;
+        color: white;
+
+    }
+
     .gradio-container .gr-button {
         width: 100%;
-        background-color: #fff;
+        background-color: green;
         border: 1px solid #ddd;
         padding: 10px;
         border-radius: 10px;
@@ -70,7 +123,7 @@ with gr.Blocks(theme=theme,css=css) as demo:
         input_vto,output_vto,move_to_cnmk_fm_try,move_to_relight_fm_try=create_virtualtryon_tab()
         #input_zest,output_zest=sam_zest_tab()
         #input_fs,output_fs=create_faceswap_tab()
-        input_cnmu,output_cnmu=create_cnmu_tab()
+        input_cnmu,output_cnmu,move_to_relight_fm_try=create_cnmu_tab()
         input_rl,output_rl=gen_relighting_tab()
         #create_train_tab()
      
@@ -87,16 +140,18 @@ with gr.Blocks(theme=theme,css=css) as demo:
     #Move to Buttons from Virtual Try On
     move_to_cnmk_fm_try.click(lambda x:x,inputs=output_vto,outputs=input_cnmu)
     move_to_relight_fm_try.click(lambda x:x,inputs=output_vto,outputs=input_rl)
-   # move_to_zest.click(lambda x:x, inputs=gen_out,outputs=input_zest)
+    #Move to Button From Control Net Mockup
+    move_to_relight_fm_try.click(lambda x:x,inputs=output_cnmu,outputs=input_rl)
+
     #Gallery 
-    image_gallery = gr.Gallery(label="Generated Images Gallery",type="pil",elem_id="output_image")
+    image_gallery = gr.Gallery(label="Generated Images Gallery",type="path",elem_id="output_image")
     #Gallery updates  get all outputs 
-    gen_out.change(update_gallery,inputs=gen_out,outputs=image_gallery)
-    output_vto.change(update_gallery,inputs=output_vto,outputs=image_gallery)
+    gen_out.change(update_gallery_local,inputs=gen_out,outputs=image_gallery)
+    output_vto.change(update_gallery_local,inputs=output_vto,outputs=image_gallery)
     #output_fs.change(update_gallery,inputs=output_fs,outputs=image_gallery)
-    output_ip.change(update_gallery,inputs=output_ip,outputs=image_gallery)
-    output_cnmu.change(update_gallery,inputs=output_cnmu,outputs=image_gallery)
-    output_rl.change(update_gallery,inputs=output_rl,outputs=image_gallery)
+    output_ip.change(update_gallery_local,inputs=output_ip,outputs=image_gallery)
+    output_cnmu.change(update_gallery_local,inputs=output_cnmu,outputs=image_gallery)
+    output_rl.change(update_gallery_local,inputs=output_rl,outputs=image_gallery)
     #output_zest.change(update_gallery,inputs=output_zest,outputs=image_gallery)
     
 

controlnetmockup_tab.py

@@ -6,34 +6,39 @@ import os
 
 
 def create_cnmu_tab():
-    with gr.TabItem("Control net mockup Generator"):
+    with gr.TabItem("Background changer"):
         with gr.Accordion("HowTo",open=False):
             gr.Markdown("""
                         
-            # Mockup Generator  \n
+            # Background Generator  \n
             ・1 Select base image  ベース画像を選択します。 \n
             ・2 Update and check image object prompt 物体のプロンプトを確認と更新\n 
             ・3 Add in background prompt   背景のプロンプトを記入 \n
             ・4 Click 'Generate' to create the image. Generateをクリックして画像を生成します。\n
 
             """)
+            gr.Image(value="HowTo/controlnetmockup.png",label=None,interactive=False)
         with gr.Row():
             with gr.Column():
                 in_img = gr.Image(label="Input Image")
                 object_prompt = gr.Textbox(label="Object Prompt",info="日本語対応は可能")
                 background_prompt = gr.Textbox(label="Background Prompt",info="日本語対応は可能")
                 style_image = gr.Image(label="Style Image",visible=False)
-                btn = gr.Button("Generate")
+                
                 with gr.Accordion("Examples",open=False):
                     gr.Examples(examples=[
                         ["Test_images/backgrounds/Output1.png"," in the stress of japan, 8k uhd, dslr, soft lighting, high quality, film grain , colorful decorative signs in"],
+                        ["Test_images/example_outputs/mockup_BG_suitshop.png","suit shop in London, wooden inertia "]
                          
                     ],
                         inputs=[style_image,background_prompt]
                     )
-            
+        with gr.Row():
+            btn = gr.Button("Generate",elem_id="gen-btn")
         with gr.Row():
             out_img= gr.Image(label="OutputImage")
+        with gr.Row():
+            move_to_relight= gr.Button("Move to Relight")
         
         btn.click(replicate_bgcontrolnet,inputs=[in_img,object_prompt,background_prompt],outputs=[out_img])
         in_img.change(replicate_caption_api,
@@ -43,10 +48,13 @@ def create_cnmu_tab():
         gr.Examples(
             examples=[
             ["Test_images/example_outputs/cnmu_input_1.webp","Test_images/example_outputs/cnmu_output_1.png","a  bowl of soup","on a table in a Chinese restraint, 8k, table, flowers in background "],
+            ["Test_images/prompt_support_examples/Jacket_1.png","Test_images/example_outputs/mockup_BG.png","a back view of a man", " a night city streets of Tokyo"],
+            ["Test_images/example_outputs/Input_3_VTO_LoritaDress.jpg","Test_images/example_outputs/mockup_BG_runway.png","a blond woman in a red and white dress is posing for a photo","on a Fashion show run way, lights "],
+            ["Test_images/prompt_support_examples/Man_1.png","Test_images/example_outputs/mockup_BG_suitshop.png","a young man in a suit posing for a photo","suit shop in London, wooden inertia "]
             ],
             inputs=[in_img,out_img,object_prompt,background_prompt]
         )
-    return in_img,out_img
+    return in_img,out_img,move_to_relight
 
 
         

gen_tab.py

@@ -20,9 +20,9 @@ prompt_dict_kp ={
 lora_style_support = [
      ["ps1","Test_images/lora_support_examples/ps1.png"],
      ["cyberpunk","Test_images/lora_support_examples/cyberpunk.png"],
-     ["Sioux","Test_images/lora_support_examples/Raylene.png"],
+     ["Sioux","Test_images/Walk_4.png"],
      ["RetroAnime1","Test_images/lora_support_examples/appleseed.png"],
-     ["RetroAnime2","Test_images/lora_support_examples/appleseed.png"],
+     ["RetroAnime2","Test_images/lora_support_examples/Galverse.png"],
       ["half-illustration","Test_images/lora_support_examples/halfillust.png"]
 ]
 
@@ -55,15 +55,15 @@ def create_gen_tab():
             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 画像サイズ アスペクト比")
-                btn = gr.Button("Generate")
-                with gr.Accordion("extra", open=False):
+                
+                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/Sit_1.png"],
+                        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]
                     )
 
@@ -84,7 +84,8 @@ def create_gen_tab():
                 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):

ipadapter_tab.py

@@ -13,21 +13,21 @@ def create_ipadaptor_tab():
         ・2 Prompt area you want the IP to be applied IPを適用するためのプロンプトエリアを選択します。 \n
         ・3 Select IP image   IP画像を選択します。 \n
         ・4 Click Generate  生成をクリックします。 \n
-          
-        ![](./HowTo/Flow_IP_ClothAdapter.jpg)
+      
         """)
-        gr.Image(value="HowTo/Flow_IP_ClothAdapter.jpg",interactive=False)
+        gr.Image(value="HowTo/Flow_IP_ClothAdapter.jpg",label=None,interactive=False)
         
 
       with gr.Row():
         with gr.Column():
           api_inp = gr.Image(label="Base Image")
-          ap_prompt = gr.Textbox(label="clothe prompt")
-          with gr.Accordion("Example People", open=False):
+          
+          with gr.Accordion("Example base images", open=False):
                     human_examples = gr.Examples(
                          examples=[
                             ["Test_images/Woman_1.png"],
                             ["Test_images/man_1.png"],
+                            ["Test_images/example_outputs/input_4.png"],
                             ["Test_images/Woman_2.png"],
                             ["Test_images/Woman_3.png"],
                             ["Test_images/man_2.png"],
@@ -39,16 +39,17 @@ def create_ipadaptor_tab():
                             ["Test_images/Jump.png"],
                             ["Test_images/Walk_1.png"],
                             ["Test_images/Walk_2.png"],
-                            ["Test_images/Walk_girl_1.png"],
-                            ["Test_images/Walk_girl_2.png"]
+                            ["Test_images/Walk_3.png"],
+                            ["Test_images/Walk_4.png"]
                         ],
-                        inputs=[api_inp]
+                        inputs=[api_inp],
+                         examples_per_page=14
                     )
         
         with gr.Column():
           ip_image = gr.Image(label="IP Adapter Image")
-          ip_btn = gr.Button("Process")
-          with gr.Accordion("Example Stles and clothes ", open=False):
+          
+          with gr.Accordion("Example IP images ", open=False):
                     ip_examples = gr.Examples(
                         examples=[
                             ["Test_images/style_1.jpg"],
@@ -59,13 +60,22 @@ def create_ipadaptor_tab():
                             ["Test_images/pattern_2.jpg"],
                             ["Test_images/pattern_3.jpg"],
                             ["Test_images/pattern_4.jpg"],
+                            ["Test_images/Jacket_1.png"],
+                            ["Test_images/Suit_4.png"],
+                            ["Test_images/dress_5.png"],
+                            ["Test_images/Shirt_1.png"],
                             ["https://replicate.delivery/pbxt/Kl23gJODaW7EuxrDzBG9dcgqRdMaYSWmBQ9UexnwPiL7AnIr/3.jpg"],
                             ["https://replicate.delivery/pbxt/Kl2WefehduxwWcQc5OrrBH6AkojQ6OqyQSKBvBLrroSpEBim/f2f0488a-180e-4d7e-9907-f26f92ac5f16.jpg"],
                             ["https://replicate.delivery/pbxt/Kl2VlUibviSP8Kq5ULLJmMOWorog1YFu0zTreqhqX97c62ku/572a1fc9-a114-4d5b-8c7c-85aa5648c7b4.jpg"],
                             ["https://replicate.delivery/pbxt/Kl2VCw1UVIJsYw9r8iqSYUMm65ePJhfYOLNolOE8CwxfRjX2/28481ff0-0829-42af-a658-fb96be2abb3d.jpg"],
                         ],
-                        inputs=[ip_image]
+                        inputs=[ip_image],
+                         examples_per_page=14
                     )
+      with gr.Row():
+        ap_prompt = gr.Textbox(label="clothes prompt",value="clothes",info="Clothe are you want to use 希望の服のエリア (日本語可能)")
+      with gr.Row():
+          ip_btn = gr.Button("Generate",elem_id="gen_btn")
       with gr.Row():
           api_out = gr.Image(label="Output",type="filepath",elem_id="output_image")
       with gr.Row(): #Move gen out row
@@ -74,4 +84,12 @@ def create_ipadaptor_tab():
             move_to_relight= gr.Button("Move to Relight")
     ip_btn.click(fal_ipadapter_api,inputs=[api_inp,ip_image,ap_prompt],outputs=api_out)
 
+    with gr.Row():
+        gr.Examples(examples=[
+            ["Test_images/example_outputs/input_4.png","Test_images/Jacket_1.png","Jacket","Test_images/example_outputs/output_clothIP_1.png"],
+            ["Test_images/Woman_2.png","Test_images/pattern_4.jpg","Dress","Test_images/example_outputs/output_clothIP_2.png"],
+            ["","","",""],
+            ["","","",""],
+        ],inputs=[api_inp,ip_image,ap_prompt,api_out])
+
     return api_inp,api_out,move_to_cnmk,move_to_relight

relighting_tab.py

@@ -6,13 +6,27 @@ from src.rep_api import replicate_iclight_BG,light_source_options
 def gen_relighting_tab():
     with gr.TabItem("Re-Lighting"):
         gr.Markdown("### Upload Images and Set Light Source")
+
+        with gr.Accordion("HowTo",open=False):
+            gr.Markdown("""
+                        
+            # Relight Generator  \n
+            ・1 Select base image  ベース画像を選択します。 \n
+            ・2 Update and check image object prompt 物体のプロンプトを確認と更新\n 
+            ・3 Add in background prompt   背景のプロンプトを記入 \n
+            ・4 Click 'Generate' to create the image. Generateをクリックして画像を生成します。\n
+
+            """)
+            gr.Image(label=None,value="HowTo/ReLight.png",interactive=False)
         
         img_input = gr.Image(label="Input Image", type="pil")
         bg_img_input = gr.Image(label="Background Image", type="pil")
         
         with gr.Row():
             with gr.Column():
-                gr.Examples(examples=[
+                gr.Examples(
+                    label="example Images ",
+                    examples=[
                     ["Test_images/anime_woman_2.png"],
                     ["Test_images/prompt_support_examples/Woman_1.png"],
                     ["Test_images/example_outputs/Input_4_VTO_ColorFullJacket.jpg"],
@@ -21,7 +35,9 @@ def gen_relighting_tab():
                 
                     ],inputs=[img_input])
             with gr.Column():
-                gr.Examples(examples=
+                gr.Examples(
+                    label="Example Light Source",
+                    examples=
                             [
                     ["https://replicate.delivery/pbxt/KxPIbJUjSmVBlvn0M3C8PAz6brN5Z0eyZSGcKIVw3XfJ6vNV/7.webp"],
                     ["Test_images/Lighting/Ligthing_1.webp"],
@@ -34,8 +50,9 @@ def gen_relighting_tab():
         prompt_input = gr.Textbox(label="Prompt", placeholder="Enter prompt here")
         light_source = gr.Dropdown(label="Light Source", choices=light_source_options, value="Use Background Image")
 
+        bg_img_input.change( lambda x: "Use Background Image",inputs=bg_img_input, outputs=light_source)
         
-        submit_btn = gr.Button("Generate")
+        submit_btn = gr.Button("Generate",elem_id="gen-btn")
         output_image = gr.Image(label="Output Image")
 
         # Link the submit button to the replicate_iclight_BG function
@@ -44,5 +61,14 @@ def gen_relighting_tab():
             inputs=[img_input, prompt_input, bg_img_input, light_source], 
             outputs=output_image
         )
-        
-        return img_input, bg_img_input
+        examples = gr.Examples(
+                examples=[
+                    ["Test_images/Walk_4.png","Test_images/Lighting/Lighing_3.webp","Use Background Image","A girl walking","Test_images/example_outputs/Re_light_output1.png"],  
+                    ["Test_images/prompt_support_examples/Woman_1.png","https://replicate.delivery/pbxt/KxPIbJUjSmVBlvn0M3C8PAz6brN5Z0eyZSGcKIVw3XfJ6vNV/7.webp","Use Background Image","a girl wearing a jacket ","Test_images/example_outputs/Re_light_output2.png"],
+                    ["Test_images/example_outputs/Input_4_VTO_ColorFullJacket.jpg",None,"Bottom Light","a girl wearing a jacket","Test_images/example_outputs/Re_light_output3.png"],
+                    ["Test_images/Woman_5.png",None,"Right Light","a image of a girl from the side","Test_images/example_outputs/Re_light_output4.png"],
+                    ["Test_images/Woman_5.png","Test_images/Lighting/Ligthing_1.webp","Use Background Image","a image of a girl from the side","Test_images/example_outputs/Re_light_output5.png"],
+                ],
+                inputs=[img_input,bg_img_input,light_source,prompt_input,output_image],
+            ) 
+        return img_input, output_image

virtualtryon_tab.py

@@ -30,7 +30,7 @@ def create_virtualtryon_tab():
                             ["Test_images/Woman_1.png"],
                             ["Test_images/prompt_support_examples/Man_1.png"],
                             ["Test_images/Woman_2.png"],
-                            ["Test_images/prompt_support_examples/Man2_2.png"],
+                            ["Test_images/prompt_support_examples/Man_2.png"],
                             ["Test_images/Woman_3.png"],
                             ["Test_images/man_1.png"],
                             ["Test_images/Woman_4.png"],
@@ -51,7 +51,9 @@ def create_virtualtryon_tab():
                     cloth_examples = gr.Examples(
                         examples=[
                             ["Test_images/pink_jumper.png","pink jumper","upper_body"],
+                             ["Test_images/Suit_1.png","Suit","upper_body"],
                             ["Test_images/Suit_2.png","Suit","upper_body"],
+                             ["Test_images/Suit_4.png","Suit","upper_body"],
                             ["Test_images/Jacket_1.png","Colorfull leather jacket","upper_body"],
                             ["Test_images/Jacket_2.png","White jacket","upper_body"],
                             ["Test_images/Shirt_1.png","Shirt","upper_body"],
@@ -60,17 +62,16 @@ def create_virtualtryon_tab():
                             ["Test_images/dress_3.png","pink dress","dresses"],
                             ["Test_images/dress_4.png","red dress","dresses"],
                              ["Test_images/dress_5.png","golden dress","dresses"],
-                              ["Test_images/Shoes_1.png","Shoes","lower_body"],
-                              ["Test_images/Shoes_2.png","Shoes","lower_body"],
-                              ["Test_images/Nike_RTFKT_Dunk_Void_1.jpg","lower_body"]
-
+                             ["Test_images/dress_6.png","brown dress","dresses"],
+                             ["Test_images/dress_7.png","pink dress","dresses"],
+                             ["Test_images/Suit_4.png","one-piece suit","upper_body"],
                         ],
                         inputs=[garm_img,garment_des,category],
-                        examples_per_page=4
+                        examples_per_page=8
                     )
 
         with gr.Row():
-            tryon_btn = gr.Button("Try-on")
+            tryon_btn = gr.Button("Try-on",elem_id="gen-btn")
         with gr.Row():
             tryon_out = gr.Image(label="Output", type="filepath",elem_id="output_image")
         with gr.Row(): #Move gen out row