Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,11 +13,9 @@ client = genai.Client(api_key=API_KEY)
|
|
13 |
def edit_image_with_gemini(image, text_input):
|
14 |
"""
|
15 |
Edits an image using Gemini 2.0 Flash Experimental API based on a given text prompt.
|
16 |
-
|
17 |
Parameters:
|
18 |
image_path (str): Path to the input image.
|
19 |
text_prompt (str): Text prompt describing the edit.
|
20 |
-
|
21 |
Returns:
|
22 |
Image: The modified image.
|
23 |
"""
|
@@ -42,11 +40,6 @@ def edit_image_with_gemini(image, text_input):
|
|
42 |
|
43 |
return "No image was generated. Try modifying your prompt.", None
|
44 |
|
45 |
-
def generate_thumbnail_prompt():
|
46 |
-
"""
|
47 |
-
Returns a predefined optimized prompt for creating a YouTube thumbnail.
|
48 |
-
"""
|
49 |
-
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."
|
50 |
|
51 |
|
52 |
# Gradio App
|
@@ -58,20 +51,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
58 |
image_input = gr.Image(type="pil", label="Upload Image")
|
59 |
text_input = gr.Textbox(placeholder="Describe your edit...", label="Edit Prompt")
|
60 |
|
61 |
-
thumbnail_master_btn = gr.Button("🎨 Thumbnail Master")
|
62 |
output_text = gr.Textbox(label="Status", interactive=False)
|
63 |
output_image = gr.Image(label="Edited Image")
|
64 |
|
65 |
with gr.Row():
|
66 |
submit_btn = gr.Button("Generate Edit")
|
67 |
clear_btn = gr.Button("Clear")
|
68 |
-
|
69 |
text_input.submit(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
|
70 |
submit_btn.click(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
|
71 |
clear_btn.click(lambda: (None, None), None, [output_text, output_image])
|
72 |
-
thumbnail_master_btn.click(generate_thumbnail_prompt, [], [text_input])
|
73 |
-
|
74 |
|
75 |
# Launch the app
|
76 |
if __name__ == "__main__":
|
77 |
-
demo.launch(debug=True)
|
|
|
13 |
def edit_image_with_gemini(image, text_input):
|
14 |
"""
|
15 |
Edits an image using Gemini 2.0 Flash Experimental API based on a given text prompt.
|
|
|
16 |
Parameters:
|
17 |
image_path (str): Path to the input image.
|
18 |
text_prompt (str): Text prompt describing the edit.
|
|
|
19 |
Returns:
|
20 |
Image: The modified image.
|
21 |
"""
|
|
|
40 |
|
41 |
return "No image was generated. Try modifying your prompt.", None
|
42 |
|
|
|
|
|
|
|
|
|
|
|
43 |
|
44 |
|
45 |
# Gradio App
|
|
|
51 |
image_input = gr.Image(type="pil", label="Upload Image")
|
52 |
text_input = gr.Textbox(placeholder="Describe your edit...", label="Edit Prompt")
|
53 |
|
|
|
54 |
output_text = gr.Textbox(label="Status", interactive=False)
|
55 |
output_image = gr.Image(label="Edited Image")
|
56 |
|
57 |
with gr.Row():
|
58 |
submit_btn = gr.Button("Generate Edit")
|
59 |
clear_btn = gr.Button("Clear")
|
|
|
60 |
text_input.submit(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
|
61 |
submit_btn.click(edit_image_with_gemini, [image_input, text_input], [output_text, output_image])
|
62 |
clear_btn.click(lambda: (None, None), None, [output_text, output_image])
|
|
|
|
|
63 |
|
64 |
# Launch the app
|
65 |
if __name__ == "__main__":
|
66 |
+
demo.launch(debug=True)
|