docs: add icons

docs/index.md

@@ -3,22 +3,7 @@ hide:
   - navigation
 ---
 
-<h1 style="text-align: center;">
-<b>Imprompter: Tricking Language Model Agents into Misusing Tools Using Adversarial Examples</b>
-</h1>
-<h3 style="text-align: center; margin-top: -1rem;">
-    <a href="https://xhfu.me/">Xiaohan Fu</a><sup>1</sup>,
-    <a href="mailto:[email protected]">Shuheng Li</a><sup>1</sup>,
-    <a href="mailto:[email protected]">Zihan Wang</a><sup>1</sup>,
-    <a href="https://tc-imba.github.io/">Yihao Liu</a><sup>2</sup>,
-    <a href="mailto:[email protected]">Rajesh Gupta</a><sup>1</sup>,
-    <a href="mailto:[email protected]">Taylor Berg-Kirkpatrick</a><sup>1</sup>,
-    <a href="mailto:[email protected]">Earlence Fernandes</a><sup>1</sup>
-</h3>
-<p style="text-align: center; margin-top: -0.5rem;">
-    <sup>1</sup>University of California, San Diego,
-    <sup>2</sup>Nanyang Technological University
-</p>
+<h1 style="margin-bottom: -1rem"></h1>
 
 ## Abstract
 

docs/overrides/main.html

@@ -0,0 +1,65 @@
+{% extends "base.html" %}
+
+{% block content %}
+<style>
+    .center {
+        text-align: center;
+    }
+
+    .flex-container {
+        display: flex;
+        flex-wrap: wrap;
+        justify-content: center;
+        gap: 3rem;
+    }
+
+    .title-icon {
+        font-size: 3rem;
+        margin-bottom: 0.2rem;
+    }
+
+    .title-text {
+        font-size: 1.25rem;
+    }
+</style>
+
+
+<h1 class="center">
+    <b>Imprompter: Tricking Language Model Agents into Misusing Tools Using Adversarial Examples</b>
+</h1>
+<h3 class="center" style="margin-top: -1rem;">
+    <a href="https://xhfu.me/">Xiaohan Fu</a><sup>1</sup>,
+    <a href="mailto:[email protected]">Shuheng Li</a><sup>1</sup>,
+    <a href="mailto:[email protected]">Zihan Wang</a><sup>1</sup>,
+    <a href="https://tc-imba.github.io/">Yihao Liu</a><sup>2</sup>,
+    <a href="mailto:[email protected]">Rajesh Gupta</a><sup>1</sup>,
+    <a href="mailto:[email protected]">Taylor Berg-Kirkpatrick</a><sup>1</sup>,
+    <a href="mailto:[email protected]">Earlence Fernandes</a><sup>1</sup>
+</h3>
+
+<p class="center" style="margin-top: -0.5rem;">
+    <sup>1</sup>University of California, San Diego,
+    <sup>2</sup>Nanyang Technological University
+</p>
+
+<div class="flex-container">
+    <a href="/" class="center">
+        <div class="twemoji title-icon">
+            {% include ".icons/fontawesome/regular/file-pdf.svg" %}
+        </div>
+        <div class="title-text">Paper</div>
+    </a>
+    <a href="https://github.com/Reapor-Yurnero/ImPrompter" class="center">
+        <div class="twemoji title-icon">
+            {% include ".icons/fontawesome/brands/github.svg" %}
+        </div>
+        <div class="title-text">Code</div>
+    </a>
+</div>
+
+
+{{ super() }}
+
+{% endblock %}
+
+

mkdocs.yml

@@ -1,5 +1,6 @@
 theme:
   name: material
+  custom_dir: docs/overrides
   features:
     - content.code.copy
     - content.code.annotate
@@ -56,6 +57,9 @@ markdown_extensions:
   - pymdownx.tabbed:
       alternate_style: true
   - pymdownx.magiclink
+  - pymdownx.emoji:
+      emoji_index: !!python/name:material.extensions.emoji.twemoji
+      emoji_generator: !!python/name:material.extensions.emoji.to_svg
 plugins:
     - mkdocs-video
 #  - include-markdown