Switch to Gemini + update requirements

app.py

@@ -209,16 +209,22 @@ async def generate_captions_api(prompt: str = Form(...)):
 # ---------------------------
 # GRADIO UI
 # ---------------------------
-with gr.Blocks(css="footer {display:none !important}") as demo:
+# ---------------------------
+# GRADIO UI
+# ---------------------------
+with gr.Blocks(css="footer {display:none !important}; .gradio-container {max-width: 100% !important}") as demo:
     gr.Markdown("# ✨ SnapLift – AI Social Media Booster")
     gr.Markdown("Upload your product photo, replace background, and auto-generate marketing captions + hashtags!")
 
+    # ---------------------------
     # Image Editor Tab
+    # ---------------------------
     with gr.Tab("📸 Image Editor"):
         with gr.Row():
-            input_img = gr.Image(type="pil", label="Upload Main Photo")
+            # ---- Left Column: Inputs & Controls ----
+            with gr.Column(scale=1):
+                input_img = gr.Image(type="pil", label="Upload Main Photo")
 
-            with gr.Column():
                 bg_choices = gr.Dropdown(
                     choices=os.listdir(BG_DIR) or ["default.png"],
                     value=(os.listdir(BG_DIR)[0] if os.listdir(BG_DIR) else None),
@@ -240,34 +246,46 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
 
                 bg_choices.change(fn=load_bg, inputs=bg_choices, outputs=bg_preview)
 
-        with gr.Row():
-            logo_upload = gr.Image(type="pil", label="Upload Logo (Optional)")
-            logo_transparency = gr.Slider(0, 100, value=70, label="Logo Transparency (%)")
-            logo_position = gr.Dropdown(
-                ["Top-Left", "Top-Right", "Bottom-Left", "Bottom-Right", "Center"],
-                value="Bottom-Right",
-                label="Logo Position"
-            )
-        
-        blend_strength = gr.Slider(0, 1, value=0.5, step=0.1, label="Blending Strength (Realism)")
-        blur_background = gr.Checkbox(label="Blur Background", value=False)
-
-            #brand_color = gr.ColorPicker(label="Brand Colour")
-
-        btn = gr.Button("✨ Generate New Photo")
-        output_imgs = gr.Gallery(label="Generated Image", elem_id="gallery", columns=1, rows=1)
+                logo_upload = gr.Image(type="pil", label="Upload Logo (Optional)")
+                logo_transparency = gr.Slider(0, 100, value=70, label="Logo Transparency (%)")
+                logo_position = gr.Dropdown(
+                    ["Top-Left", "Top-Right", "Bottom-Left", "Bottom-Right", "Center"],
+                    value="Bottom-Right",
+                    label="Logo Position"
+                )
+
+                blend_strength = gr.Slider(0, 1, value=0.5, step=0.1, label="Blending Strength (Realism)")
+                blur_background = gr.Checkbox(label="Blur Background", value=False)
 
+                btn = gr.Button("✨ Generate New Photo")
+
+            # ---- Right Column: Output ----
+            with gr.Column(scale=1):
+                output_imgs = gr.Gallery(
+                    label="Generated Image",
+                    elem_id="gallery",
+                    columns=1,
+                    rows=1
+                )
+
+        # Action binding
         btn.click(
             fn=process_image,
             inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength],
             outputs=output_imgs
         )
 
+    # ---------------------------
     # Caption Generator Tab
+    # ---------------------------
     with gr.Tab("✍️ Caption Generator"):
-        prompt = gr.Textbox(label="Enter product/promotion text", value="Promote my skincare product")
-        btn2 = gr.Button("💡 Suggest Captions + Hashtags")
-        caption_box = gr.Textbox(label="Suggested Posts (multi-platform)", lines=12)
+        with gr.Row():
+            with gr.Column(scale=1):
+                prompt = gr.Textbox(label="Enter product/promotion text", value="Promote my skincare product")
+                btn2 = gr.Button("💡 Suggest Captions + Hashtags")
+            with gr.Column(scale=1):
+                caption_box = gr.Textbox(label="Suggested Posts (multi-platform)", lines=12)
+
         btn2.click(fn=generate_caption, inputs=[prompt], outputs=[caption_box])
 
 # ---------------------------