Switch to Gemini + update requirements

.env

@@ -0,0 +1 @@
+GOOGLE_API_KEY=AIzaSyAhN_jl2FRSq8qBPNws87LJWEGY3UtS_6o

app.py

@@ -12,6 +12,16 @@ import google.generativeai as genai  # Gemini SDK
 import gradio as gr
 import uvicorn
 
+from dotenv import load_dotenv
+
+# Load environment variables from .env
+load_dotenv()
+
+# Confirm API key is loaded
+print("Gemini API Key Loaded:", os.getenv("GOOGLE_API_KEY") is not None)
+
+
+
 # ---------------------------
 # CONFIG
 # ---------------------------
@@ -118,6 +128,35 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
     gr.Markdown("Upload your product photo, replace background, and auto-generate marketing captions + hashtags!")
 
     # Image Editor Tab
+
+        # Image Editor Tab
+    with gr.Tab("📸 Image Editor"):
+        with gr.Row():
+            input_img = gr.Image(type="pil", label="Upload Photo")
+
+            with gr.Column():
+                bg_choices = gr.Dropdown(
+                    choices=os.listdir(BG_DIR),
+                    value=os.listdir(BG_DIR)[0] if os.listdir(BG_DIR) else None,
+                    label="Choose Background"
+                )
+                bg_preview = gr.Image(
+                    label="Background Preview",
+                    type="filepath"
+                )
+
+        # Update preview whenever background changes
+        def update_preview(bg_choice):
+            return os.path.join(BG_DIR, bg_choice) if bg_choice else None
+
+        bg_choices.change(fn=update_preview, inputs=bg_choices, outputs=bg_preview)
+
+        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)
+
+    
+    '''
     with gr.Tab("📸 Image Editor"):
         with gr.Row():
             input_img = gr.Image(type="pil", label="Upload Photo")
@@ -137,7 +176,7 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
         caption_box = gr.Textbox(label="Suggested Posts (multi-platform)", lines=12)
         btn2.click(fn=generate_caption, inputs=[prompt], outputs=[caption_box])
 
-# ---------------------------
+    '''
 # START SERVER
 # ---------------------------
 if __name__ == "__main__":