Upload 13 files

app.py

@@ -4,9 +4,13 @@ from train_tab import create_train_tab
 from virtualtryon_tab import create_virtualtryon_tab
 from faceswap_tab import create_faceswap_tab
 from ipadapter_tab import create_ipadaptor_tab
+from controlnetmockup_tab import create_cnmu_tab
+#from mail_tab import create_mail_tab
+
 from pattern_ip_adapter import sam_zest_tab
 from dotenv import load_dotenv, find_dotenv
 import os 
+
 from src.utils import convert_to_pil
 gallery_list=[]
 
@@ -18,30 +22,56 @@ def update_gallery(img):
 
 
 theme= gr.themes.Soft(
-    radius_size=gr.themes.sizes.radius_none,
+    
     primary_hue=gr.themes.colors.emerald, 
     secondary_hue=gr.themes.colors.green)
-css = """
-    h1{font-size: 2em}
-    h3{margin-top: 10}
-    #component-1{text-align:left}
-    .main_ui_logged_out{opacity: 0.3; pointer-events: none}
-    .tabitem{border: 20px}
-    .tabitem h1 {
-    font-size: 5em; /* Adjust this value to make the headings bigger */
-}
-
-    .group_padding{padding: .55em}
-    """
+css  = """
+    .gradio-container img {
+        display: block;
+        width: 100%;
+        height: auto;
+        max-height: 200px;
+        object-fit: cover;
+        border-radius: 10px;
+    }
+
+    .gradio-container p {
+        margin: 0;
+        font-size: 16px;
+        font-weight: bold;
+    }
+
+    .gradio-container .gr-button {
+        width: 100%;
+        background-color: #fff;
+        border: 1px solid #ddd;
+        padding: 10px;
+        border-radius: 10px;
+    }
+    .output_image {
+       display: block;
+       margin-left: auto;
+       margin-right: auto;
+        width: 100%;
+        height: auto;
+     
+    }
+    .gradio-container .gr-button:hover {
+        background-color: #f0f0f0;
+    }
+"""
 
 _ = load_dotenv(find_dotenv())
 with gr.Blocks(theme=theme,css=css) as demo:
+
+
     with gr.Tabs() as tabs:
-        gen_out,move_to_vto,move_to_fs,move_to_ip,move_to_zest= create_gen_tab()
+        gen_out,move_to_vto,move_to_fs,move_to_ip= create_gen_tab()
         input_ip,output_ip=create_ipadaptor_tab()
         input_vto,output_vto=create_virtualtryon_tab()
-        input_zest,output_zest=sam_zest_tab()
+        #input_zest,output_zest=sam_zest_tab()
         input_fs,output_fs=create_faceswap_tab()
+        input_cnmu,output_cnmu=create_cnmu_tab()
         #create_train_tab()
      
 
@@ -49,15 +79,16 @@ with gr.Blocks(theme=theme,css=css) as demo:
     move_to_vto.click(lambda x:x, inputs=gen_out,outputs=input_vto)      
     move_to_fs.click(lambda x:x, inputs=gen_out,outputs=input_fs)
     move_to_ip.click(lambda x:x, inputs=gen_out,outputs=input_ip)
-    move_to_zest.click(lambda x:x, inputs=gen_out,outputs=input_zest)
-
+   # move_to_zest.click(lambda x:x, inputs=gen_out,outputs=input_zest)
     #Gallery 
-    image_gallery = gr.Gallery(label="Generated Images Gallery",type="pil")
+    image_gallery = gr.Gallery(label="Generated Images Gallery",type="pil",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)
     output_fs.change(update_gallery,inputs=output_fs,outputs=image_gallery)
     output_ip.change(update_gallery,inputs=output_ip,outputs=image_gallery)
-    output_zest.change(update_gallery,inputs=output_zest,outputs=image_gallery)
+    output_cnmu.change(update_gallery,inputs=output_cnmu,outputs=image_gallery)
+    #output_zest.change(update_gallery,inputs=output_zest,outputs=image_gallery)
+    
 
 demo.launch(share=True,show_error=True,debug=True) #,auth=[("username", "password"),(os.getenv("APP_USER"),os.getenv("APP_PW"))])

controlnetmockup_tab.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from src.utils import add_to_prompt
+from src.rep_api import replicate_bgcontrolnet,replicate_caption_api
+import os
+
+
+
+def create_cnmu_tab():
+    with gr.TabItem("Control net mockup Generator"):
+        with gr.Accordion("HowTo",open=False):
+            gr.Markdown("How to Goes Here")
+        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"],
+                    ],
+                        inputs=[style_image,background_prompt]
+                    )
+            
+        with gr.Row():
+            out_img= gr.Image(label="OutputImage")
+        
+        btn.click(replicate_bgcontrolnet,inputs=[in_img,object_prompt,background_prompt],outputs=[out_img])
+        in_img.change(replicate_caption_api,
+                        inputs=[in_img,gr.Textbox("blip",visible=False),gr.Textbox("What item is in this image reply with a one word if u can, i.e Shoe, person, car ....",visible=False)],
+                        outputs=[object_prompt])
+    return in_img,out_img
+
+
+        

gen_tab.py

@@ -20,8 +20,9 @@ prompt_dict_kp ={
 lora_style_support = [
      ["ps1","Test_images/lora_support_examples/ps1.png"],
      ["cyberpunk","Test_images/lora_support_examples/cyberpunk.png"],
-     ["Raylean","Test_images/lora_support_examples/Raylene.png"],
-     ["AppleSeed","Test_images/lora_support_examples/appleseed.png"],
+     ["Sioux","Test_images/lora_support_examples/Raylene.png"],
+     ["RetroAnime1","Test_images/lora_support_examples/appleseed.png"],
+     ["RetroAnime2","Test_images/lora_support_examples/appleseed.png"],
       ["half-illustration","Test_images/lora_support_examples/halfillust.png"]
 ]
 
@@ -37,10 +38,33 @@ prompt_support = [
 
 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.\n"
+                                "5. Click 'Generate' to create the image. \n"
+                                "6. click button to move image to other tab. \n")
+                    
+                with gr.Column():
+                    
+                    gr.Markdown("### 画像生成器の使い方\n"
+                                "1. テキストボックスにプロンプトを入力します。\n"
+                                "2. アスペクト比を選択します。\n"
+                                "3. スタイルロアを選択します。\n"
+                                "4. 'Extra'でコントロールネットのオプションを選択します。\n"
+                                "5. 'Generate'をクリックして画像を生成します。\n"
+                                "6. ボタンをクリックして画像を他のタブに移動します。\n")
+        #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="Aspect Ratio")
+                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):
                     #Control net 
@@ -48,11 +72,15 @@ def create_gen_tab():
                     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"],
+                        inputs=[control_net_img]
+                    )
 
               
             with gr.Column():
                 
-                style_mode = gr.Dropdown(list(model_dict.keys()),label="Style lore",value="Base",info="style model")
+                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):
@@ -61,7 +89,7 @@ def create_gen_tab():
                         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 safytensor route goes here or replicate link goes here",value=None)
+                    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")
                 
@@ -78,24 +106,21 @@ def create_gen_tab():
                         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)
-            
-            #gr.Button(key,interactive=False)
-            #for value in values:
-            #    gr.Button(value).click(add_to_prompt, inputs=[inp, gr.Textbox(value,visible=False)], outputs=inp)
+         
 
         with gr.Row():
-            gen_out = gr.Image(label="Generated Image",type="filepath")
-       # with gr.Accordion("Gallery", open=False):
-        #    gen_gallery = gr.Gallery(label="Generated Images Gallery", type="filepath")
+            gen_out = gr.Image(label="Generated Image",type="filepath",elem_id="output_image")
+       
 
         with gr.Row(): #Move gen out row
             move_to_vto =gr.Button("Move to Virtual Try On")    
-            move_to_fs =gr.Button("Move to Face Swap") 
             move_to_ip =gr.Button("Move to Cloth IP Adapter")     
-            move_to_zest=gr.Button("Move to Zest IP Adapter")
+            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_fs,move_to_ip,move_to_zest
+        
+        return gen_out,move_to_vto,move_to_fs,move_to_ip #,move_to_zest

ipadapter_tab.py

@@ -46,11 +46,15 @@ def create_ipadaptor_tab():
                             ["Test_images/pattern_2.jpg"],
                             ["Test_images/pattern_3.jpg"],
                             ["Test_images/pattern_4.jpg"],
+                            ["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]
                     )
       with gr.Row():
-          api_out = gr.Image(label="Output",type="filepath")
+          api_out = gr.Image(label="Output",type="filepath",elem_id="output_image")
 
     ip_btn.click(fal_ipadapter_api,inputs=[api_inp,ip_image,ap_prompt],outputs=api_out)
 

mail_tab.py

@@ -0,0 +1,102 @@
+import gradio as gr
+import smtplib
+import numpy as np
+from email.mime.text import MIMEText
+from email.mime.image import MIMEImage
+from email.mime.multipart import MIMEMultipart
+from email.utils import formatdate
+import os 
+
+def send_email(email, selected_images):
+    # Email configuration (replace with your SMTP server details)
+    smtp_server = "smtp.gmail.com"
+    smtp_port = 587
+    # Create the email message
+    msg = MIMEMultipart()
+    msg['Subject'] = 'Selected Image'
+    msg['From'] = os.getenv("sender_email")
+    msg['To'] = email
+    msg['Date'] = formatdate()
+
+    # Add text to the email
+    text = MIMEText(f"""
+    今日はブースを訪れてくれてありがとうございました。。
+  　以下は、あなたが生成した{len(selected_images)}枚の画像です。
+    """,_charset="utf-8")
+    msg.attach(text)
+    print(selected_images)
+    for i,image in enumerate(selected_images):
+      image=np.array(image)[0] 
+      print(image)
+      with open(image, 'rb') as f:
+          img_data = f.read()
+      image = MIMEImage(img_data, name=f"image_{i+1}.jpg")
+      msg.attach(image)
+
+    # Send the email
+    try:
+        with smtplib.SMTP(smtp_server, smtp_port) as server:
+            server.starttls()
+            server.login( os.getenv("sender_email"), os.getenv("sender_password"))
+            server.send_message(msg)
+        return "Email sent successfully!"
+    except smtplib.SMTPAuthenticationError:
+        raise Exception("SMTP Authentication failed. Please check your email and password.")
+    except smtplib.SMTPException as e:
+        raise Exception(f"SMTP error occurred: {str(e)}")
+    except Exception as e:
+        raise Exception(f"Failed to send email: {str(e)}")
+    
+
+
+
+examples= ["Test_images/Woman_1.png","Test_images/prompt_support_examples/Man_1.png"]    
+select_list=[]      
+
+from src.utils import convert_to_pil
+def update_gallery(img,select_list=select_list):
+    #img=convert_to_pil(img)
+    if select_list is None:
+        select_list=[]
+    select_list.append(img)
+    return select_list
+
+def clear_selection(select_list=select_list):
+    if select_list is not None:
+        select_list=[]
+    return select_list
+
+
+def create_mail_tab():
+    
+    with gr.TabItem("Mail", id="mail"):
+        email_input = gr.Textbox(label="Enter your email:")
+        image_gallery = gr.Gallery(label="Selected Images", show_label=True, elem_id="gallery")
+        with gr.Row():
+            clear_button = gr.Button("Clear Selection")
+            send_button = gr.Button("Send Email")
+        result_output = gr.Textbox(label="Result")
+
+        # Examples
+        with gr.Row():
+            print(examples)
+            examples=np.array(examples)
+            for example in examples:
+                print(example)
+                img=gr.Image(value=example, interactive=False, label=os.path.basename(example),type="filepath")
+                img.select(update_gallery,inputs=img,outputs=image_gallery)
+
+        # Connect the send button to the send_email function
+        send_button.click(
+            fn=send_email,
+            inputs=[email_input, image_gallery],
+            outputs=result_output
+        )
+
+        # Clear the selected images
+        clear_button.click(
+            fn=clear_selection,
+            inputs=None,
+            outputs=image_gallery,
+          
+        )

model_dict.json

@@ -2,7 +2,8 @@
     "Base":["xlabs-ai/flux-dev-realism:39b3434f194f87a900d1bc2b6d4b983e90f0dde1d5022c27b52c143d670758fa","https://huggingface.co/XLabs-AI/flux-RealismLora/resolve/main/lora.safetensors","REALISM"],
     "ps1":["veryvanya/flux-ps1-style:e785bcd326ea7a2c6efef8c43122a289f2c344370a4b74b2c7fa1a5ff4c38fea","https://huggingface.co/veryVANYA/ps1-style-flux/resolve/main/ps1_style_flux_v1.safetensors","ps1 game screenshot"],
     "cyberpunk":["fofr/flux-80s-cyberpunk:5d0cefd0746b833042b384c3a310bc4d1f9d1304ec59ba93e75097d40b967180","https://huggingface.co/fofr/flux-80s-cyberpunk/resolve/main/lora.safetensors","style of 80s cyberpunk"],
-    "Raylean": ["jarvis-labs2024/flux-raylene:5574556226d11e0f10855a957d91f118a9178c8fc77e7e7b18830627ce3184f1","https://replicate.delivery/yhqm/jpM5dp9JqOoYPdvF2lXsdWC0cV0XBHWQHmflBYz9L3IegyTTA/trained_model.tar"," SOK"],
-    "AppleSeed": ["jarvis-labs2024/flux-appleseed:0aecb9fdfb17a2517112cc70b4a1898aa7791da84a010419782ce7043481edec","https://replicate.delivery/yhqm/6v6JOdXtArJZEBbsjPoYSC6lezM2Is12eghIoQBhlnWNI0UTA/trained_model.tar","style of APPLESEED anime"],
+    "Sioux": ["jarvis-labs2024/sioux-flux:19afc63be3ff9838cf9d1225617f8ab0d6f8a9b2b8d6e836ee08d4a0f109c33f","https://replicate.delivery/yhqm/l8drADSeN6S2ekAF5cofeB7yn4C6X57r0VlvYjEbFVjTJbpNB/trained_model.tar"," SOK"],
+    "RetroAnime1": ["jarvis-labs2024/flux-appleseed:0aecb9fdfb17a2517112cc70b4a1898aa7791da84a010419782ce7043481edec","https://replicate.delivery/yhqm/6v6JOdXtArJZEBbsjPoYSC6lezM2Is12eghIoQBhlnWNI0UTA/trained_model.tar","style of APPLESEED anime"],
+    "RetroAnime2": ["galverse/mama-flux-v1:dec77f91f71155f16c6bfe7df2cbacec18d0e4816ee772be48738ea9f9423a30","https://replicate.delivery/yhqm/Jg4kWsYyHnIwMl7XgeWY280o6lfNk3LMh0MxOSo2mUV2q5VTA/trained_model.tar","Retro anime style, GALVERSE"],
     "half-illustration":["davisbrown/flux-half-illustration:687458266007b196a490e79a77bae4b123c1792900e1cb730a51344887ad9832","https://huggingface.co/davisbro/half_illustration/resolve/main/flux_train_replicate.safetensors?download=true","In the style of TOK,vibrant illustration"]
 }

virtualtryon_tab.py

@@ -60,8 +60,18 @@ def create_virtualtryon_tab():
         with gr.Row():
             tryon_btn = gr.Button("Try-on")
         with gr.Row():
-            tryon_out = gr.Image(label="Output", type="filepath")
-            
+            tryon_out = gr.Image(label="Output", type="filepath",elem_id="output_image")
+        with gr.Row():
+            tryon_examples = gr.Examples(
+                examples=[
+                    ["Test_images/example_outputs/input_1.webp",True,"Test_images/pink_jumper.png","pink jumper","upper_body","Test_images/example_outputs/input_1_VTO_Pink.jpg"],  
+                    ["Test_images/example_outputs/input_3.png",True,"Test_images/Jacket_2.png","jacket","upper_body","Test_images/example_outputs/input_3_VTO_whiteJacket.jpg"],  
+                    ["Test_images/example_outputs/input_4.png",True,"Test_images/Jacket_1.png","colorfull leather jacket","upper_body","Test_images/example_outputs/Input_4_VTO_ColorFullJacket.jpg"], 
+                    ["Test_images/example_outputs/input_4.png",True,"Test_images/dress_1.png","dress","dresses","Test_images/example_outputs/Input_3_VTO_LoritaDress.jpg"],   
+                    ["Test_images/example_outputs/input_4.png",True,"Test_images/Shirt_1.png"," Shirt","upper_body","Test_images/example_outputs/Input_3_VTO_Shirt.jpg"],     
+                ],
+                inputs=[human_img,crop,garm_img,garment_des,category,tryon_out],
+            ) 
        
     tryon_btn.click(virtual_try_on, inputs=[crop, seed, steps, category,  garm_img, human_img,garment_des],outputs=tryon_out)