Upload 13 files

.gitattributes

@@ -1,35 +1,38 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+data/1.jpg filter=lfs diff=lfs merge=lfs -text
+data/1777043.jpg filter=lfs diff=lfs merge=lfs -text
+data/2807615.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,12 +1,12 @@
----
-title: My3
-emoji: 🦀
-colorFrom: blue
-colorTo: pink
-sdk: gradio
-sdk_version: 5.20.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Gemini-2.5-Flash-Nano-Banana
+emoji: 📚
+colorFrom: indigo
+colorTo: blue
+sdk: gradio
+sdk_version: 5.34.0
+app_file: app.py
+pinned: true
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,381 +1,376 @@
-import logging
-from fastapi import FastAPI, UploadFile, File, HTTPException, Form, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, JSONResponse
-import requests
-import base64
-import os
-import time
-import jwt
-from pathlib import Path
-from typing import List
-import io
-import razorpay
-from razorpay.errors import SignatureVerificationError
-from supabase import create_client, Client
-from pydantic import BaseModel
-from typing import Optional
-
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Initialize FastAPI app
-app = FastAPI(title="Kling AI Multi-Image Generator API with Razorpay")
-
-# Enable CORS for the frontend - Updated for Lovable domains
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=[
-        "https://hivili.web.app", 
-        "http://localhost:3000",
-        "https://*.lovable.dev",
-        "https://*.sandbox.lovable.dev",
-        # Add pattern matching for Lovable subdomains
-    ],
-    allow_origin_regex=r"https://.*\.lovable\.dev|https://.*\.sandbox\.lovable\.dev",
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# ===== API CONFIGURATION =====
-ACCESS_KEY_ID = os.getenv("ACCESS_KEY_ID", "AFyHfnQATghFdCMyAG3gRPbNY4TNKFGB")
-ACCESS_KEY_SECRET = os.getenv("ACCESS_KEY_SECRET", "TTepeLyBterLNM3brYPGmdndBnnyKJBA")
-API_BASE_URL = "https://api-singapore.klingai.com"
-CREATE_TASK_ENDPOINT = f"{API_BASE_URL}/v1/images/multi-image2image"
-
-# ===== RAZORPAY CONFIGURATION =====
-RAZORPAY_KEY_ID = os.getenv("RAZORPAY_KEY_ID")
-RAZORPAY_KEY_SECRET = os.getenv("RAZORPAY_KEY_SECRET")
-razorpay_client = razorpay.Client(auth=(RAZORPAY_KEY_ID, RAZORPAY_KEY_SECRET))
-
-# ===== SUPABASE CONFIGURATION =====
-SUPABASE_URL = os.getenv("SUPABASE_URL")
-SUPABASE_KEY = os.getenv("SUPABASE_KEY")
-supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY) if SUPABASE_URL and SUPABASE_KEY else None
-
-# Pydantic models for JSON input validation
-class CreateOrderRequest(BaseModel):
-    amount: int
-
-class VerifyPaymentRequest(BaseModel):
-    razorpay_order_id: str
-    razorpay_payment_id: str
-    razorpay_signature: str
-    user_id: Optional[str] = None
-
-# ===== AUTHENTICATION =====
-def generate_jwt_token():
-    """Generate JWT token for API authentication"""
-    payload = {
-        "iss": ACCESS_KEY_ID,
-        "exp": int(time.time()) + 1800,  # 30 minutes expiration
-        "nbf": int(time.time()) - 5      # Not before 5 seconds ago
-    }
-    return jwt.encode(payload, ACCESS_KEY_SECRET, algorithm="HS256")
-
-# ===== IMAGE PROCESSING =====
-def prepare_image_base64(image_content: bytes):
-    """Convert image bytes to base64 without prefix"""
-    try:
-        return base64.b64encode(image_content).decode('utf-8')
-    except Exception as e:
-        logger.error(f"Image processing failed: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Image processing failed: {str(e)}")
-
-def validate_image(image_content: bytes):
-    """Validate image meets API requirements"""
-    try:
-        size_mb = len(image_content) / (1024 * 1024)
-        if size_mb > 10:
-            raise HTTPException(status_code=400, detail="Image too large (max 10MB)")
-        return True, ""
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Image validation error: {str(e)}")
-
-# ===== API FUNCTIONS =====
-def create_multi_image_task(subject_images: List[bytes], prompt: str):
-    """Create multi-image generation task"""
-    headers = {
-        "Authorization": f"Bearer {generate_jwt_token()}",
-        "Content-Type": "application/json"
-    }
-    
-    subject_image_list = []
-    for img_content in subject_images:
-        if img_content:
-            base64_img = prepare_image_base64(img_content)
-            if base64_img:
-                subject_image_list.append({"subject_image": base64_img})
-    
-    if len(subject_image_list) < 2:
-        raise HTTPException(status_code=400, detail="At least 2 subject images required")
-    
-    payload = {
-        "model_name": "kling-v2",
-        "prompt": prompt,
-        "subject_image_list": subject_image_list,
-        "n": 1,
-        "aspect_ratio": "1:1"
-    }
-    
-    try:
-        response = requests.post(CREATE_TASK_ENDPOINT, json=payload, headers=headers)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        logger.error(f"API request failed: {str(e)}")
-        if hasattr(e, 'response') and e.response:
-            logger.error(f"API response: {e.response.text}")
-        raise HTTPException(status_code=500, detail=f"API Error: {str(e)}")
-
-def check_task_status(task_id: str):
-    """Check task completion status"""
-    headers = {"Authorization": f"Bearer {generate_jwt_token()}"}
-    status_url = f"{API_BASE_URL}/v1/images/multi-image2image/{task_id}"
-    
-    try:
-        response = requests.get(status_url, headers=headers)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        raise HTTPException(status_code=500, detail=f"Status check failed: {str(e)}")
-
-# ===== RAZORPAY FUNCTIONS =====
-def create_razorpay_order(amount: int):
-    """Create a Razorpay order"""
-    try:
-        if amount <= 0:
-            raise ValueError("Amount must be a positive integer")
-        order_data = {
-            "amount": amount * 100,  # Convert INR to paise
-            "currency": "INR",
-            "payment_capture": 1  # Auto-capture payment
-        }
-        order = razorpay_client.order.create(data=order_data)
-        logger.info(f"Razorpay order created successfully: {order['id']}")
-        return order
-    except Exception as e:
-        logger.error(f"Failed to create Razorpay order: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Failed to create order: {str(e)}")
-
-def verify_payment_signature(order_id: str, payment_id: str, signature: str):
-    """Verify Razorpay payment signature"""
-    try:
-        params_dict = {
-            "razorpay_order_id": order_id,
-            "razorpay_payment_id": payment_id,
-            "razorpay_signature": signature
-        }
-        razorpay_client.utility.verify_payment_signature(params_dict)
-        logger.info(f"Payment signature verified successfully for order: {order_id}")
-        return True
-    except SignatureVerificationError as e:
-        logger.error(f"Payment signature verification failed: {str(e)}")
-        return False
-    except Exception as e:
-        logger.error(f"Error verifying payment signature: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Verification error: {str(e)}")
-
-# ===== MAIN PROCESSING =====
-async def generate_image(subject_images: List[bytes], prompt: str):
-    """Handle complete image generation workflow"""
-    for img_content in subject_images:
-        if img_content:
-            validate_image(img_content)
-    
-    task_response = create_multi_image_task(subject_images, prompt)
-    if task_response.get("code") != 0:
-        raise HTTPException(status_code=500, detail=f"API error: {task_response.get('message', 'Unknown error')}")
-    
-    task_id = task_response["data"]["task_id"]
-    logger.info(f"Task created: {task_id}")
-    
-    for _ in range(60):
-        task_data = check_task_status(task_id)
-        status = task_data["data"]["task_status"]
-        
-        if status == "succeed":
-            image_url = task_data["data"]["task_result"]["images"][0]["url"]
-            try:
-                response = requests.get(image_url)
-                response.raise_for_status()
-                output_dir = Path("/tmp")
-                output_dir.mkdir(exist_ok=True)
-                output_path = output_dir / f"kling_output_{task_id}.png"
-                with open(output_path, "wb") as f:
-                    f.write(response.content)
-                return output_path
-            except Exception as e:
-                raise HTTPException(status_code=500, detail=f"Failed to download result: {str(e)}")
-                
-        elif status in ("failed", "canceled"):
-            error_msg = task_data["data"].get("task_status_msg", "Unknown error")
-            raise HTTPException(status_code=500, detail=f"Task failed: {error_msg}")
-            
-        time.sleep(10)
-    
-    raise HTTPException(status_code=500, detail="Task timed out after 10 minutes")
-
-# ===== API ENDPOINTS =====
-@app.post("/generate")
-async def generate_image_endpoint(
-    prompt: str = Form(...),
-    images: List[UploadFile] = File(...)
-):
-    """Endpoint to generate an image from multiple input images and a prompt"""
-    try:
-        if len(images) < 2:
-            raise HTTPException(status_code=400, detail="At least 2 images are required")
-        if len(images) > 4:
-            raise HTTPException(status_code=400, detail="Maximum 4 images allowed")
-
-        image_contents = [await image.read() for image in images]
-        output_path = await generate_image(image_contents, prompt)
-
-        return FileResponse(
-            path=output_path,
-            media_type="image/png",
-            filename=f"kling_output_{Path(output_path).stem}.png"
-        )
-    except Exception as e:
-        logger.error(f"Error in /generate: {str(e)}")
-        raise HTTPException(status_code=500, detail=str(e))
-
-@app.post("/create-razorpay-order")
-async def create_order_endpoint(
-    request: Request,
-    amount: Optional[int] = Form(None),
-    body: Optional[CreateOrderRequest] = None
-):
-    """Create a Razorpay order (supports form-data and JSON)"""
-    logger.info("Received create order request")
-    
-    if not RAZORPAY_KEY_ID or not RAZORPAY_KEY_SECRET:
-        logger.error("Razorpay configuration missing")
-        raise HTTPException(status_code=500, detail="Razorpay configuration missing")
-    
-    # Handle JSON body if provided
-    try:
-        if body and body.amount:
-            amount = body.amount
-        elif not amount:
-            # Try to parse JSON from request body
-            try:
-                json_body = await request.json()
-                amount = json_body.get('amount')
-            except:
-                pass
-                
-        if not amount or amount <= 0:
-            raise HTTPException(status_code=422, detail="Missing or invalid 'amount' parameter")
-
-        logger.info(f"Creating order with amount: {amount}")
-        order = create_razorpay_order(amount)
-        
-        response_data = {
-            "id": order["id"],
-            "amount": order["amount"],
-            "currency": order["currency"],
-            "key_id": RAZORPAY_KEY_ID
-        }
-        
-        logger.info(f"Order created successfully: {order['id']}")
-        return JSONResponse(content=response_data)
-        
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Error creating order: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Failed to create order: {str(e)}")
-
-@app.post("/verify-razorpay-payment")
-async def verify_payment_endpoint(
-    request: Request,
-    razorpay_order_id: Optional[str] = Form(None),
-    razorpay_payment_id: Optional[str] = Form(None),
-    razorpay_signature: Optional[str] = Form(None),
-    user_id: Optional[str] = Form(None),
-    body: Optional[VerifyPaymentRequest] = None
-):
-    """Verify Razorpay payment signature (supports form-data and JSON)"""
-    logger.info("Received payment verification request")
-    
-    try:
-        # Handle JSON body if provided
-        if body:
-            razorpay_order_id = razorpay_order_id or body.razorpay_order_id
-            razorpay_payment_id = razorpay_payment_id or body.razorpay_payment_id
-            razorpay_signature = razorpay_signature or body.razorpay_signature
-            user_id = user_id or body.user_id
-        else:
-            # Try to parse JSON from request body
-            try:
-                json_body = await request.json()
-                razorpay_order_id = razorpay_order_id or json_body.get('razorpay_order_id')
-                razorpay_payment_id = razorpay_payment_id or json_body.get('razorpay_payment_id')
-                razorpay_signature = razorpay_signature or json_body.get('razorpay_signature')
-                user_id = user_id or json_body.get('user_id')
-            except:
-                pass
-
-        # Validate required fields
-        if not all([razorpay_order_id, razorpay_payment_id, razorpay_signature]):
-            missing_fields = []
-            if not razorpay_order_id: missing_fields.append("razorpay_order_id")
-            if not razorpay_payment_id: missing_fields.append("razorpay_payment_id")
-            if not razorpay_signature: missing_fields.append("razorpay_signature")
-            
-            logger.error(f"Missing required fields: {missing_fields}")
-            raise HTTPException(
-                status_code=422, 
-                detail=f"Missing required fields: {', '.join(missing_fields)}"
-            )
-
-        logger.info(f"Verifying payment for order_id: {razorpay_order_id}")
-        is_valid = verify_payment_signature(razorpay_order_id, razorpay_payment_id, razorpay_signature)
-        
-        if is_valid:
-            if user_id and supabase:
-                logger.info(f"Updating Supabase for user_id: {user_id}")
-                try:
-                    supabase.table("users").update({"is_premium": True}).eq("user_id", user_id).execute()
-                    logger.info(f"Successfully updated premium status for user: {user_id}")
-                except Exception as e:
-                    logger.error(f"Failed to update Supabase: {str(e)}")
-                    # Don't fail the payment verification if Supabase update fails
-            
-            return JSONResponse(content={"success": True, "message": "Payment verified successfully"})
-        else:
-            logger.warning(f"Payment verification failed for order: {razorpay_order_id}")
-            return JSONResponse(content={"success": False, "message": "Payment verification failed"}, status_code=400)
-            
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Error verifying payment: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Verification error: {str(e)}")
-
-@app.get("/")
-async def index():
-    return {
-        "status": "Kling AI Multi-Image Generator API with Razorpay is running",
-        "endpoints": {
-            "generate": "POST /generate",
-            "create_order": "POST /create-razorpay-order",
-            "verify_payment": "POST /verify-razorpay-payment"
-        }
-    }
-
-@app.get("/health")
-async def health_check():
-    return {
-        "status": "healthy",
-        "razorpay_configured": bool(RAZORPAY_KEY_ID and RAZORPAY_KEY_SECRET),
-        "supabase_configured": bool(SUPABASE_URL and SUPABASE_KEY)
-    }
-
-if __name__ == "__main__":
-    import uvicorn
+import logging
+from fastapi import FastAPI, UploadFile, File, HTTPException, Form, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse, JSONResponse
+import base64
+import os
+import time
+import jwt
+from pathlib import Path
+from typing import List
+import io
+import razorpay
+from razorpay.errors import SignatureVerificationError
+from supabase import create_client, Client
+from pydantic import BaseModel
+from typing import Optional
+from PIL import Image
+from io import BytesIO
+import google.generativeai as genai
+import tempfile
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize FastAPI app
+app = FastAPI(title="Gemini Image Editing API with Razorpay")
+
+# Enable CORS for frontend
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=[
+        "https://hivili.web.app",
+        "http://localhost:3000",
+        "https://*.lovable.dev",
+        "https://*.sandbox.lovable.dev",
+    ],
+    allow_origin_regex=r"https://.*\.lovable\.dev|https://.*\.sandbox\.lovable\.dev",
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# ===== API CONFIGURATION =====
+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")
+razorpay_client = razorpay.Client(auth=(RAZORPAY_KEY_ID, RAZORPAY_KEY_SECRET)) if RAZORPAY_KEY_ID and RAZORPAY_KEY_SECRET else None
+
+# ===== SUPABASE CONFIGURATION =====
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY) if SUPABASE_URL and SUPABASE_KEY else None
+
+# Pydantic models for JSON input validation
+class CreateOrderRequest(BaseModel):
+    amount: int
+
+class VerifyPaymentRequest(BaseModel):
+    razorpay_order_id: str
+    razorpay_payment_id: str
+    razorpay_signature: str
+    user_id: Optional[str] = None
+
+class GenerateImageRequest(BaseModel):
+    prompt: str
+    user_id: Optional[str] = None
+
+# ===== AUTHENTICATION =====
+def generate_jwt_token():
+    """Generate JWT token for API authentication"""
+    payload = {
+        "iss": "gemini-image-editor",
+        "exp": int(time.time()) + 1800,  # 30 minutes expiration
+        "nbf": int(time.time()) - 5      # Not before 5 seconds ago
+    }
+    return jwt.encode(payload, GEMINI_API_KEY, algorithm="HS256")
+
+# ===== IMAGE PROCESSING =====
+def prepare_image_base64(image_content: bytes):
+    """Convert image bytes to base64 without prefix"""
+    try:
+        return base64.b64encode(image_content).decode('utf-8')
+    except Exception as e:
+        logger.error(f"Image processing failed: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Image processing failed: {str(e)}")
+
+def validate_image(image_content: bytes):
+    """Validate image meets API requirements"""
+    try:
+        size_mb = len(image_content) / (1024 * 1024)
+        if size_mb > 10:
+            raise HTTPException(status_code=400, detail="Image too large (max 10MB)")
+        img = Image.open(BytesIO(image_content))
+        if img.format.lower() not in ['png', 'jpg', 'jpeg', 'webp']:
+            raise HTTPException(status_code=400, detail="Unsupported image format. Use PNG, JPG, JPEG, or WEBP")
+        return True, ""
+    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):
+    """Generate or edit image using Gemini API"""
+    try:
+        contents = images + [prompt]
+        response = genai.GenerativeModel(GEMINI_MODEL).generate_content(contents)
+
+        text_response = ""
+        image_path = None
+
+        for part in response.candidates[0].content.parts:
+            if part.text:
+                text_response += part.text + "\n"
+            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))
+                    generated_image.save(temp_path)
+                    image_path = temp_path
+                    logger.info(f"Generated image saved to: {temp_path} with prompt: {prompt}")
+
+        if image_path:
+            return image_path, ""
+        else:
+            return None, text_response.strip()
+    except Exception as e:
+        logger.error(f"Gemini API error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Gemini API error: {str(e)}")
+
+# ===== RAZORPAY FUNCTIONS =====
+def create_razorpay_order(amount: int):
+    """Create a Razorpay order"""
+    try:
+        if not razorpay_client:
+            raise HTTPException(status_code=500, detail="Razorpay configuration missing")
+        if amount <= 0:
+            raise ValueError("Amount must be a positive integer")
+        order_data = {
+            "amount": amount * 100,  # Convert INR to paise
+            "currency": "INR",
+            "payment_capture": 1  # Auto-capture payment
+        }
+        order = razorpay_client.order.create(data=order_data)
+        logger.info(f"Razorpay order created successfully: {order['id']}")
+        return order
+    except Exception as e:
+        logger.error(f"Failed to create Razorpay order: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to create order: {str(e)}")
+
+def verify_payment_signature(order_id: str, payment_id: str, signature: str):
+    """Verify Razorpay payment signature"""
+    try:
+        if not razorpay_client:
+            raise HTTPException(status_code=500, detail="Razorpay configuration missing")
+        params_dict = {
+            "razorpay_order_id": order_id,
+            "razorpay_payment_id": payment_id,
+            "razorpay_signature": signature
+        }
+        razorpay_client.utility.verify_payment_signature(params_dict)
+        logger.info(f"Payment signature verified successfully for order: {order_id}")
+        return True
+    except SignatureVerificationError as e:
+        logger.error(f"Payment signature verification failed: {str(e)}")
+        return False
+    except Exception as e:
+        logger.error(f"Error verifying payment signature: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Verification error: {str(e)}")
+
+# ===== MAIN PROCESSING =====
+async def generate_image(images: List[bytes], prompt: str, user_id: Optional[str] = None):
+    """Handle complete image generation workflow"""
+    # Validate images
+    for img_content in images:
+        if img_content:
+            validate_image(img_content)
+    
+    # Convert bytes to PIL Images
+    pil_images = []
+    for img_content in images:
+        try:
+            img = Image.open(BytesIO(img_content))
+            if img.mode == "RGBA":
+                img = img.convert("RGBA")
+            pil_images.append(img)
+        except Exception as e:
+            logger.error(f"Image conversion failed: {str(e)}")
+            raise HTTPException(status_code=400, detail=f"Image conversion failed: {str(e)}")
+
+    if len(pil_images) < 1:
+        raise HTTPException(status_code=400, detail="At least one image required")
+
+    # Check user premium status if Supabase is configured
+    if user_id and supabase:
+        try:
+            user_data = supabase.table("users").select("is_premium").eq("user_id", user_id).execute()
+            if not user_data.data or not user_data.data[0].get("is_premium"):
+                raise HTTPException(status_code=403, detail="Premium subscription required")
+        except Exception as e:
+            logger.error(f"Supabase user check failed: {str(e)}")
+            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)
+
+    if image_path:
+        return image_path
+    else:
+        raise HTTPException(status_code=500, detail=f"Image generation failed: {text_response or 'Unknown error'}")
+
+# ===== API ENDPOINTS =====
+@app.post("/generate")
+async def generate_image_endpoint(
+    prompt: str = Form(...),
+    images: List[UploadFile] = File(...),
+    user_id: Optional[str] = Form(None)
+):
+    """Endpoint to generate or edit an image using Gemini API"""
+    try:
+        if len(images) < 1:
+            raise HTTPException(status_code=400, detail="At least one image required")
+        if len(images) > 4:
+            raise HTTPException(status_code=400, detail="Maximum 4 images allowed")
+
+        image_contents = [await image.read() for image in images]
+        output_path = await generate_image(image_contents, prompt, user_id)
+
+        return FileResponse(
+            path=output_path,
+            media_type="image/png",
+            filename=f"gemini_output_{Path(output_path).stem}.png"
+        )
+    except HTTPException as e:
+        raise
+    except Exception as e:
+        logger.error(f"Error in /generate: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/create-razorpay-order")
+async def create_order_endpoint(
+    request: Request,
+    amount: Optional[int] = Form(None),
+    body: Optional[CreateOrderRequest] = None
+):
+    """Create a Razorpay order (supports form-data and JSON)"""
+    logger.info("Received create order request")
+    
+    try:
+        if not razorpay_client:
+            raise HTTPException(status_code=500, detail="Razorpay configuration missing")
+        
+        # Handle JSON body if provided
+        if body and body.amount:
+            amount = body.amount
+        elif not amount:
+            try:
+                json_body = await request.json()
+                amount = json_body.get('amount')
+            except:
+                pass
+                
+        if not amount or amount <= 0:
+            raise HTTPException(status_code=422, detail="Missing or invalid 'amount' parameter")
+
+        logger.info(f"Creating order with amount: {amount}")
+        order = create_razorpay_order(amount)
+        
+        response_data = {
+            "id": order["id"],
+            "amount": order["amount"],
+            "currency": order["currency"],
+            "key_id": RAZORPAY_KEY_ID
+        }
+        
+        logger.info(f"Order created successfully: {order['id']}")
+        return JSONResponse(content=response_data)
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error creating order: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to create order: {str(e)}")
+
+@app.post("/verify-razorpay-payment")
+async def verify_payment_endpoint(
+    request: Request,
+    razorpay_order_id: Optional[str] = Form(None),
+    razorpay_payment_id: Optional[str] = Form(None),
+    razorpay_signature: Optional[str] = Form(None),
+    user_id: Optional[str] = Form(None),
+    body: Optional[VerifyPaymentRequest] = None
+):
+    """Verify Razorpay payment signature (supports form-data and JSON)"""
+    logger.info("Received payment verification request")
+    
+    try:
+        # Handle JSON body if provided
+        if body:
+            razorpay_order_id = razorpay_order_id or body.razorpay_order_id
+            razorpay_payment_id = razorpay_payment_id or body.razorpay_payment_id
+            razorpay_signature = razorpay_signature or body.razorpay_signature
+            user_id = user_id or body.user_id
+        else:
+            try:
+                json_body = await request.json()
+                razorpay_order_id = razorpay_order_id or json_body.get('razorpay_order_id')
+                razorpay_payment_id = razorpay_payment_id or json_body.get('razorpay_payment_id')
+                razorpay_signature = razorpay_signature or json_body.get('razorpay_signature')
+                user_id = user_id or json_body.get('user_id')
+            except:
+                pass
+
+        # Validate required fields
+        if not all([razorpay_order_id, razorpay_payment_id, razorpay_signature]):
+            missing_fields = []
+            if not razorpay_order_id: missing_fields.append("razorpay_order_id")
+            if not razorpay_payment_id: missing_fields.append("razorpay_payment_id")
+            if not razorpay_signature: missing_fields.append("razorpay_signature")
+            
+            logger.error(f"Missing required fields: {missing_fields}")
+            raise HTTPException(
+                status_code=422, 
+                detail=f"Missing required fields: {', '.join(missing_fields)}"
+            )
+
+        logger.info(f"Verifying payment for order_id: {razorpay_order_id}")
+        is_valid = verify_payment_signature(razorpay_order_id, razorpay_payment_id, razorpay_signature)
+        
+        if is_valid:
+            if user_id and supabase:
+                logger.info(f"Updating Supabase for user_id: {user_id}")
+                try:
+                    supabase.table("users").update({"is_premium": True}).eq("user_id", user_id).execute()
+                    logger.info(f"Successfully updated premium status for user: {user_id}")
+                except Exception as e:
+                    logger.error(f"Failed to update Supabase: {str(e)}")
+            
+            return JSONResponse(content={"success": True, "message": "Payment verified successfully"})
+        else:
+            logger.warning(f"Payment verification failed for order: {razorpay_order_id}")
+            return JSONResponse(content={"success": False, "message": "Payment verification failed"}, status_code=400)
+            
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error verifying payment: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Verification error: {str(e)}")
+
+@app.get("/")
+async def index():
+    return {
+        "status": "Gemini Image Editing API with Razorpay is running",
+        "endpoints": {
+            "generate": "POST /generate",
+            "create_order": "POST /create-razorpay-order",
+            "verify_payment": "POST /verify-razorpay-payment"
+        }
+    }
+
+@app.get("/health")
+async def health_check():
+    return {
+        "status": "healthy",
+        "razorpay_configured": bool(RAZORPAY_KEY_ID and RAZORPAY_KEY_SECRET),
+        "supabase_configured": bool(SUPABASE_URL and SUPABASE_KEY),
+        "gemini_configured": bool(GEMINI_API_KEY)
+    }
+
+if __name__ == "__main__":
+    import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)

index.html

@@ -0,0 +1,34 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+  <meta charset="UTF-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+  <title>Transformers.js - Background Removal</title>
+  <script type="module" crossorigin src="/assets/index-E_M5nW8h.js"></script>
+  <link rel="stylesheet" crossorigin href="/assets/index-SojaDS6z.css">
+</head>
+
+<body>
+  <h1>Background Removal w/ <a href="http://github.com/xenova/transformers.js" target="_blank">🤗 Transformers.js</a>
+  </h1>
+  <h4>Runs locally in your browser, powered by the <a href="https://huggingface.co/briaai/RMBG-1.4" target="_blank">RMBG V1.4 model</a> from <a
+      href="https://bria.ai/" target="_blank">BRIA AI</a>
+  </h4>
+  <div id="container">
+    <label id="upload-button" for="upload">
+      <svg width="25" height="25" viewBox="0 0 25 25" fill="none" xmlns="http://www.w3.org/2000/svg">
+        <path fill="#000"
+          d="M3.5 24.3a3 3 0 0 1-1.9-.8c-.5-.5-.8-1.2-.8-1.9V2.9c0-.7.3-1.3.8-1.9.6-.5 1.2-.7 2-.7h18.6c.7 0 1.3.2 1.9.7.5.6.7 1.2.7 2v18.6c0 .7-.2 1.4-.7 1.9a3 3 0 0 1-2 .8H3.6Zm0-2.7h18.7V2.9H3.5v18.7Zm2.7-2.7h13.3c.3 0 .5 0 .6-.3v-.7l-3.7-5a.6.6 0 0 0-.6-.2c-.2 0-.4 0-.5.3l-3.5 4.6-2.4-3.3a.6.6 0 0 0-.6-.3c-.2 0-.4.1-.5.3l-2.7 3.6c-.1.2-.2.4 0 .7.1.2.3.3.6.3Z">
+        </path>
+      </svg>
+      Click to upload image
+      <label id="example">(or try example)</label>
+    </label>
+  </div>
+  <label id="status"></label>
+  <input id="upload" type="file" accept="image/*" />
+
+</body>
+
+</html>

requirements.txt

@@ -1,10 +1,10 @@
-fastapi==0.115.0
-uvicorn==0.30.6
-python-multipart==0.0.9
-requests==2.32.3
-PyJWT==2.9.0
-Pillow==10.4.0
-razorpay==1.4.1
-supabase==2.9.1
-pydantic==2.8.2
-google-generativeai
+google-generativeai==0.8.3
+fastapi==0.115.0
+uvicorn==0.30.6
+python-multipart==0.0.9
+requests==2.32.3
+PyJWT==2.9.0
+Pillow==10.4.0
+razorpay==1.4.1
+supabase==2.9.1
+pydantic==2.8.2

style.css

@@ -0,0 +1,408 @@
+/* Creating a more modern looking UI for the Gemini Image Editing App */
+/* Global Styles */
+:root {
+  --primary-color: #4f46e5;
+  --primary-light: #6366f1;
+  --primary-dark: #3730a3;
+  --secondary-color: #9333ea;
+  --accent-color: #ec4899;
+  --text-color: #0f172a;
+  --text-light: #64748b;
+  --bg-color: #f8fafc;
+  --card-bg: #ffffff;
+  --border-color: #e2e8f0;
+  --shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+  --transition: all 0.3s ease;
+  --radius: 12px;
+  --header-gradient: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+}
+
+/* Dark mode support */
+.dark {
+  --text-color: #e2e8f0;
+  --text-light: #94a3b8;
+  --bg-color: #0f172a;
+  --card-bg: #1e293b;
+  --border-color: #334155;
+}
+
+/* Base container styling */
+.gradio-container {
+  font-family: 'Inter', system-ui, -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
+  background-color: var(--bg-color);
+  color: var(--text-color);
+  max-width: 1200px;
+  margin: 0 auto;
+  padding: 2rem;
+  box-shadow: var(--shadow);
+  border-radius: var(--radius);
+}
+
+/* Header styling */
+.header-container {
+  background: var(--header-gradient);
+  border-radius: var(--radius);
+  padding: 2rem;
+  margin-bottom: 2rem;
+  box-shadow: var(--shadow);
+  color: white;
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  gap: 20px;
+}
+
+.header-container img {
+  width: 100px;
+  height: 100px;
+  filter: drop-shadow(0 0 8px rgba(255, 255, 255, 0.5));
+  transition: var(--transition);
+}
+
+.header-container img:hover {
+  transform: scale(1.05) rotate(5deg);
+}
+
+.header-container h1 {
+  font-size: 2.5rem;
+  font-weight: 700;
+  margin-bottom: 0.5rem;
+  background: linear-gradient(45deg, #fff, #f0f0ff);
+  -webkit-background-clip: text;
+  background-clip: text;
+  color: transparent;
+  display: inline-block;
+}
+
+.header-container a {
+  color: white;
+  text-decoration: none;
+  font-weight: 600;
+  border-bottom: 2px solid rgba(255, 255, 255, 0.5);
+  transition: var(--transition);
+  padding-bottom: 2px;
+}
+
+.header-container a:hover {
+  border-color: white;
+}
+
+/* Accordion styling */
+.gr-accordion {
+  border: none !important;
+  background: var(--card-bg);
+  border-radius: var(--radius);
+  overflow: hidden;
+  box-shadow: var(--shadow);
+  margin-bottom: 1.5rem;
+  transition: var(--transition);
+}
+
+.gr-accordion:hover {
+  box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05);
+}
+
+.gr-accordion-title {
+  background-color: var(--primary-color);
+  color: white !important;
+  padding: 1rem 1.5rem;
+  font-weight: 600;
+  cursor: pointer;
+}
+
+
+/* Input fields */
+.gr-input, .gr-textarea {
+  width: 100%;
+  padding: 0.75rem 1rem;
+  border: 1px solid var(--border-color);
+  border-radius: var(--radius);
+  background-color: var(--card-bg);
+  transition: var(--transition);
+  font-size: 1rem;
+  color: var(--text-color);
+}
+
+.gr-input:focus, .gr-textarea:focus {
+  border-color: var(--primary-color);
+  box-shadow: 0 0 0 3px rgba(79, 70, 229, 0.2);
+  outline: none;
+}
+
+.gr-form .gr-form-label {
+  font-weight: 600;
+  color: var(--text-color);
+  margin-bottom: 0.5rem;
+}
+
+/* Adding Dark mode handling */
+/* Generate button - Modified to use header gradient in both light and dark mode */
+.gr-button.gr-button-primary, 
+button.generate-btn,
+#component-54 > div.svelte-1pf7t34 > div > button {
+  background: var(--header-gradient) !important;
+  color: white !important;
+  border: none !important;
+  border-radius: var(--radius) !important;
+  padding: 0.75rem 1.5rem !important;
+  font-size: 1rem !important;
+  font-weight: 600 !important;
+  cursor: pointer !important;
+  transition: var(--transition) !important;
+  width: 100% !important;
+  text-transform: uppercase !important;
+  letter-spacing: 1px !important;
+  box-shadow: 0 4px 6px -1px rgba(79, 70, 229, 0.4) !important;
+}
+
+.gr-button.gr-button-primary:hover,
+button.generate-btn:hover,
+#component-54 > div.svelte-1pf7t34 > div > button:hover {
+  transform: translateY(-2px) !important;
+  box-shadow: 0 10px 15px -3px rgba(79, 70, 229, 0.4) !important;
+}
+
+.gr-button.gr-button-primary:active,
+button.generate-btn:active,
+#component-54 > div.svelte-1pf7t34 > div > button:active {
+  transform: translateY(1px) !important;
+}
+
+/* Gallery output */
+.gr-gallery {
+  background-color: var(--card-bg);
+  border-radius: var(--radius);
+  padding: 1rem;
+  box-shadow: var(--shadow);
+  min-height: 300px;
+}
+
+.gr-gallery img {
+  border-radius: calc(var(--radius) - 4px);
+  transition: var(--transition);
+  object-fit: contain !important;
+  max-height: 100% !important;
+  max-width: 100% !important;
+}
+
+.gr-gallery img:hover {
+  transform: scale(1.02);
+}
+
+/* Output text area */
+.output-text {
+  background-color: var(--card-bg);
+  border-radius: var(--radius);
+  padding: 1.5rem;
+  box-shadow: var(--shadow);
+  line-height: 1.6;
+  min-height: 100px;
+  color: var(--text-color);
+}
+
+/* Examples section - Fixed for dark mode */
+.gr-examples-header {
+  font-weight: 600 !important;
+  margin: 2rem 0 1rem !important;
+  padding-bottom: 0.5rem !important;
+  /* Removed border-bottom to avoid duplicate lines */
+  color: var(--primary-light) !important;
+  font-size: 1.5rem !important;
+  display: block !important;
+  visibility: visible !important;
+}
+
+/* Handling duplicate lines coming up below the examples header */
+.gradio-container hr {
+  display: none !important;
+}
+
+/* Fix horizontal separator styling if needed */
+.gradio-container hr:first-of-type {
+  display: block !important;
+  border-top: 2px solid var(--primary-light) !important;
+  margin: 1rem 0 2rem 0 !important;
+}
+
+/* Fix dark mode examples grid */
+.gr-examples {
+  display: grid;
+  grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
+  gap: 1rem;
+  margin-top: 1.5rem;
+}
+
+/* Fix dark mode examples background */
+.gr-sample {
+  background-color: var(--card-bg) !important;
+  border-radius: var(--radius);
+  overflow: hidden;
+  box-shadow: var(--shadow);
+  transition: var(--transition);
+  cursor: pointer;
+}
+
+.gr-sample:hover {
+  transform: translateY(-2px);
+  box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1);
+}
+
+.gr-sample img {
+  width: 100%;
+  height: 150px;
+  object-fit: cover;
+}
+
+/* Examples table styling fixes for dark mode */
+table {
+  background-color: transparent !important;
+  color: var(--text-color) !important;
+}
+
+table tr {
+  background-color: var(--card-bg) !important;
+  color: var(--text-color) !important;
+}
+
+table td {
+  background-color: var(--card-bg) !important;
+  color: var(--text-color) !important;
+  border-color: var(--border-color) !important;
+}
+
+/* Responsive adjustments */
+@media (max-width: 768px) {
+  .gradio-container {
+    padding: 1rem;
+  }
+  
+  .header-container {
+    flex-direction: column;
+    text-align: center;
+    padding: 1.5rem;
+  }
+  
+  .header-container h1 {
+    font-size: 2rem;
+  }
+}
+
+/* Custom scrollbar */
+::-webkit-scrollbar {
+  width: 8px;
+  height: 8px;
+}
+
+::-webkit-scrollbar-track {
+  background: var(--bg-color);
+}
+
+::-webkit-scrollbar-thumb {
+  background: var(--primary-light);
+  border-radius: 4px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+  background: var(--primary-color);
+}
+
+/* Ensure proper viewport handling for mobile */
+@media screen and (max-width: 767px) {
+  body, html {
+    overflow-x: hidden !important;
+    width: 100vw !important;
+    max-width: 100% !important;
+  }
+  
+  /* Force central alignment for the entire app container */
+  .gradio-container {
+    padding: 1rem !important;
+    margin: 0 auto !important;
+    max-width: 100% !important;
+    width: 100% !important;
+    left: 0 !important;
+    right: 0 !important;
+    box-sizing: border-box !important;
+    overflow-x: hidden !important;
+  }
+  
+  /* Fix header container on mobile */
+  .header-container {
+    flex-direction: column !important;
+    text-align: center !important;
+    padding: 1rem !important;
+    width: 100% !important;
+    box-sizing: border-box !important;
+    margin-left: auto !important;
+    margin-right: auto !important;
+  }
+  
+  /* Ensure all content blocks are properly centered */
+  .main-content, 
+  .input-column, 
+  .output-column,
+  .gr-form,
+  .gr-panel,
+  .gr-box,
+  .gr-input,
+  .gr-text-input,
+  .gr-gallery,
+  .gr-button,
+  #component-54 > div.svelte-1pf7t34 > div > button {
+    width: 100% !important;
+    max-width: 100% !important;
+    margin-left: auto !important;
+    margin-right: auto !important;
+    box-sizing: border-box !important;
+  }
+  
+ 
+  /* Fix examples area */
+  .gr-examples {
+    grid-template-columns: 1fr !important;
+    width: 100% !important;
+  }
+  
+  /* Fix example tables on mobile */
+  table {
+    table-layout: fixed !important;
+    width: 100% !important;
+  }
+  
+  table td {
+    word-break: break-word !important;
+  }
+}
+
+/* Fix for extreme narrow screens (small Android phones) */
+@media screen and (max-width: 380px) {
+  .header-container h1 {
+    font-size: 1.8rem !important;
+  }
+  
+  .header-container p {
+    font-size: 0.9rem !important;
+  }
+  
+  /* Stack everything vertically */
+  .gr-panel {
+    display: flex !important;
+    flex-direction: column !important;
+  }
+}
+
+/* Custom fix for the container to always stay centered on any device */
+#gradio-app {
+  display: flex !important;
+  justify-content: center !important;
+  width: 100% !important;
+  max-width: 100% !important;
+  margin: 0 auto !important;
+  overflow-x: hidden !important;
+}
+
+/* Ensure the root container doesn't cause horizontal scroll */
+.root {
+  max-width: 100% !important;
+  overflow-x: hidden !important;
+}