hansoneze commited on
Commit
675c27f
·
1 Parent(s): 0813fab

Switch to Gemini + update requirements

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -178,6 +178,20 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
178
  )
179
  bg_upload = gr.Image(type="pil", label="Or Upload Your Background (Optional)")
180
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
181
  with gr.Row():
182
  logo_upload = gr.Image(type="pil", label="Upload Logo (Optional)")
183
  logo_transparency = gr.Slider(0, 100, value=70, label="Logo Transparency (%)")
@@ -190,6 +204,7 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
190
 
191
  btn = gr.Button("✨ Generate New Photo")
192
  output_imgs = gr.Gallery(label="Generated Image", elem_id="gallery", columns=1, rows=1)
 
193
  btn.click(
194
  fn=process_image,
195
  inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, brand_color],
 
178
  )
179
  bg_upload = gr.Image(type="pil", label="Or Upload Your Background (Optional)")
180
 
181
+ # 👉 Live preview for selected background
182
+ bg_preview = gr.Image(type="pil", label="Background Preview", interactive=False)
183
+
184
+ def load_bg(choice):
185
+ if not choice:
186
+ return None
187
+ path = os.path.join(BG_DIR, choice)
188
+ if os.path.exists(path):
189
+ from PIL import Image
190
+ return Image.open(path)
191
+ return None
192
+
193
+ bg_choices.change(fn=load_bg, inputs=bg_choices, outputs=bg_preview)
194
+
195
  with gr.Row():
196
  logo_upload = gr.Image(type="pil", label="Upload Logo (Optional)")
197
  logo_transparency = gr.Slider(0, 100, value=70, label="Logo Transparency (%)")
 
204
 
205
  btn = gr.Button("✨ Generate New Photo")
206
  output_imgs = gr.Gallery(label="Generated Image", elem_id="gallery", columns=1, rows=1)
207
+
208
  btn.click(
209
  fn=process_image,
210
  inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, brand_color],