Begin adding content

app.py

@@ -121,6 +121,7 @@ def check_password():
                 # Retrieve and store user name and team
                 st.session_state["name"] = user_db.loc[user_db.username == st.session_state["username"], "name"].values[0]
                 st.session_state["team"] = user_db.loc[user_db.username == st.session_state["username"], "team"].values[0]
+                st.session_state["profile_pic"] = user_db.loc[user_db.username == st.session_state["username"], "profile_pic"].values[0]
 
                 # Don't store the username or password
                 del st.session_state["password"]

menu.py

@@ -4,15 +4,57 @@ import streamlit as st
 
 def authenticated_menu():
 
+    st.markdown("""
+        <style>
+        .circle-image {
+            width: 100px;
+            height: 100px;
+            border-radius: 50%;
+            overflow: hidden;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            border: 2px solid black;
+            margin: 0 auto 10px auto;
+        }
+        
+        .circle-image img {
+            width: 100%;
+            height: 100%;
+            object-fit: cover;
+        }
+                
+        .username {
+          font-size: 20px;
+          font-weight: bold;
+          text-align: center;
+          margin-top: 0px;
+        }
+        </style>
+        """, unsafe_allow_html=True)
+    
+    # Show the user's profile picture
+    st.sidebar.markdown(f'<div class="circle-image"><img src="{st.session_state.profile_pic}" /></div>', unsafe_allow_html=True)
+
+    # Show the user's name
+    # st.sidebar.markdown(f"Logged in as {st.session_state.name}.")
+    st.sidebar.markdown(f'<div class="username">{st.session_state.name}</div>', unsafe_allow_html=True)
+    st.sidebar.markdown("---")
+
     # Show a navigation menu for authenticated users
     # st.sidebar.page_link("app.py", label="Switch Accounts", icon="🔒")
-    st.sidebar.page_link("pages/user.py", label="Profile", icon="👤")
+    st.sidebar.page_link("pages/about.py", label="About", icon="📖")
     st.sidebar.page_link("pages/input.py", label="Input", icon="💡")
     st.sidebar.page_link("pages/validate.py", label="Validate", icon="✅")
     st.sidebar.page_link("pages/explore.py", label="Explore", icon="🔍")
     if st.session_state.role in ["admin"]:
         st.sidebar.page_link("pages/admin.py", label="Manage Users", icon="🔧")
 
+    # Show the logout button
+    st.sidebar.markdown("---")
+    st.sidebar.button("Log Out", on_click=lambda: st.session_state.clear())
+    
+
 
 def unauthenticated_menu():
 

pages/about.py

@@ -0,0 +1,17 @@
+import streamlit as st
+from menu import menu_with_redirect
+
+# Path manipulation
+from pathlib import Path
+
+# Custom and other imports
+import project_config
+
+# Redirect to app.py if not logged in, otherwise show the navigation menu
+menu_with_redirect()
+
+# Header
+st.image(str(project_config.MEDIA_DIR / 'about_header.svg'), use_column_width=True)
+
+# Main content
+st.markdown(f"Hello, {st.session_state.name}! Welcome to GRAVITY, a GRaph AI VISualization Tool to query and visualize knowledge graph-grounded biomedical AI models.")

pages/explore.py

@@ -3,13 +3,15 @@ from menu import menu_with_redirect
 
 # Path manipulation
 from pathlib import Path
-import sys
 
 # Custom and other imports
 import project_config
-from utils import add_logo
 
 # Redirect to app.py if not logged in, otherwise show the navigation menu
 menu_with_redirect()
 
-st.title("Explore")
+# Header
+st.image(str(project_config.MEDIA_DIR / 'explore_header.svg'), use_column_width=True)
+
+# Main content
+st.markdown(f"Hello, {st.session_state.name}!")

pages/input.py

@@ -3,13 +3,15 @@ from menu import menu_with_redirect
 
 # Path manipulation
 from pathlib import Path
-import sys
 
 # Custom and other imports
 import project_config
-from utils import add_logo
 
 # Redirect to app.py if not logged in, otherwise show the navigation menu
 menu_with_redirect()
 
-st.title("Input")
+# Header
+st.image(str(project_config.MEDIA_DIR / 'input_header.svg'), use_column_width=True)
+
+# Main content
+st.markdown(f"Hello, {st.session_state.name}!")

pages/user.py

@@ -1,8 +0,0 @@
-import streamlit as st
-from menu import menu_with_redirect
-
-# Redirect to app.py if not logged in, otherwise show the navigation menu
-menu_with_redirect()
-
-st.title("Welcome to Gravity")
-st.markdown(f"Hello, {st.session_state.name}!")

pages/validate.py

@@ -3,13 +3,15 @@ from menu import menu_with_redirect
 
 # Path manipulation
 from pathlib import Path
-import sys
 
 # Custom and other imports
 import project_config
-from utils import add_logo
 
 # Redirect to app.py if not logged in, otherwise show the navigation menu
 menu_with_redirect()
 
-st.title("Validate")
+# Header
+st.image(str(project_config.MEDIA_DIR / 'validate_header.svg'), use_column_width=True)
+
+# Main content
+st.markdown(f"Hello, {st.session_state.name}!")