Fix Gallery component for Gradio compatibility

app.py

@@ -3,7 +3,7 @@ from datetime import datetime
 from PIL import Image
 import gradio as gr
 from rembg import remove
-import openai
+from openai import OpenAI
 
 # ---------------------------
 # CONFIG
@@ -17,8 +17,8 @@ LIFETIME = 24 * 60 * 60  # 24 hours
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 os.makedirs(RESULTS_DIR, exist_ok=True)
 
-# Set OpenAI API key (must be added in Hugging Face Secrets as OPENAI_API_KEY)
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# OpenAI API key (must be added in Hugging Face Secrets as OPENAI_API_KEY)
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # ---------------------------
 # HELPERS
@@ -40,8 +40,10 @@ def replace_background(input_path, bg_choice):
     check_size(input_path)
     input_img = Image.open(input_path).convert("RGBA")
     fg = remove(input_img)
+
     bg_path = os.path.join(BG_DIR, bg_choice)
     bg = Image.open(bg_path).convert("RGBA").resize(fg.size)
+
     result = Image.alpha_composite(bg, fg)
     timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
     result_path = os.path.join(RESULTS_DIR, f"result_{timestamp}.png")
@@ -52,18 +54,15 @@ def replace_background(input_path, bg_choice):
 # ---------------------------
 # IMAGE PROCESSOR
 # ---------------------------
-def process_image(input_img, bg_choices):
+def process_image(input_img, bg_choice):
     if input_img is None:
         return []
+
     temp_path = os.path.join(UPLOAD_DIR, f"upload_{int(time.time())}.png")
     input_img.save(temp_path)
 
-    results = []
-    for bg_choice in bg_choices:  # Loop through all selected backgrounds
-        result_path = replace_background(temp_path, bg_choice)
-        results.append(result_path)
-
-    return results
+    result_path = replace_background(temp_path, bg_choice)
+    return [result_path]
 
 # ---------------------------
 # CAPTION GENERATOR
@@ -76,7 +75,7 @@ def generate_caption(prompt="Promote my product"):
             "Format output clearly as:\nInstagram:\nFacebook:\nTikTok:\n"
         )
 
-        response = openai.chat.completions.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[{"role": "user", "content": full_prompt}],
             temperature=0.7,
@@ -100,13 +99,13 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
     with gr.Tab("📸 Image Editor"):
         with gr.Row():
             input_img = gr.Image(type="pil", label="Upload Photo")
-            bg_choices = gr.CheckboxGroup(
+            bg_choices = gr.Dropdown(
                 choices=os.listdir(BG_DIR),
-                value=[os.listdir(BG_DIR)[0]],
-                label="Choose Background(s)"
+                value=os.listdir(BG_DIR)[0] if os.listdir(BG_DIR) else None,
+                label="Choose Background"
             )
-        btn = gr.Button("✨ Generate New Photos")
-        output_imgs = gr.Gallery(label="Generated Images", elem_id="gallery", columns=3, rows=2)
+        btn = gr.Button("✨ Generate New Photo")
+        output_imgs = gr.Gallery(label="Generated Image", elem_id="gallery", columns=1, rows=1)
         btn.click(fn=process_image, inputs=[input_img, bg_choices], outputs=output_imgs)
 
     # Caption Generator Tab