Update app.py

app.py

@@ -42,6 +42,11 @@ def edit_image_with_gemini(image, text_input):
 
     return "No image was generated. Try modifying your prompt.", None
 
+def generate_thumbnail_prompt():
+    """
+    Returns a predefined optimized prompt for creating a YouTube thumbnail.
+    """
+    return "Generate a bold, eye-catching YouTube thumbnail with vibrant colors, large text, and a strong contrast. Make sure it stands out and is attention-grabbing."
 
 
 # Gradio App
@@ -53,16 +58,25 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         image_input = gr.Image(type="pil", label="Upload Image")
         text_input = gr.Textbox(placeholder="Describe your edit...", label="Edit Prompt")
 
+# Thumbnail Master button (appears below the text box)
+    thumbnail_master_btn = gr.Button("🎨 Thumbnail Master")
+
+    
     output_text = gr.Textbox(label="Status", interactive=False)
     output_image = gr.Image(label="Edited Image")
 
     with gr.Row():
         submit_btn = gr.Button("Generate Edit")
         clear_btn = gr.Button("Clear")
+        
     text_input.submit(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
     submit_btn.click(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
     clear_btn.click(lambda: (None, None), None, [output_text, output_image])
 
+        # Clicking "Thumbnail Master" fills in the optimized prompt
+    thumbnail_master_btn.click(generate_thumbnail_prompt, [], [text_input])
+
+
 # Launch the app
 if __name__ == "__main__":
     demo.launch(debug=True)