Update app.py

app.py

@@ -1,41 +1,7 @@
 import os
 import streamlit as st
-import stripe
-import requests
 import time
-
-# 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://39701j614g.execute-api.us-east-1.amazonaws.com/dev/?sessionID={CHECKOUT_SESSION_ID}&key=123",
-            cancel_url="https://wyn-education.streamlit.app/",
-        )
-        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
+from utils.helper import create_checkout_session, check_payment_status
 
 # Streamlit app
 st.set_page_config(layout="wide")