Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-
 import streamlit as st
 import stripe
 import requests
@@ -39,30 +38,43 @@ def check_payment_status(session_id):
         return False
 
 # Streamlit app
-st.title("Stripe Payment Integration")
+st.set_page_config(layout="wide")
+st.title("Stripe Payment Integration 💳")
 
+# Sidebar for payment processing
+st.sidebar.title("Payment Section 🛒")
 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"):
+if st.sidebar.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)
+        st.session_state.payment_checked = False
+        st.sidebar.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...")
+    st.sidebar.write(f"Your session ID: {st.session_state.session_id}")
+    if st.sidebar.button("Check Payment Status"):
+        st.sidebar.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
+            st.success("Payment successful! 🎉")
+            st.sidebar.write("Payment successful! 🎉")
         else:
-            st.error("Payment not completed yet. Please try again.")
-else:
-    st.info("Create a checkout session to get the session ID.")
+            st.sidebar.error("Payment not completed yet. Please try again. ❌")
 
+# Display the paid content on the main page if payment is successful
+if st.session_state.payment_checked:
+    st.write("Here's the paid content. 🏆")
+    if st.button("Click for More Info"):
+        st.write("Here is the additional information for paid users! 📄")
+else:
+    st.info("Create a checkout session to get the session ID. 📋")