Add report a bug and get help link + about us text

hexviz/pages/🗺️Identify_Interesting_Heads.py

@@ -3,9 +3,9 @@ import streamlit as st
 from hexviz.attention import get_attention, get_sequence, get_structure
 from hexviz.models import Model, ModelType
 from hexviz.plot import plot_tiled_heatmap
-from hexviz.view import select_model, select_pdb
+from hexviz.view import menu_items, select_model, select_pdb
 
-st.set_page_config(layout="wide")
+st.set_page_config(layout="wide", menu_items=menu_items)
 st.subheader("Find interesting heads and layers")
 
 

hexviz/view.py

@@ -1,5 +1,10 @@
 import streamlit as st
 
+menu_items = {
+    "Get Help": "https://huggingface.co/spaces/aksell/hexviz/discussions/new", 
+    "Report a bug": "https://huggingface.co/spaces/aksell/hexviz/discussions/new", 
+    "About": "Created by [Aksel Lenes](https://github.com/aksell/) from Noelia Ferruz's group at the Institute of Molecular Biology of Barcelona. Read more at https://www.aiproteindesign.com/"
+    }
 
 def get_selecte_model_index(models):
     selected_model_name = st.session_state.get("selected_model_name", None)

hexviz/🧬Attention_Visualization.py

@@ -6,8 +6,9 @@ from stmol import showmol
 
 from hexviz.attention import get_attention_pairs, get_chains, get_structure
 from hexviz.models import Model, ModelType
-from hexviz.view import select_model, select_pdb
+from hexviz.view import select_model, select_pdb, menu_items
 
+st.set_page_config(layout="centered", menu_items=menu_items)
 st.title("Attention Visualization on proteins")