Update app.py

app.py

@@ -4,29 +4,43 @@ import streamlit as st
 # Hardcoded credentials
 USERNAME = os.environ["USERNAME"]
 PASSWORD = os.environ["PASSWORD"]
+BASE_CONTENT_CODE_ASSIST_T2_MICRO = os.environ["BASE_CONTENT_CODE_ASSIST_T2_MICRO"]
 
-# Sidebar for login with emojis
-st.sidebar.title("🔒 AIXNet Login")
-with st.sidebar.form(key='login_form'):
-    username = st.text_input("👤 Username")
-    password = st.text_input("🔑 Password", type="password")
-    login_button = st.form_submit_button(label="🔓 Login")
+# Initialize session state
+if 'logged_in' not in st.session_state:
+    st.session_state.logged_in = False
 
-if login_button:
-    if username == USERNAME and password == PASSWORD:
-        st.sidebar.success("🎉 Login successful!")
-        
-        # Display table after successful login
-        st.title("📋 AIXNet Tasks")
-        data = {
-            "📝 Task": ["💻 Code assist"],
-            "🖥️ Instance Type": ["t2.micro"],
-            "💰 Price": ["$0.05 / hour"],
-            "🌐 IPv4": ["some_url_here"]
-        }
+# Sidebar for login/logout with emojis
+st.sidebar.title("🔒 AIXNet")
 
-        st.table(data)
-    else:
-        st.sidebar.error("❌ Invalid username or password. Please try again.")
+if st.session_state.logged_in:
+    st.sidebar.success("🎉 You are logged in!")
+    if st.sidebar.button("🔓 Log out"):
+        st.session_state.logged_in = False
+        st.sidebar.info("You have logged out.")
 else:
-    st.sidebar.info("👉 Please enter your credentials to log in.")
+    with st.sidebar.form(key='login_form'):
+        username = st.text_input("👤 Username")
+        password = st.text_input("🔑 Password", type="password")
+        login_button = st.form_submit_button(label="🔓 Log in")
+
+    if login_button:
+        if username == USERNAME and password == PASSWORD:
+            st.session_state.logged_in = True
+            st.sidebar.success("🎉 Login successful!")
+        else:
+            st.sidebar.error("❌ Invalid username or password. Please try again.")
+
+# Display table only if logged in
+if st.session_state.logged_in:
+    st.title("📋 AIXNet Tasks")
+    data = {
+        "📝 Task": ["💻 Code assist"],
+        "🖥️ Instance Type": ["t2.micro"],
+        "💰 Price": ["$0.05 / hour"],
+        "🌐 IPv4": [BASE_CONTENT_CODE_ASSIST_T2_MICRO]
+    }
+
+    st.table(data)
+else:
+    st.info("👉 Please log in to view the tasks.")