Update app.py

app.py

@@ -1,36 +1,91 @@
-import gradio as gr
-import torch
-import modin.pandas as pd
-import numpy as np
-from diffusers import DiffusionPipeline
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-if torch.cuda.is_available():
-    torch.cuda.max_memory_allocated(device=device)
-    torch.cuda.empty_cache()
-    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
-    pipe.enable_xformers_memory_efficient_attention()
-    pipe = pipe.to(device)
-    torch.cuda.empty_cache()
-else:
-    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
-    pipe = pipe.to(device)
-
-def genie(prompt, steps, seed):
-    generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
-    int_image = pipe(prompt=prompt, generator=generator, num_inference_steps=steps, guidance_scale=0.0).images[0]
-    return int_image
-
-gr.Interface(
-    fn=genie,
-    inputs=[
-        gr.Textbox(label="What you want the AI to generate. 77 Token Limit."),
-        gr.Slider(1, maximum=5, value=2, step=1, label="Number of Iterations"),
-        gr.Slider(minimum=0, step=1, maximum=999999999999999999, randomize=True),
-    ],
-    outputs="image",
-    title="Stable Diffusion Turbo CPU or GPU",
-    description="SDXL Turbo CPU or GPU. Currently running on CPU. <br><br><b>WARNING: This model is capable of producing NSFW (Softcore) images.</b>",
-    article="Created by [Your Name]."
-).launch(debug=True, max_threads=80)
+# app.py
+from flask import Flask, request, jsonify
+import razorpay
+import os
+import hmac
+import hashlib
+import json
+
+app = Flask(__name__)
+
+# Get keys from Hugging Face Secrets (Settings -> Variables)
+RAZORPAY_KEY_ID = os.getenv("RAZORPAY_KEY_ID", "")
+RAZORPAY_KEY_SECRET = os.getenv("RAZORPAY_KEY_SECRET", "")
+
+if not RAZORPAY_KEY_ID or not RAZORPAY_KEY_SECRET:
+    print("WARNING: Razorpay keys not found in env. Set RAZORPAY_KEY_ID and RAZORPAY_KEY_SECRET in Space secrets.")
+
+client = razorpay.Client(auth=(RAZORPAY_KEY_ID, RAZORPAY_KEY_SECRET))
+
+def cors_json(resp):
+    # Helper to add CORS headers to Flask response dict
+    response = jsonify(resp)
+    response.headers.add("Access-Control-Allow-Origin", "*")
+    response.headers.add("Access-Control-Allow-Headers", "Content-Type")
+    return response
+
+@app.route("/create-order", methods=["POST", "OPTIONS"])
+def create_order():
+    # Preflight
+    if request.method == "OPTIONS":
+        response = app.response_class(status=204)
+        response.headers.add("Access-Control-Allow-Origin", "*")
+        response.headers.add("Access-Control-Allow-Headers", "Content-Type")
+        return response
+
+    data = request.get_json(force=True) or {}
+    amount = int(data.get("amount", 0))   # amount in rupees from frontend
+    if amount <= 0:
+        return cors_json({"error": "Invalid amount"}), 400
+
+    try:
+        options = {
+            "amount": amount * 100,  # paise
+            "currency": "INR",
+            "receipt": f"receipt_{os.urandom(4).hex()}",
+            "payment_capture": 1
+        }
+        order = client.order.create(options)
+        return cors_json(order)
+    except Exception as e:
+        return cors_json({"error": str(e)}), 500
+
+@app.route("/verify-payment", methods=["POST", "OPTIONS"])
+def verify_payment():
+    # Verifies signature from Razorpay checkout
+    if request.method == "OPTIONS":
+        response = app.response_class(status=204)
+        response.headers.add("Access-Control-Allow-Origin", "*")
+        response.headers.add("Access-Control-Allow-Headers", "Content-Type")
+        return response
+
+    data = request.get_json(force=True) or {}
+    try:
+        razorpay_order_id = data.get("razorpay_order_id")
+        razorpay_payment_id = data.get("razorpay_payment_id")
+        razorpay_signature = data.get("razorpay_signature")
+
+        if not (razorpay_order_id and razorpay_payment_id and razorpay_signature):
+            return cors_json({"success": False, "message": "Missing fields"}), 400
+
+        payload = f"{razorpay_order_id}|{razorpay_payment_id}"
+        generated_signature = hmac.new(
+            RAZORPAY_KEY_SECRET.encode("utf-8"),
+            payload.encode("utf-8"),
+            hashlib.sha256
+        ).hexdigest()
+
+        if generated_signature == razorpay_signature:
+            return cors_json({"success": True})
+        else:
+            return cors_json({"success": False, "message": "Invalid signature"}), 400
+    except Exception as e:
+        return cors_json({"success": False, "message": str(e)}), 500
+
+@app.route("/health", methods=["GET"])
+def health():
+    return cors_json({"ok": True, "app": "razorpay-backend"})
+
+if __name__ == "__main__":
+    # Hugging Face Spaces runs on port 7860
+    app.run(host="0.0.0.0", port=7860)