Switch to Gemini + update requirements

app.py

@@ -3,7 +3,7 @@ from datetime import datetime
 from PIL import Image
 import gradio as gr
 from rembg import remove
-from openai import OpenAI
+import google.generativeai as genai  # Gemini SDK
 
 # ---------------------------
 # CONFIG
@@ -17,8 +17,9 @@ LIFETIME = 24 * 60 * 60  # 24 hours
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 os.makedirs(RESULTS_DIR, exist_ok=True)
 
-# OpenAI API key (must be added in Hugging Face Secrets as OPENAI_API_KEY)
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+# Gemini API key (must be added in Hugging Face Secrets as GEMINI_API_KEY)
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+model = genai.GenerativeModel("gemini-1.5-flash")
 
 # ---------------------------
 # HELPERS
@@ -65,7 +66,7 @@ def process_image(input_img, bg_choice):
     return [result_path]
 
 # ---------------------------
-# CAPTION GENERATOR
+# CAPTION GENERATOR (Gemini)
 # ---------------------------
 def generate_caption(prompt="Promote my product"):
     try:
@@ -75,15 +76,8 @@ def generate_caption(prompt="Promote my product"):
             "Format output clearly as:\nInstagram:\nFacebook:\nTikTok:\n"
         )
 
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": full_prompt}],
-            temperature=0.7,
-            max_tokens=300
-        )
-
-        captions = response.choices[0].message.content.strip()
-        return captions
+        response = model.generate_content(full_prompt)
+        return response.text.strip()
 
     except Exception as e:
         return f"❌ Error generating captions: {str(e)}"

requirements.txt

@@ -5,4 +5,5 @@ Pillow
 rembg
 onnxruntime
 requests
-openai
+google-generativeai
+