Update app.py

app.py

@@ -29,8 +29,6 @@ def add_auth(
             login_button_color=login_button_color,
         )
 
-
-
 def require_auth(
     login_button_text: str = "Login with Google",
     login_button_color: str = "#FD504D",
@@ -116,45 +114,46 @@ st.set_page_config(
 add_auth(required=True, login_button_color="#FFA500",subscribe_color_button="#FFA500")
 
 if st.session_state.email is not '':
-      st.write(f'User: {st.session_state.email}')
-
-
-if "buttons" in st.session_state:
-   st.session_state.buttons = st.session_state.buttons
-
-
-
-from PIL import Image
-import base64
-
-def get_base64_of_bin_file(bin_file):
-    with open(bin_file, 'rb') as f:
-        data = f.read()
-    return base64.b64encode(data).decode()
-
-def set_png_as_page_bg(png_file):
-    bin_str = get_base64_of_bin_file(png_file)
-    page_bg_img = '''
-    <style>
-    .stApp {
-        background-image: url("data:image/png;base64,%s");
-        background-size: cover;
-    }
-    </style>
-    ''' % bin_str
-    st.markdown(page_bg_img, unsafe_allow_html=True)
-
-
-if st.toggle('Set background image', True):
-    set_png_as_page_bg('saucer.webp')  # Replace with your background image path
-
-
-pg = st.navigation([
-            st.Page("rag_search.py", title="Smart-Search (Retrieval Augmented Generations)", icon="🔍"),
-            st.Page("parsing.py", title="UAP Feature Extraction (Shape, Speed, Color)", icon="📄"),
-            st.Page("analyzing.py", title="Statistical Analysis (UMAP+HDBSCAN, XGBoost, V-Cramer)", icon="🧠"),
-            st.Page("magnetic.py", title="Magnetic Anomaly Detection (InterMagnet Stations)", icon="🧲"),
-            st.Page("map.py", title="Interactive Map (Tracking variations, Proximity with Military Bases, Nuclear Facilities)", icon="🗺️"),
-        ])
-
-pg.run()
+        st.write('')
+        st.write(f'User: {st.session_state.email}')
+        
+        if "buttons" in st.session_state:
+           st.session_state.buttons = st.session_state.buttons
+        
+        from PIL import Image
+        import base64
+        
+        def get_base64_of_bin_file(bin_file):
+            with open(bin_file, 'rb') as f:
+                data = f.read()
+            return base64.b64encode(data).decode()
+        
+        def set_png_as_page_bg(png_file):
+            bin_str = get_base64_of_bin_file(png_file)
+            page_bg_img = '''
+            <style>
+            .stApp {
+                background-image: url("data:image/png;base64,%s");
+                background-size: cover;
+            }
+            </style>
+            ''' % bin_str
+            st.markdown(page_bg_img, unsafe_allow_html=True)
+        
+        
+        if st.toggle('Set background image', True):
+            set_png_as_page_bg('saucer.webp')  # Replace with your background image path
+        
+        if st.session_state.user_subscribed:     
+            pg = st.navigation([
+                        st.Page("rag_search.py", title="Smart-Search (Retrieval Augmented Generations)", icon="🔍"),
+                        st.Page("parsing.py", title="UAP Feature Extraction (Shape, Speed, Color)", icon="📄"),
+                        st.Page("analyzing.py", title="Statistical Analysis (UMAP+HDBSCAN, XGBoost, V-Cramer)", icon="🧠"),
+                        st.Page("magnetic.py", title="Magnetic Anomaly Detection (InterMagnet Stations)", icon="🧲"),
+                        st.Page("map.py", title="Interactive Map (Tracking variations, Proximity with Military Bases, Nuclear Facilities)", icon="🗺️"),
+                    ])
+            pg.run()
+        else:
+            st.write('Please subscribe to access tools')
+    else:
+        st.write('Please login to your account')