update extract prompt

app.py

@@ -94,18 +94,24 @@ def generate_image(
             continue
 
         for prompt_index in range(prompt_count):
-            edit_prompt = ""
+            old_edit_prompt = ""
             for image_index in range(image_count):
                 completed_count += 1
                 styleImageUrl = prefix + style_image[0]
                 result_image, edit_prompt, error_msg = change_image_style(
-                    inputImageUrl, styleImageUrl, edit_prompt, extract_model, extract_prompt, prompt_prefix
+                    inputImageUrl, styleImageUrl, old_edit_prompt, extract_model, extract_prompt, prompt_prefix
                 )
 
                 if result_image:
                     success_count += 1
                     result_image_list.append(result_image)
-                    markdownStr += f"{edit_prompt} <img src='{result_image}' style='zoom: 33%;' />\n"
+                    if old_edit_prompt:
+                        markdownStr += f"<img src='{result_image}' style='zoom: 33%;' />"
+                    else:
+                        markdownStr += f"{edit_prompt}<img src='{result_image}' style='zoom: 33%;' />"
+
+                    old_edit_prompt = edit_prompt
+                        
                 else:
                     error_count += 1
                     error_list.append(f"Error processing style image {style_index + 1}, prompt {prompt_index + 1}, image {image_index + 1}: {error_msg}")
@@ -140,7 +146,7 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     image_count = gr.Number(label="Image Count", value=3, precision=0)
             extract_model = gr.Dropdown(choices=extract_model_tag_list, label="Extract Model", value="o3")
-            extract_prompt = gr.Textbox(lines=2, label="Extracted Prompt", value="Accurately extract everything from this photo into a prompt especially the hairstyle excluding the backgrounds. I want to replicate it perfectly. need to be a complete paragraph")
+            extract_prompt = gr.Textbox(lines=2, label="Extracted Prompt", value="Accurately extract the hairstyle and makeup from this photo into a prompt especially excluding the backgrounds. And Turn to the side, wearing deep blue Off-shoulder Knitted top.  I want to replicate it perfectly. need to be a complete paragraph")
             prompt_prefix = gr.Textbox(lines=1, label="Prompt Prefix", value="change the backgrounds to match this:")
 
             submit = gr.Button("Submit")