update demo

app.py

@@ -134,14 +134,13 @@ def run(
                 image.save(f"{image_name}_{i}{ext}")
     return output_image
 
-
 def get_example():
-    case = [
+    cases = [
         [
             "The sun rises slightly, the dew on the rose petals in the garden is clear, a crystal ladybug is crawling to the dew, the background is the early morning garden, macro lens.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             None,
             None,
@@ -160,7 +159,7 @@ def get_example():
             "A snow maiden with pale translucent skin, frosty white lashes, and a soft expression of longing",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             None,
             None,
@@ -179,7 +178,7 @@ def get_example():
             "Add a fisherman hat to the woman's head",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/flux5.png"),
             None,
@@ -198,7 +197,7 @@ def get_example():
             " replace the sword with a hammer.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(
                 ROOT_DIR,
@@ -221,7 +220,7 @@ def get_example():
             # "Transform the sculpture into jade",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(
                 ROOT_DIR, "example_images/46e79704-c88e-4e68-97b4-b4c40cd29826.png"
@@ -242,7 +241,7 @@ def get_example():
             "Make he smile",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(
                 ROOT_DIR, "example_images/vicky-hladynets-C8Ta0gwPbQg-unsplash.jpg"
@@ -263,7 +262,7 @@ def get_example():
             "Change the background to classroom",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/ComfyUI_temp_mllvz_00071_.png"),
             None,
@@ -282,7 +281,7 @@ def get_example():
             "Raise his hand",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(
                 ROOT_DIR,
@@ -304,7 +303,7 @@ def get_example():
             "Generate a photo of an anime-style figurine placed on a desk. The figurine model should be based on the character photo provided in the attachment, accurately replicating the full-body pose, facial expression, and clothing style of the character in the photo, ensuring the entire figurine is fully presented. The overall design should be exquisite and detailed, soft gradient colors and a delicate texture, leaning towards a Japanese anime style, rich in details, with a realistic quality and beautiful visual appeal.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/RAL_0315.JPG"),
             None,
@@ -323,7 +322,7 @@ def get_example():
             "Change the dress to blue.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/1.png"),
             None,
@@ -342,7 +341,7 @@ def get_example():
             "Remove the cat",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(
                 ROOT_DIR,
@@ -364,7 +363,7 @@ def get_example():
             "In a cozy café, the anime figure is sitting in front of a laptop, smiling confidently.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/ComfyUI_00254_.png"),
             None,
@@ -383,7 +382,7 @@ def get_example():
             "Create a wedding figure based on the girl in the first image and the man in the second image. Set the background as a wedding hall, with the man dressed in a suit and the girl in a white wedding dress. Ensure that the original faces remain unchanged and are accurately preserved. The man should adopt a realistic style, whereas the girl should maintain their classic anime style.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/1_20241127203215.png"),
             os.path.join(ROOT_DIR, "example_images/000050281.jpg"),
@@ -402,7 +401,7 @@ def get_example():
             "Let the girl  and the boy get married in the church. ",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/8FtFUxRzXqaguVRGzkHvN.png"),
             os.path.join(ROOT_DIR, "example_images/01194-20240127001056_1024x1536.png"),
@@ -421,7 +420,7 @@ def get_example():
             "Let the man from image1 and the woman from image2 kiss and hug",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/1280X1280.png"),
             os.path.join(ROOT_DIR, "example_images/000077066.jpg"),
@@ -440,7 +439,7 @@ def get_example():
             "Please let the person in image 2 hold the toy from the first image in a parking lot.",
             1024,
             1024,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/04.jpg"),
             os.path.join(ROOT_DIR, "example_images/000365954.jpg"),
@@ -459,7 +458,7 @@ def get_example():
             "Make the girl pray in the second image.",
             1024,
             682,
-            'euler',
+            "euler",
             50,
             os.path.join(ROOT_DIR, "example_images/000440817.jpg"),
             os.path.join(ROOT_DIR, "example_images/000119733.jpg"),
@@ -474,11 +473,55 @@ def get_example():
             1024 * 1024,
             0,
         ],
+        [
+            "Edit the first image: add the man from the second image. The man is talking with a woman in the kitchen",
+            1024,
+            682,
+            "euler",
+            50,
+            os.path.join(
+                ROOT_DIR,
+                "example_images/woman_kitchen.webp",
+            ),
+            os.path.join(ROOT_DIR, "example_images/a_man.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "Edit the first image: replace the woman with the person from the second image. The person is watching his phone in the kitchen",
+            1024,
+            682,
+            "euler",
+            50,
+            os.path.join(
+                ROOT_DIR,
+                "example_images/woman_kitchen.webp",
+            ),
+            os.path.join(ROOT_DIR, "example_images/a_man.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
         # [
         #     "Add the bird from image 1 to the desk in image 2",
         #     1024,
         #     682,
-        #     'euler',
+        #     "euler",
         #     50,
         #     os.path.join(
         #         ROOT_DIR,
@@ -496,33 +539,33 @@ def get_example():
         #     1024 * 1024,
         #     0,
         # ],
-        # [
-        #     "Replace the apple in the first image with the cat from the second image",
-        #     1024,
-        #     780,
-        #     'euler',
-        #     50,
-        #     os.path.join(ROOT_DIR, "example_images/apple.png"),
-        #     os.path.join(
-        #         ROOT_DIR,
-        #         "example_images/468404374-d52ec1a44aa7e0dc9c2807ce09d303a111c78f34da3da2401b83ce10815ff872.png",
-        #     ),
-        #     None,
-        #     NEGATIVE_PROMPT,
-        #     5.0,
-        #     2.0,
-        #     0.0,
-        #     1.0,
-        #     1,
-        #     2048,
-        #     1024 * 1024,
-        #     0,
-        # ],
+        [
+            "Edit the first image: replace the apple with the cat from the second image.",
+            1024,
+            780,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/apple.png"),
+            os.path.join(
+                ROOT_DIR,
+                "example_images/468404374-d52ec1a44aa7e0dc9c2807ce09d303a111c78f34da3da2401b83ce10815ff872.png",
+            ),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            2.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
         # [
         #     "Replace the woman in the second image with the woman from the first image",
         #     1024,
         #     747,
-        #     'euler',
+        #     "euler",
         #     50,
         #     os.path.join(
         #         ROOT_DIR, "example_images/byward-outfitters-B97YFrsITyo-unsplash.jpg"
@@ -541,6 +584,139 @@ def get_example():
         #     1024 * 1024,
         #     0,
         # ],
+        [
+            "She holds the sunflower close to her chest, eyes closed and smiling softly, standing on a sunlit balcony with the awakening city skyline behind her.",
+            1024,
+            1536,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/woman1.webp"),
+            None,
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            2.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "She does push-ups on the moonlit beach, staring intensely at the camera. Moonlight highlights her strong, focused expression, with waves crashing beside her on the glistening sand.",
+            1024,
+            1536,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/woman1.webp"),
+            None,
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "She is sitting on a motorcycle, framed by a daytime backdrop with swaying palm trees.",
+            1024,
+            1536,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/woman1.webp"),
+            None,
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            2.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "The two people shown in the images are sitting in a theater, watching the screen. One person points at the other person.",
+            1024,
+            1024,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/man1.webp"),
+            os.path.join(ROOT_DIR, "example_images/woman1.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "The two people shown in the images are sitting in a theater, watching the screen. One person points at the other person.",
+            1024,
+            1024,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/saml-altman-openai-ceo.webp"),
+            os.path.join(ROOT_DIR, "example_images/elon-twitter-new-ceo.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "The two people shown in the images are drinking wine, their eyes locked in an intense gaze at the camera, seated at an elegantly set table in a dimly lit room. Soft, golden light spills from a nearby chandelier, casting a warm glow over the polished wood surface, while the ambient sound of quiet conversation fills the air.",
+            1024,
+            1024,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/man1.webp"),
+            os.path.join(ROOT_DIR, "example_images/woman1.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
+        [
+            "The two people shown in the images are drinking wine, their eyes locked in an intense gaze at the camera, seated at an elegantly set table in a dimly lit room. Soft, golden light spills from a nearby chandelier, casting a warm glow over the polished wood surface, while the ambient sound of quiet conversation fills the air.",
+            1024,
+            1024,
+            "euler",
+            50,
+            os.path.join(ROOT_DIR, "example_images/saml-altman-openai-ceo.webp"),
+            os.path.join(ROOT_DIR, "example_images/elon-twitter-new-ceo.webp"),
+            None,
+            NEGATIVE_PROMPT,
+            5.0,
+            3.0,
+            0.0,
+            1.0,
+            1,
+            2048,
+            1024 * 1024,
+            0,
+        ],
         [
             "The cat is sitting on the table. The bird is perching on the edge of the table.",
             800,
@@ -567,7 +743,7 @@ def get_example():
             0,
         ],
     ]
-    return case
+    return cases
 
 
 def run_for_examples(