Update src/streamlit_app.py

src/streamlit_app.py

@@ -387,8 +387,9 @@ def main():
     st.set_page_config(page_title="Jacobian Scopes Demo", page_icon="🔍", layout="centered")
     st.title("🔍 Jacobian Scopes Demo")
     st.markdown(
-        "<b>Interactive demonstrations for Jacobian Scopes: a suite of token-level attribution tools for LLMs.</b> "
-        '<b>Check out our <a href="https://arxiv.org/abs/2601.16407">paper</a> to learn more.</b>',
+        'Interactive demonstrations for <a href="https://arxiv.org/abs/2601.16407"><b>Jacobian Scopes: token-level causal attributions in LLMs</b></a>. <br>'
+        'Github Repo: <a href="https://github.com/AntonioLiu97/JacobianScopes">https://github.com/AntonioLiu97/JacobianScopes</a>',
+        
         unsafe_allow_html=True,
     )
     # Keep scope columns on one line (Streamlit stacks them below ~640px by default)
@@ -613,10 +614,25 @@ def main():
         plt.close(fig_pred)
 
     st.divider()
-    st.caption(
-        "© 2026 Toni Jianbang Liu. All rights reserved. "
-        "Cite us at https://arxiv.org/abs/2601.16407"
-    )
+    with st.expander("Citation Information", expanded=True):
+        st.markdown("**Jacobian Scopes Demo © 2026 Toni Jianbang Liu.**")
+        st.markdown("If you use this demo in your work, please cite:")
+        st.markdown(
+            "Liu, T. J., Zadeoğlu, B., Boullé, N., Sarfati, R., & Earls, C. J. (2026). "
+            "*Jacobian Scopes: token-level causal attributions in LLMs.* arXiv preprint arXiv:2601.16407."
+        )
+        st.markdown("**BibTeX:**")
+        st.code(
+            """@misc{liu2026jacobianscopestokenlevelcausal,
+      title={Jacobian Scopes: token-level causal attributions in LLMs}, 
+      author={Toni J. B. Liu and Baran Zadeoğlu and Nicolas Boullé and Raphaël Sarfati and Christopher J. Earls},
+      year={2026},
+      eprint={2601.16407},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2601.16407}, }""",
+            language=None,
+        )
 
 
 if __name__ == "__main__":