diegocp01 commited on
Commit
bcc6b54
·
verified ·
1 Parent(s): 6113a1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -12
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)