Update app.py

app.py

@@ -16,7 +16,8 @@ from pydantic import BaseModel
 from typing import Optional
 from PIL import Image
 from io import BytesIO
-from google import genai
+import google.generativeai as genai
+import tempfile
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -44,6 +45,9 @@ app.add_middleware(
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "AIzaSyDL5Rilo7ptJpUOZdY6wy8PJYUcVcnDADs")
 GEMINI_MODEL = "gemini-2.5-flash-image-preview"
 
+# Configure Gemini API
+genai.configure(api_key=GEMINI_API_KEY)
+
 # ===== RAZORPAY CONFIGURATION =====
 RAZORPAY_KEY_ID = os.getenv("RAZORPAY_KEY_ID")
 RAZORPAY_KEY_SECRET = os.getenv("RAZORPAY_KEY_SECRET")
@@ -100,24 +104,19 @@ def validate_image(image_content: bytes):
     except Exception as e:
         raise HTTPException(status_code=400, detail=f"Image validation error: {str(e)}")
 
-def generate_gemini_image(images: List[Image.Image], prompt: str, api_key: str):
+def generate_gemini_image(images: List[Image.Image], prompt: str):
     """Generate or edit image using Gemini API"""
     try:
-        client = genai.Client(api_key=api_key)
         contents = images + [prompt]
-        
-        response = client.models.generate_content(
-            model=GEMINI_MODEL,
-            contents=contents,
-        )
+        response = genai.GenerativeModel(GEMINI_MODEL).generate_content(contents)
 
         text_response = ""
         image_path = None
 
         for part in response.candidates[0].content.parts:
-            if part.text is not None:
+            if part.text:
                 text_response += part.text + "\n"
-            elif part.inline_data is not None:
+            elif hasattr(part, 'inline_data') and part.inline_data:
                 with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
                     temp_path = tmp.name
                     generated_image = Image.open(BytesIO(part.inline_data.data))
@@ -207,7 +206,7 @@ async def generate_image(images: List[bytes], prompt: str, user_id: Optional[str
             raise HTTPException(status_code=500, detail=f"User verification failed: {str(e)}")
 
     # Generate image using Gemini API
-    image_path, text_response = generate_gemini_image(pil_images, prompt, GEMINI_API_KEY)
+    image_path, text_response = generate_gemini_image(pil_images, prompt)
 
     if image_path:
         return image_path