Update app.py

app.py

@@ -1,5 +1,5 @@
 import logging
-from fastapi import FastAPI, UploadFile, File, HTTPException, Form, Request
+from fastapi import FastAPI, UploadFile, File, HTTPException, Form
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse, JSONResponse
 import requests
@@ -12,6 +12,7 @@ from typing import List
 import io
 import razorpay
 from razorpay.errors import SignatureVerificationError
+from supabase import create_client, Client
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -23,7 +24,7 @@ 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"],  # Update with your frontend URL
+    allow_origins=["https://hivili.web.app", "https://your-lovable-app.lovable.dev"],  # Replace with your Lovable.dev URL
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -36,10 +37,15 @@ 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_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
+
 # ===== AUTHENTICATION =====
 def generate_jwt_token():
     """Generate JWT token for API authentication"""
@@ -118,12 +124,12 @@ def check_task_status(task_id: str):
         raise HTTPException(status_code=500, detail=f"Status check failed: {str(e)}")
 
 # ===== RAZORPAY FUNCTIONS =====
-def create_razorpay_order(amount: int, currency: str = "INR"):
+def create_razorpay_order(amount: int):
     """Create a Razorpay order"""
     try:
         order_data = {
-            "amount": amount * 100,  # Amount in paise (e.g., 500 INR = 50000 paise)
-            "currency": currency,
+            "amount": amount * 100,  # Convert INR to paise
+            "currency": "INR",
             "payment_capture": 1  # Auto-capture payment
         }
         order = razorpay_client.order.create(data=order_data)
@@ -214,16 +220,15 @@ 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"""
+@app.post("/create-razorpay-order")
+async def create_razorpay_order(amount: int = Form(...)):
+    """Create a Razorpay order"""
     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"],
+            "id": order["id"],
             "amount": order["amount"],
             "currency": order["currency"],
             "key_id": RAZORPAY_KEY_ID
@@ -232,21 +237,22 @@ async def create_order(amount: int = Form(...)):
         logger.error(f"Error creating order: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.post("/verify-payment")
-async def verify_payment(
+@app.post("/verify-razorpay-payment")
+async def verify_razorpay_payment(
     razorpay_order_id: str = Form(...),
     razorpay_payment_id: str = Form(...),
-    razorpay_signature: str = Form(...)
+    razorpay_signature: str = Form(...),
+    user_id: str = Form(None)  # Optional for Supabase
 ):
     """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"})
+            if user_id and supabase:
+                supabase.table("users").update({"is_premium": True}).eq("user_id", user_id).execute()
+            return JSONResponse(content={"success": True})
         else:
-            raise HTTPException(status_code=400, detail="Payment verification failed")
+            return JSONResponse(content={"success": False}, status_code=400)
     except Exception as e:
         logger.error(f"Error verifying payment: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))