Update pages/1_english_.py

pages/1_english_.py

@@ -88,49 +88,44 @@ from utils.ui_helper import *
 st.set_page_config(layout="wide")
 st.title("Stripe Payment Integration")
 
-# Sidebar for session ID input
+# Sidebar for session ID input and payment actions
 with st.sidebar:
-    st.header("Enter your Session ID")
-    session_id_input = st.text_input("Session ID", "")
-    if st.button("Check Payment Status", key="sidebar_check"):
+    st.header("Payment and Session ID")
+    
+    # Initialize session states if not already set
+    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
+    
+    # Session ID input field
+    session_id_input = st.text_input("Enter your Session ID", st.session_state.session_id or "")
+    
+    # Button to check payment status
+    if st.button("Check Payment Status"):
         if session_id_input:
             st.session_state.session_id = session_id_input
-            st.session_state.payment_checked = False
+            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.session_state.payment_checked = True
+            else:
+                st.error("Payment not completed yet. Please try again.")
         else:
             st.warning("Please enter a valid session ID.")
 
-# Initializing session states
-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 create a new checkout session
-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.sidebar.write(f"Your session ID: {st.session_state.session_id}")
-        st.sidebar.markdown(f"[Proceed to Checkout]({checkout_url})", unsafe_allow_html=True)
-
-# Display session ID if available
-if st.session_state.session_id:
-    st.write(f"Your session ID: {st.session_state.session_id}")
-
-    # Button to check payment status
-    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.")
-            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.")
+    # Button to create a new checkout session
+    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.write(f"Your session ID: {st.session_state.session_id}")
+            st.markdown(f"[Proceed to Checkout]({checkout_url})", unsafe_allow_html=True)
 
-# Run the main_algo_trader only if the payment has been checked and is successful
+# Main panel: Only show main_algo_trader() if payment is successful
 if st.session_state.payment_checked:
     main_algo_trader()
+else:
+    st.write("Please complete the payment to access the content.")