Update app.py

app.py

@@ -130,16 +130,16 @@ CAPTION_TYPE_MAP = {
     ],
 
     ("style_prompt", "formal", False, False): [
-        "Analyze this image through the lens of {style} art. Describe how it aligns with or diverges from {style_characteristics}.",
-        "Examine this artwork in the context of the {style} movement, focusing on {style_focus} and how these elements are represented or reinterpreted in the image."
+        "Generate a detailed stable diffusion prompt to recreate this image, including style, composition, and key elements.",
+        "Craft a comprehensive prompt for an AI art generator to produce an image in the same style and mood as this artwork."
     ],
     ("style_prompt", "formal", False, True): [
-        "Within {word_count} words, compare this image to the {style} style, highlighting elements that reflect or contrast with its key characteristics.",
-        "Compose a concise {word_count}-word analysis of how this artwork relates to the {style} movement, noting its adherence to or departure from typical {style} elements."
+        "Within {word_count} words, create a precise stable diffusion prompt capturing the essence of this image.",
+        "Compose a concise AI art prompt of {word_count} words to replicate this artwork's style and content."
     ],
     ("style_prompt", "formal", True, False): [
-        "Write a {length} critique of this image, exploring its relationship to the {style} movement. Discuss composition, technique, and thematic elements in this context.",
-        "Develop a {length} analysis of how this artwork incorporates or challenges the principles of {style}, considering its visual language and artistic approach."
+        "Write a {length} stable diffusion prompt that thoroughly describes this image's style, subject, and artistic techniques.",
+        "Develop a {length} detailed prompt for AI art generation, breaking down the key visual elements and artistic approach of this image."
     ],
     ("style_prompt", "informal", False, False): [
         "Imagine this image is in an exhibition of {style} art. Describe what makes it fit in or stand out from other {style} pieces.",
@@ -291,7 +291,10 @@ def stream_chat(input_image: Image.Image, caption_type: str, caption_tone: str,
 	
 	# Add style prompt details if applicable
 	if caption_type == "style_prompt":
-		prompt_str += f" Include details about using a {lens_type} lens, {film_stock} film stock, {composition} composition, and {lighting} lighting."
+		prompt_str += (f" The prompt should specifically include details about using a {lens_type} lens, "
+		               f"{film_stock} film stock, {composition} composition, and {lighting} lighting. "
+		               f"Format the output as a comma-separated list of descriptors and modifiers, "
+		               f"suitable for direct input into a Stable Diffusion interface.")
 
 	print(f"Prompt: {prompt_str}")
 
@@ -345,6 +348,10 @@ def stream_chat(input_image: Image.Image, caption_type: str, caption_tone: str,
 
 	caption = tokenizer.batch_decode(generate_ids, skip_special_tokens=False, clean_up_tokenization_spaces=False)[0]
 
+	# For style_prompt, format the output for easy copying into image generation platforms
+	if caption_type == "style_prompt":
+		caption = "Stable Diffusion Prompt: " + caption.replace("\n", ", ")
+
 	return caption.strip()
 
 css = """
@@ -478,7 +485,17 @@ with gr.Blocks(theme="Hev832/Applio", css=css) as demo:
                 run_button = gr.Button("Make My Caption!")
             
             with gr.Column():
-                output_caption = gr.Textbox(label="Your Amazing Caption Appears Here", lines=10)
+                output_caption = gr.Textbox(label="Your Image Generation Prompt (Copy this for Stable Diffusion)", lines=10)
+
+        gr.Markdown("""
+        ## How to Use Your Generated Prompt:
+        1. For "Style Prompt" captions, the output is formatted for direct use in Stable Diffusion.
+        2. Simply copy the entire text from the output box.
+        3. Paste it into your preferred Stable Diffusion interface or any other AI image generation platform.
+        4. Adjust or add to the prompt as desired to fine-tune your image generation.
+
+        Remember, you can always regenerate or modify the prompt to get different results!
+        """)
 
         run_button.click(fn=stream_chat, inputs=[input_image, caption_type, caption_tone, caption_length, lens_type, film_stock, composition, lighting], outputs=[output_caption])