Switch to Gemini + update requirements

app.py

@@ -221,7 +221,7 @@ with gr.Blocks(css="""
     # ---------------------------
     # Theme Toggle
     # ---------------------------
-    theme_state = gr.State("light")  # default theme
+    theme_state = gr.State("light")
 
     def toggle_theme(current):
         return "dark" if current == "light" else "light"
@@ -229,7 +229,7 @@ with gr.Blocks(css="""
     with gr.Row():
         with gr.Column():
             gr.Markdown("<h1 style='text-align:center; font-size:2.2em;'>✨ SnapLift – AI Social Media Booster</h1>")
-            gr.Markdown("<p style='text-align:center; font-size:1.1em; color:#555;'>Upload your product photo, replace background, and auto-generate <b>marketing captions</b> + <b>hashtags</b>!</p>")
+            gr.Markdown("<p style='text-align:center; font-size:1.1em; color:#555;'>Upload or capture your product photo, replace background, and auto-generate <b>marketing captions</b> + <b>hashtags</b>!</p>")
         with gr.Column(scale=0.2):
             theme_btn = gr.Button("🌙 Toggle Theme")
 
@@ -240,10 +240,14 @@ with gr.Blocks(css="""
     # ---------------------------
     with gr.Tab("📸 Image Editor"):
         with gr.Row(equal_height=True):
-            # ---- Left Column: Inputs & Controls ----
             with gr.Column(scale=1):
                 with gr.Group(elem_classes="box"):
-                    input_img = gr.Image(type="pil", label="📤 Upload Main Photo")
+                    input_img = gr.Image(
+                        type="pil",
+                        label="📤 Upload or Capture Main Photo",
+                        sources=["upload", "camera"],
+                        tool="editor"
+                    )
 
                 with gr.Accordion("🎨 Background Options", open=True):
                     bg_choices = gr.Dropdown(
@@ -251,7 +255,12 @@ with gr.Blocks(css="""
                         value=(os.listdir(BG_DIR)[0] if os.listdir(BG_DIR) else None),
                         label="Choose Background"
                     )
-                    bg_upload = gr.Image(type="pil", label="Or Upload Your Background")
+                    bg_upload = gr.Image(
+                        type="pil",
+                        label="📤 Upload or Capture Background",
+                        sources=["upload", "camera"],
+                        tool="editor"
+                    )
                     bg_preview = gr.Image(type="pil", label="Background Preview", interactive=False)
 
                     def load_bg(choice):
@@ -277,7 +286,26 @@ with gr.Blocks(css="""
                     blend_strength = gr.Slider(0, 1, value=0.5, step=0.1, label="Blending Strength")
                     blur_background = gr.Checkbox(label="Blur Background", value=False)
 
-                btn = gr.Button("🚀 Generate New Photo", elem_classes="box")
+                with gr.Accordion("💾 Export Options", open=True):
+                    export_format = gr.Dropdown(
+                        ["PNG", "JPG", "PDF"],
+                        value="PNG",
+                        label="Export Format"
+                    )
+                    export_size = gr.Dropdown(
+                        [
+                            "Original",
+                            "Instagram (1080x1080)",
+                            "Facebook (1200x628)",
+                            "TikTok (1080x1920)",
+                            "LinkedIn (1200x1200)",
+                            "Twitter (1600x900)"
+                        ],
+                        value="Original",
+                        label="Social Media Size"
+                    )
+
+                btn = gr.Button("🚀 Generate & Export", elem_classes="box")
 
             # ---- Right Column: Output ----
             with gr.Column(scale=1):
@@ -288,12 +316,49 @@ with gr.Blocks(css="""
                     elem_id="output-img",
                     interactive=False
                 )
-                download_btn = gr.File(label="⬇️ Download Final Image")
-
-        # Bind button to function
+                download_btn = gr.File(label="⬇️ Download HD Export")
+
+        # ---------------------------
+        # Wrapper to handle export formats + resizing
+        # ---------------------------
+        def process_and_export(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength, export_format, export_size):
+            result = process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength)
+            if isinstance(result, list):
+                result = result[0]
+
+            img = Image.open(result)
+
+            # Resize if needed
+            size_map = {
+                "Instagram (1080x1080)": (1080, 1080),
+                "Facebook (1200x628)": (1200, 628),
+                "TikTok (1080x1920)": (1080, 1920),
+                "LinkedIn (1200x1200)": (1200, 1200),
+                "Twitter (1600x900)": (1600, 900)
+            }
+            if export_size in size_map:
+                img = img.resize(size_map[export_size], Image.LANCZOS)
+
+            # Prepare export path
+            timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+            if export_format == "PNG":
+                out_path = os.path.join(RESULTS_DIR, f"export_{timestamp}.png")
+                img.save(out_path, "PNG", quality=95)
+            elif export_format == "JPG":
+                out_path = os.path.join(RESULTS_DIR, f"export_{timestamp}.jpg")
+                img.convert("RGB").save(out_path, "JPEG", quality=95)
+            elif export_format == "PDF":
+                out_path = os.path.join(RESULTS_DIR, f"export_{timestamp}.pdf")
+                img.convert("RGB").save(out_path, "PDF", resolution=300.0)
+            else:
+                out_path = result
+
+            return result, out_path
+
+        # Bind button
         btn.click(
-            fn=process_image,
-            inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength],
+            fn=process_and_export,
+            inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength, export_format, export_size],
             outputs=[output_img, download_btn]
         )