Update app.py

app.py

@@ -1,7 +1,7 @@
 import logging
-from fastapi import FastAPI, UploadFile, File, HTTPException, Form
+from fastapi import FastAPI, UploadFile, File, HTTPException, Form, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse
+from fastapi.responses import FileResponse, JSONResponse
 import requests
 import base64
 import os
@@ -10,18 +10,20 @@ import jwt
 from pathlib import Path
 from typing import List
 import io
+import razorpay
+from razorpay.errors import SignatureVerificationError
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Initialize FastAPI app
-app = FastAPI(title="Kling AI Multi-Image Generator API")
+app = FastAPI(title="Kling AI Multi-Image Generator API with Razorpay")
 
 # Enable CORS for the frontend
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["https://hivili.web.app"],  # Allow your frontend
+    allow_origins=["https://hivili.web.app"],  # Update with your frontend URL
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -33,6 +35,11 @@ ACCESS_KEY_SECRET = os.getenv("ACCESS_KEY_SECRET", "TTepeLyBterLNM3brYPGmdndBnny
 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")  # Set in environment variables
+RAZORPAY_KEY_SECRET = os.getenv("RAZORPAY_KEY_SECRET")  # Set in environment variables
+razorpay_client = razorpay.Client(auth=(RAZORPAY_KEY_ID, RAZORPAY_KEY_SECRET))
+
 # ===== AUTHENTICATION =====
 def generate_jwt_token():
     """Generate JWT token for API authentication"""
@@ -110,6 +117,38 @@ def check_task_status(task_id: str):
     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, currency: str = "INR"):
+    """Create a Razorpay order"""
+    try:
+        order_data = {
+            "amount": amount * 100,  # Amount in paise (e.g., 500 INR = 50000 paise)
+            "currency": currency,
+            "payment_capture": 1  # Auto-capture payment
+        }
+        order = razorpay_client.order.create(data=order_data)
+        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)
+        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"""
@@ -175,9 +214,46 @@ async def generate_image_endpoint(
         logger.error(f"Error in /generate: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
+@app.post("/create-order")
+async def create_order(amount: int = Form(...)):
+    """Create a Razorpay order for payment"""
+    if not RAZORPAY_KEY_ID or not RAZORPAY_KEY_SECRET:
+        raise HTTPException(status_code=500, detail="Razorpay configuration missing")
+    
+    try:
+        order = create_razorpay_order(amount)
+        return JSONResponse(content={
+            "order_id": order["id"],
+            "amount": order["amount"],
+            "currency": order["currency"],
+            "key_id": RAZORPAY_KEY_ID
+        })
+    except Exception as e:
+        logger.error(f"Error creating order: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/verify-payment")
+async def verify_payment(
+    razorpay_order_id: str = Form(...),
+    razorpay_payment_id: str = Form(...),
+    razorpay_signature: str = Form(...)
+):
+    """Verify Razorpay payment signature"""
+    try:
+        is_valid = verify_payment_signature(razorpay_order_id, razorpay_payment_id, razorpay_signature)
+        if is_valid:
+            # Here you can update user subscription status in your database
+            # For example, update a user's plan to "premium" or increment credits
+            return JSONResponse(content={"status": "success", "message": "Payment verified successfully"})
+        else:
+            raise HTTPException(status_code=400, detail="Payment verification failed")
+    except Exception as e:
+        logger.error(f"Error verifying payment: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.get("/")
 async def index():
-    return {"status": "Kling AI Multi-Image Generator API is running"}
+    return {"status": "Kling AI Multi-Image Generator API with Razorpay is running"}
 
 if __name__ == "__main__":
     import uvicorn