Update app.py

app.py

@@ -1,52 +1,42 @@
+import os
+
 import streamlit as st
 import stripe
-import os
+import requests
+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 retrieve product information from Stripe
-def get_product(product_id):
-    try:
-        product = stripe.Product.retrieve(product_id)
-        price = stripe.Price.list(product=product_id)
-        return product, price.data[0]
-    except Exception as e:
-        st.error(f"Error fetching product information: {e}")
-        return None, None
-
-# Function to create a checkout session
-def create_checkout_session(price_id):
+# Function to simulate checking payment status (for demo purposes)
+def check_payment_status(session_id):
     try:
-        session = stripe.checkout.Session.create(
-            payment_method_types=['card'],
-            line_items=[{
-                'price': price_id,
-                'quantity': 1,
-            }],
-            mode='payment',
-            success_url='https://buy.stripe.com/14kbLG4FV5s509OdQV',
-            cancel_url='https://www.future-minds.io/',
-        )
-        return session.url
-    except Exception as e:
-        st.error(f"Error creating checkout session: {e}")
-        return None
+        session = stripe.checkout.Session.retrieve(session_id)
+        return session.payment_status == 'paid'
+    except stripe.error.StripeError as e:
+        st.error(f"Error checking payment status: {e}")
+        return False
 
 # Streamlit app
-st.title("Stripe Product Checkout")
-
-product_id = st.text_input("Enter Stripe Product ID", value="your-product-id")
-
-if st.button("Get Product Info"):
-    product, price = get_product(product_id)
-    if product:
-        st.write("### Product Information")
-        st.write(f"**Name:** {product['name']}")
-        st.write(f"**Description:** {product['description']}")
-        st.write(f"**Price:** ${price['unit_amount'] / 100:.2f}")
-
-        if st.button("Checkout"):
-            checkout_url = create_checkout_session(price['id'])
-            if checkout_url:
-                st.markdown(f"[Proceed to Checkout]({checkout_url})", unsafe_allow_html=True)
+st.title("Stripe Payment Integration")
+
+# 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)
+
+# 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:
+        st.write("Checking payment status, please wait...")
+        time.sleep(2)  # Simulating delay for payment processing
+
+        if check_payment_status(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.")