Add image files using Git LFS

app copy.py

@@ -1,94 +0,0 @@
-import gradio as gr
-import asyncio
-import fal_client
-from dotenv import load_dotenv
-import os
-from pathlib import Path
-import time
-
-load_dotenv()
-os.environ["FAL_KEY"] = os.getenv("FAL_API_KEY")
-
-async def generate_paris_images(image1_path: str, image2_path: str, woman_prompt: str, man_prompt: str, batch_size: int, progress=gr.Progress()):
-    start_time = time.time()
-    print("Progress: 5% - Starting Paris image generation...")
-    progress(0.05, desc="Starting Paris image generation...")
-    
-    # Upload all images in parallel
-    upload_tasks = [
-        fal_client.upload_file_async(str(image1_path)),
-        fal_client.upload_file_async(str(image2_path)),
-        fal_client.upload_file_async("test_images/woman.png"),
-        fal_client.upload_file_async("test_images/man.png"),
-        fal_client.upload_file_async("test_images/clipspace-mask-4736783.png"),
-        fal_client.upload_file_async("test_images/clipspace-mask-4722992.png")
-    ]
-    
-    [image1_url, image2_url, woman_img, man_img, mask1_img, mask2_img] = await asyncio.gather(*upload_tasks)
-    
-    print("Progress: 40% - Uploaded all images")
-    progress(0.4, desc="Uploaded all images")
-
-    handler = await fal_client.submit_async(
-        "comfy/LVE/paris-couple",
-        arguments={
-            "loadimage_1": image1_url,
-            "loadimage_2": image2_url,
-            "loadimage_3": woman_img,
-            "loadimage_4": mask1_img,
-            "loadimage_5": mask2_img,
-            "loadimage_6": man_img,
-            "woman_prompt": woman_prompt,
-            "man_prompt": man_prompt,
-            "batch_size": batch_size
-        }
-    )
-
-    print("Progress: 60% - Processing images...")
-    progress(0.6, desc="Processing images...")
-    
-    result = await handler.get()
-    print(result)
-
-    end_time = time.time()
-    processing_time = end_time - start_time
-    print(f"Progress: 100% - Generation completed in {processing_time:.2f} seconds")
-    progress(1.0, desc=f"Generation completed in {processing_time:.2f} seconds")
-    
-    # Return all generated image URLs and processing time
-    # Get the first key from outputs dynamically
-    return (
-        [img["url"] for img in result["outputs"][next(iter(result["outputs"]))]["images"]] if "outputs" in result and result["outputs"] else [],
-        f"Processing time: {processing_time:.2f} seconds"
-    )
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        image1_input = gr.Image(label="Upload Woman Image", type="filepath", value="test_images/user3-f.jpg")
-        image2_input = gr.Image(label="Upload Man Image", type="filepath", value="test_images/user3.jpg")
-    
-    with gr.Row():
-        woman_prompt = gr.Textbox(
-            label="Woman Prompt",
-            value="Close-up, portrait photo, a woman, Paris nighttime romance scene, wearing an elegant black dress with a shawl, standing beneath the same canopy of twinkling lights along the Champs-Élysées, the Eiffel Tower glowing bright in the distance, soft mist rising from the street, looking at the camera."
-        )
-        man_prompt = gr.Textbox(
-            label="Man Prompt",
-            value="Close-up, portrait photo, a man, Paris nighttime romance scene, wearing a tailored suit with a crisp white shirt, standing beneath a canopy of twinkling lights along the Champs-Élysées, the Eiffel Tower glowing bright in the distance, soft mist rising from the street, looking at the camera."
-        )
-    
-    batch_size = gr.Slider(minimum=1, maximum=8, value=4, step=1, label="Batch Size")
-    
-    generate_btn = gr.Button("Generate")
-    image_output = gr.Gallery(label="Generated Image")
-    time_output = gr.Textbox(label="Processing Time")
-    
-    generate_btn.click(
-        fn=generate_paris_images,
-        inputs=[image1_input, image2_input, woman_prompt, man_prompt, batch_size],
-        outputs=[image_output, time_output]
-    )
-
-if __name__ == "__main__":
-    print("Starting Gradio interface...")
-    demo.launch()

app.py

@@ -5,52 +5,61 @@ from dotenv import load_dotenv
 import os
 from pathlib import Path
 import time
+import json
 
 load_dotenv()
 os.environ["FAL_KEY"] = os.getenv("FAL_API_KEY")
 
-async def generate_paris_images(product_name: str, image1_path: str, image2_path: str, woman_prompt: str, man_prompt: str, batch_size: int, progress=gr.Progress()):
+async def generate_paris_images(product_name: str, image1_path: str, image2_path: str, woman_prompt: str, man_prompt: str, girl_name: str, girl_hair_length: str, girl_hair_style: str, girl_hair_color: str, boy_name: str, boy_hair_length: str, boy_hair_style: str, boy_hair_color: str, batch_size: int, progress=gr.Progress()):
     start_time = time.time()
     print("Progress: 5% - Starting Paris image generation...")
     progress(0.05, desc="Starting Paris image generation...")
     
-    product_number = product_name.split(" ")[1]
-    man_img = f"Movie Pose Style/Option {product_number}/man{product_number}.png"
-    woman_img = f"Movie Pose Style/Option {product_number}/woman{product_number}.png"
-    man_mask = f"Movie Pose Style/Option {product_number}/clip-m.png"
-    woman_mask = f"Movie Pose Style/Option {product_number}/clip-f.png"
-
-    print(f"man_img: {man_img}")
-    print(f"woman_img: {woman_img}")
-    print(f"man_mask: {man_mask}")
-    print(f"woman_mask: {woman_mask}")
-
     # Upload all images in parallel
     upload_tasks = [
         fal_client.upload_file_async(str(image1_path)),
         fal_client.upload_file_async(str(image2_path)),
-        fal_client.upload_file_async(man_img),
-        fal_client.upload_file_async(woman_img),
-        fal_client.upload_file_async(woman_mask),
-        fal_client.upload_file_async(man_mask)
+        fal_client.upload_file_async("template/man_pose.png"),
+        fal_client.upload_file_async("template/woman_pose.png"),
+        fal_client.upload_file_async("template/woman_clip_mask.png"),
+        fal_client.upload_file_async("template/man_clip_mask.png")
     ]
     
-    [image1_url, image2_url, man_img, woman_img, mask1_img, mask2_img] = await asyncio.gather(*upload_tasks)
+    [image1_url, image2_url, man_pose_img, woman_pose_img, woman_clip_mask, man_clip_mask] = await asyncio.gather(*upload_tasks)
     
     print("Progress: 40% - Uploaded all images")
     progress(0.4, desc="Uploaded all images")
 
+    # Replace {hair_feature} placeholders with user hair descriptions
+    woman_hair_desc = f"{girl_hair_length} {girl_hair_style} {girl_hair_color} hair,"
+    print(f"Final woman hair description: {woman_hair_desc}")
+
+    # Handle bald case for man's hair description
+    if boy_hair_length == "Bald":
+        man_hair_desc = "bald,"
+    else:
+        man_hair_desc = f"{boy_hair_length} {boy_hair_style} {boy_hair_color} hair,"
+    print(f"Final man hair description: {man_hair_desc}")
+    
+    woman_prompt = woman_prompt.replace("{hair_feature}", woman_hair_desc)
+    man_prompt = man_prompt.replace("{hair_feature}", man_hair_desc)
+
+    print(f"Final woman prompt: {woman_prompt}")
+    print(f"Final man prompt: {man_prompt}")
+
     handler = await fal_client.submit_async(
         "comfy/LVE/paris-couple",
         arguments={
             "loadimage_1": image1_url,
             "loadimage_2": image2_url,
-            "loadimage_3": woman_img,
-            "loadimage_4": mask1_img,
-            "loadimage_5": mask2_img,
-            "loadimage_6": man_img,
+            "loadimage_3": woman_pose_img,
+            "loadimage_4": woman_clip_mask,
+            "loadimage_5": man_clip_mask,
+            "loadimage_6": man_pose_img,
             "woman_prompt": woman_prompt,
             "man_prompt": man_prompt,
+            "girl_name": girl_name,
+            "boy_name": boy_name,
             "batch_size": batch_size
         }
     )
@@ -66,61 +75,96 @@ async def generate_paris_images(product_name: str, image1_path: str, image2_path
     print(f"Progress: 100% - Generation completed in {processing_time:.2f} seconds")
     progress(1.0, desc=f"Generation completed in {processing_time:.2f} seconds")
     
+    # Fix the URL extraction logic
+    image_215 = []
+    image_818 = []
+    if "outputs" in result:
+        if "215" in result["outputs"]:
+            image_215 = [img["url"] for img in result["outputs"]["215"]["images"]]
+        if "818" in result["outputs"]:
+            image_818 = [img["url"] for img in result["outputs"]["818"]["images"]]
+
+    print(f"Image 215: {image_215}")
+    print(f"Image 818: {image_818}")
+
     # Return all generated image URLs and processing time
     # Get the first key from outputs dynamically
     return (
-        [img["url"] for img in result["outputs"][next(iter(result["outputs"]))]["images"]] if "outputs" in result and result["outputs"] else [],
+        image_215,
+        image_818,
         f"Processing time: {processing_time:.2f} seconds"
     )
 
 def change_product_preview(product_name):
-    if product_name == "Pose 1":
-        return "Movie Pose Style/Option 1/image.png"
-    elif product_name == "Pose 2":
-        return "Movie Pose Style/Option 2/image.png"
-    elif product_name == "Pose 3":
-        return "Movie Pose Style/Option 3/image.png"
-    elif product_name == "Pose 4":
-        return "Movie Pose Style/Option 4/image.png"
-    elif product_name == "Pose 5":
-        return "Movie Pose Style/Option 5/image.png"
-    elif product_name == "Pose 6":
-        return "Movie Pose Style/Option 6/image.png"
+    # Load prompts from JSON file
+    with open('prompt.json', 'r') as f:
+        prompts = json.load(f)
+    
+    # Find the matching prompt data
+    prompt_data = next((item for item in prompts if item['title'] == product_name), None)
+    
+    if prompt_data:
+        return (
+            f"thumbnail/{product_name}.png",
+            prompt_data['woman'],
+            prompt_data['man']
+        )
+    return None, "", ""
 
 with gr.Blocks() as demo:
     with gr.Row():
-        product_name = gr.Dropdown(label="Product Name", choices=["Pose 1", "Pose 2", "Pose 3", "Pose 4", "Pose 5", "Pose 6"], value="Pose 1")
-        product_preview = gr.Image(label="Product Preview", type="filepath", value="Movie Pose Style/Option 1/image.png", height=500, width=500)
+        product_name = gr.Dropdown(label="Product Name", choices=["Winter", "Classy", "Night Out", "Romantic"], value="Winter")
+        product_preview = gr.Image(label="Product Preview", type="filepath", value="thumbnail/Winter.png", height=500, width=500)
     with gr.Row():
         image1_input = gr.Image(label="Upload Woman Image", type="filepath", value="user3-f.jpg")
         image2_input = gr.Image(label="Upload Man Image", type="filepath", value="user3-m.jpg")
     
     with gr.Row():
-        woman_prompt = gr.Textbox(
-            label="Woman Prompt",
-            value="Close-up, intimate portrait, a woman wearing a delicate pink floral embellished dress with a flowing tulle skirt, long blonde wavy hair cascading down her back, standing on the same snowy cobblestone street, the Eiffel Tower in the background, soft snowflakes falling around them, gentle streetlights casting a warm glow, her arms wrapped around her partner's neck, faces close together in an intimate moment, sharing a gentle smile."
-        )
-        man_prompt = gr.Textbox(
-            label="Man Prompt",
-            value="Close-up, intimate portrait, a man wearing a crisp white button-down shirt with rolled-up sleeves, khaki pants, standing on a snowy cobblestone street, the Eiffel Tower dusted with snow in the background, soft snowflakes falling around them, gentle streetlights casting a warm glow, his hands tenderly embracing his partner's waist, looking lovingly at her with a warm smile."
-        )
+        with gr.Column():
+            woman_prompt = gr.Textbox(
+                label="Woman Prompt",
+                value="Close-up, portrait photo, a woman, {hair_feature} wearing a cream-colored wool coat, chunky knit scarf, and matching earmuffs, standing on the same snow-dusted cobblestone street, illuminated Eiffel Tower glowing golden in the background, snowflakes sparkling in the warm streetlight glow, looking at camera with gentle smile."
+            )
+            girl_name = gr.Textbox(
+                label="Girl Name",
+                value="julie delpy"
+            )
+            girl_hair_length = gr.Dropdown(label="Girl Hair Length", choices=["Short", "Medium", "Long"], value="Long")
+            girl_hair_style = gr.Dropdown(label="Girl Hair Style", choices=["Straight", "Wavy", "Curly"], value="Straight")
+            girl_hair_color = gr.Dropdown(label="Girl Hair Color", choices=["Blonde", "Brown", "Black", "Brunette", "Redhead", "Bronde"], value="Bronde")
+        with gr.Column():
+            man_prompt = gr.Textbox(
+                label="Man Prompt",
+                value="Close-up, portrait photo, a man, {hair_feature} wearing a dark navy wool peacoat, cashmere scarf, and leather gloves, standing on a snow-dusted cobblestone street, illuminated Eiffel Tower in the background glowing golden against the night sky, gentle snowflakes catching the warm glow of vintage streetlamps, looking at camera with confident expression."
+            )
+            boy_name = gr.Textbox(
+                label="Boy Name",
+                value="ethan hawke"
+            )
+            boy_hair_length = gr.Dropdown(label="Boy Hair Length", choices=["Short", "Medium", "Long", "Bald"], value="Short")
+            boy_hair_style = gr.Dropdown(label="Boy Hair Style", choices=["None", "Undercut", "Mullet", "French Crop", "Slicked Back", "Fade", "Buzz Cut"], value="Undercut")
+            boy_hair_color = gr.Dropdown(label="Boy Hair Color", choices=["None", "Blonde", "Brown", "Black", "Brunette", "Redhead"], value="Black")
     
     batch_size = gr.Slider(minimum=1, maximum=8, value=4, step=1, label="Batch Size")
     
     generate_btn = gr.Button("Generate")
-    image_output = gr.Gallery(label="Generated Image")
+
+    with gr.Row():
+        image_output = gr.Gallery(label="Generated Image Raw")
+        image_output_processed = gr.Gallery(label="Generated Image Final")
+
     time_output = gr.Textbox(label="Processing Time")
     
     generate_btn.click(
         fn=generate_paris_images,
-        inputs=[product_name, image1_input, image2_input, woman_prompt, man_prompt, batch_size],
-        outputs=[image_output, time_output]
+        inputs=[product_name, image1_input, image2_input, woman_prompt, man_prompt, girl_name, girl_hair_length, girl_hair_style, girl_hair_color, boy_name, boy_hair_length, boy_hair_style, boy_hair_color, batch_size],
+        outputs=[image_output, image_output_processed, time_output]
     )
 
     product_name.change(
         fn=change_product_preview,
         inputs=[product_name],
-        outputs=[product_preview]
+        outputs=[product_preview, woman_prompt, man_prompt]
     )
 
 if __name__ == "__main__":

prompt.json

@@ -0,0 +1,22 @@
+[
+    {
+    "title": "Winter",
+    "man": "Close-up, portrait photo, a man, {hair_feature} wearing a dark navy wool peacoat, cashmere scarf, and leather gloves, standing on a snow-dusted cobblestone street, illuminated Eiffel Tower in the background glowing golden against the night sky, gentle snowflakes catching the warm glow of vintage streetlamps, looking at camera with confident expression.",
+    "woman": "Close-up, portrait photo, a woman, {hair_feature} wearing a cream-colored wool coat, chunky knit scarf, and matching earmuffs, standing on the same snow-dusted cobblestone street, illuminated Eiffel Tower glowing golden in the background, snowflakes sparkling in the warm streetlight glow, looking at camera with gentle smile."
+  },
+  {
+    "title": "Classy",
+    "man": "Close-up, portrait photo, a man, {hair_feature} wearing a perfectly tailored black tuxedo with silk lapels and white pocket square, standing beneath ornate Art Nouveau streetlamps on Champs-Élysées, Eiffel Tower's golden lights twinkling in the distance, misty evening atmosphere, looking at camera with sophisticated expression.",
+    "woman": "Close-up, portrait photo, a woman, {hair_feature} wearing a floor-length black evening gown with delicate lace details and silk wrap, standing beneath the same ornate streetlamps, Eiffel Tower's golden lights sparkling behind, misty evening atmosphere, looking at camera with elegant poise."
+  },
+  {
+    "title": "Night Out",
+    "man": "Close-up, portrait photo, a man, {hair_feature} wearing a stylish casual outfit with dark fitted jeans and a well-tailored button-down shirt, standing on a charming Parisian street corner, Eiffel Tower illuminated against the night sky, warm streetlights creating ambient glow, looking at camera with relaxed confidence.",
+    "woman": "Close-up, portrait photo, a woman, {hair_feature} wearing a chic casual ensemble with high-waisted jeans and an elegant blouse, standing on the same charming street corner, illuminated Eiffel Tower in background, warm streetlights creating soft ambiance, looking at camera with casual sophistication."
+  },
+  {
+    "title": "Romantic",
+    "man": "Close-up, portrait photo, a man, {hair_feature} wearing a tailored navy double-breasted peacoat, burgundy cashmere sweater, and charcoal dress pants, standing on Pont Alexandre III bridge, ornate golden statues illuminated behind him, Eiffel Tower glowing majestically in the distance, nighttime Seine reflecting the bridge lights, looking at camera with charming expression.",
+    "woman": "Close-up, portrait photo, a woman, {hair_feature} wearing a plaid black and white blazer, red beret, flowing brown hair, and statement earrings, standing on the same ornate bridge, golden statues illuminated behind her, Eiffel Tower glowing in the distance, nighttime Seine reflecting the bridge lights, looking at camera with playful sophistication."
+  }
+]