stripe helper func updated

utils/helper.py

@@ -379,3 +379,48 @@ def call_gpt(prompt: str, content: str) -> str:
 
     # Extracts and returns the response content from the model's completion
     return response.choices[0].message.content
+
+
+import os
+import time
+
+import requests
+import streamlit as st
+import stripe
+
+# Set your secret key. Remember to switch to your live secret key in production!
+stripe.api_key = os.environ["STRIPE_API_KEY"]
+
+# Set the product id.
+stripe_price_id = os.environ["STRIPE_PRICE_ID"]
+
+
+# Function to create a Stripe Checkout Session
+def create_checkout_session():
+    try:
+        session = stripe.checkout.Session.create(
+            payment_method_types=["card"],
+            line_items=[
+                {
+                    "price": stripe_price_id,  # Replace with your actual Stripe price ID
+                    "quantity": 1,
+                }
+            ],
+            mode="payment",
+            success_url="https://your-website.com/success?session_id={CHECKOUT_SESSION_ID}",
+            cancel_url="https://your-website.com/cancel",
+        )
+        return session.id, session.url
+    except Exception as e:
+        st.error(f"Error creating checkout session: {e}")
+        return None, None
+
+
+# Function to check payment status
+def check_payment_status(session_id):
+    try:
+        session = stripe.checkout.Session.retrieve(session_id)
+        return session.payment_status == "paid"
+    except Exception as e:
+        st.error(f"Error checking payment status: {e}")
+        return False