stripe api integration with state

pages/1_english_.py

@@ -1,8 +1,46 @@
+# import streamlit as st
+# from utils.helper import *
+# from utils.ui_helper import *
+
+
+# # Streamlit app
+# st.set_page_config(layout="wide")
+# 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("Create Checkout Session"):
+#     session_id, checkout_url = create_checkout_session()
+#     if session_id and checkout_url:
+#         st.session_state.session_id = session_id
+#         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
+
+#         if check_payment_status(st.session_state.session_id):
+#             st.success("Payment successful!")
+#             st.write("Here's the paid content.")
+
+#             # run ui
+#             main_algo_trader()
+#         else:
+#             st.error("Payment not completed yet. Please try again.")
+# else:
+#     st.info("Create a checkout session to get the session ID.")
+
+
 import streamlit as st
+import time
 from utils.helper import *
 from utils.ui_helper import *
 
-
 # Streamlit app
 st.set_page_config(layout="wide")
 st.title("Stripe Payment Integration")
@@ -10,31 +48,33 @@ st.title("Stripe Payment Integration")
 if 'session_id' not in st.session_state:
     st.session_state.session_id = None
 
+if 'payment_checked' not in st.session_state:
+    st.session_state.payment_checked = False
+
 # 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.session_state.payment_checked = False
         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:
-    # Status
     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
 
-    # Checking
-    st.write("Checking payment status, please wait...")
-    time.sleep(2)  # Simulating delay for payment processing
-
-    # Main content
-    if check_payment_status(st.session_state.session_id):
-        st.success("Payment successful!")
-        st.write("Here's the paid content.")
-
-        # run ui
-        main_algo_trader()
-    else:
-        st.error("Payment not completed yet. Please try again.")
+        if check_payment_status(st.session_state.session_id):
+            st.success("Payment successful!")
+            st.write("Here's the paid content.")
+            st.session_state.payment_checked = True
+        else:
+            st.error("Payment not completed yet. Please try again.")
 else:
     st.info("Create a checkout session to get the session ID.")
 
+# Run the main_algo_trader only if the payment has been checked and is successful
+if st.session_state.payment_checked:
+    main_algo_trader()