Update app.py

app.py

@@ -8,35 +8,58 @@ import time
 # Set your secret key. Remember to switch to your live secret key in production!
 stripe.api_key = os.environ["STRIPE_API_KEY"]
 
-# Function to simulate checking payment status (for demo purposes)
+# Function to create a Stripe Checkout Session
+def create_checkout_session():
+    try:
+        session = stripe.checkout.Session.create(
+            payment_method_types=['card'],
+            line_items=[{
+                'price': 'your-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 stripe.error.StripeError as e:
+    except Exception as e:
         st.error(f"Error checking payment status: {e}")
         return False
 
 # Streamlit app
 st.title("Stripe Payment Integration")
 
+if 'session_id' not in st.session_state:
+    st.session_state.session_id = None
+
 # Button to redirect to Stripe Checkout
-if st.button("Go to Checkout"):
-    checkout_url = "https://buy.stripe.com/14kbLG4FV5s509OdQV"
-    st.markdown(f"[Proceed to Checkout]({checkout_url})", unsafe_allow_html=True)
+if st.button("Create Checkout Session"):
+    session_id, checkout_url = create_checkout_session()
+    if session_id and checkout_url:
+        st.session_state.session_id = session_id
+        st.write(f"Checkout URL: {checkout_url}")
+        st.markdown(f"[Proceed to Checkout]({checkout_url})", unsafe_allow_html=True)
 
 # Input for session ID to check payment status
-session_id = st.text_input("Enter your Stripe Session ID after payment:")
-
-if st.button("Check Payment Status"):
-    if session_id:
+if st.session_state.session_id:
+    st.write(f"Your session ID: {st.session_state.session_id}")
+    if st.button("Check Payment Status"):
         st.write("Checking payment status, please wait...")
         time.sleep(2)  # Simulating delay for payment processing
 
-        if check_payment_status(session_id):
+        if check_payment_status(st.session_state.session_id):
             st.success("Payment successful!")
             st.write("Here's the paid content.")
         else:
             st.error("Payment not completed yet. Please try again.")
-    else:
-        st.error("Please enter a valid session ID.")
+else:
+    st.info("Create a checkout session to get the session ID.")