Update app.py

app.py

@@ -44,41 +44,26 @@ st.title("Stripe Payment Integration")
 if 'session_id' not in st.session_state:
     st.session_state.session_id = None
 
-def list_prices_for_product(product_id):
-    try:
-        prices = stripe.Price.list(product=product_id)
-        return prices
-    except Exception as e:
-        print(f"Error listing prices: {e}")
-        return None
-
-# Example usage
-product_id = stripe_product_id  # Replace with your actual Product ID
-prices = list_prices_for_product(product_id)
-
-if prices:
-    for price in prices.data:
-        print(f"Price ID: {price.id}, Amount: {price.unit_amount}, Currency: {price.currency}")
+# Button to redirect to Stripe Checkout
+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)
 
-# # Button to redirect to Stripe Checkout
-# 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
+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
 
-# # Input for session ID to check payment status
-# 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(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.info("Create a checkout session to get the 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.info("Create a checkout session to get the session ID.")