Update app.py

app.py

@@ -34,15 +34,9 @@ except KeyError:
     st.error("API_URL environment variable is not set.")
     st.stop()
 
-# Try to get the Bearer token from environment variables, if not found set to a default value
-try:
-    BEARER_TOKEN = os.environ["BEARER_TOKEN"]
-except KeyError:
-    st.error("BEARER_TOKEN environment variable is not set.")
-    st.stop()
+
 
 print("API_URL:", os.environ["API_URL"])
-print("BEARER_TOKEN:", os.environ["BEARER_TOKEN"])
 
 page_bg_img = '''
 <style>
@@ -199,25 +193,27 @@ if st.session_state["jwt_token"] is None:
 else:
     # User is logged in, continue with the app
     if st.button("Log out", key="logout_button"):
-        st.session_state["jwt_token"] = None
-        st.session_state["session_id"] = None  # Reset session ID
+        # Clear all user-specific session state
+        st.session_state.clear()
         st.success("Logged out successfully!")
         st.rerun()
 
-# Example usage of the jwt_token in API requests
-headers = {
-    "Authorization": f"Bearer {st.session_state['jwt_token']}",
-    "Content-Type": "application/json",
-    "Cache-Control": "no-store"
-}
 
 # Function to get the auth headers using the current jwt_token
 def get_auth_headers():
     jwt_token = st.session_state.get("jwt_token")
     if jwt_token:
-        return {"Authorization": f"Bearer {jwt_token}"}
+        return {
+            "Authorization": f"Bearer {jwt_token}",
+            "Content-Type": "application/json",
+            "Cache-Control": "no-store",
+        }
     else:
-        return {}
+        return {
+            "Content-Type": "application/json",
+            "Cache-Control": "no-store",
+        }
+
         
 if "session_id" not in st.session_state:
     st.session_state["session_id"] = str(uuid.uuid4())
@@ -414,7 +410,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
 
     # Use the original headers for the API request
     with st.spinner("Generating audio ..."):
-        response = time_post_request(API_URL, headers, payload)
+        response = time_post_request(API_URL, headers=auth_headers, payload=payload)
 
     placeholder1 = st.empty()
     if response.status_code != 200: