From 82370ffdf3d133a18ee89daceab7bf533eed94e2 Mon Sep 17 00:00:00 2001
From: xiangjinyu <xiangjinyu@fuzhi.ai>
Date: Thu, 13 Feb 2025 20:54:04 +0800
Subject: [PATCH] add paper link and github link to SPO app

---
 metagpt/ext/spo/app.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/metagpt/ext/spo/app.py b/metagpt/ext/spo/app.py
index ac7aed3d6..20895a420 100644
--- a/metagpt/ext/spo/app.py
+++ b/metagpt/ext/spo/app.py
@@ -76,7 +76,25 @@ def main():
     if "optimization_results" not in st.session_state:
         st.session_state.optimization_results = []
 
-    st.title("SPO | Self-Supervised Prompt Optimization 🤖")
+    st.markdown(
+        """
+        <div style="background-color: #f0f2f6; padding: 20px; border-radius: 10px; margin-bottom: 25px">
+            <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 10px">
+                <h1 style="margin: 0;">SPO | Self-Supervised Prompt Optimization 🤖</h1>
+            </div>
+            <div style="display: flex; gap: 20px; align-items: center">
+                <a href="https://arxiv.org/pdf/2502.06855" target="_blank" style="text-decoration: none;">
+                    <img src="https://img.shields.io/badge/Paper-PDF-red.svg" alt="Paper">
+                </a>
+                <a href="https://github.com/geekan/MetaGPT/blob/main/examples/spo/README.md" target="_blank" style="text-decoration: none;">
+                    <img src="https://img.shields.io/badge/GitHub-Repository-blue.svg" alt="GitHub">
+                </a>
+                <span style="color: #666;">A framework for self-supervised prompt optimization</span>
+            </div>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
 
     # Sidebar for configurations
     with st.sidebar: