Update app.py

app.py

@@ -11,6 +11,24 @@ STEPFUN_API_KEY = os.environ.get("STEPFUN_API_KEY", "")
 STEPFUN_BASE_URL = "https://api.stepfun.com/v1"
 MODEL_NAME = "step-3.5-flash"
 HF_CONFIG_URL = "https://huggingface.co/stepfun-ai/Step-3.5-Flash/raw/main/config.json"
+STEPFUN_LOGO_URL = "https://huggingface.co/stepfun-ai/Step-3.5-Flash/resolve/main/stepfun.svg"
+STEPFUN_LOGO_PATH = "/tmp/stepfun_logo.svg"
+
+
+def download_logo():
+    """下载 StepFun logo 到本地"""
+    if not os.path.exists(STEPFUN_LOGO_PATH):
+        try:
+            response = httpx.get(STEPFUN_LOGO_URL, timeout=10.0)
+            if response.status_code == 200:
+                with open(STEPFUN_LOGO_PATH, "wb") as f:
+                    f.write(response.content)
+        except:
+            pass
+
+
+# 启动时下载 logo
+download_logo()
 
 st.set_page_config(
     page_title="Step-3.5-Flash",
@@ -176,7 +194,7 @@ def main():
             with st.chat_message("user"):
                 st.markdown(msg["content"])
         elif msg["role"] == "assistant":
-            with st.chat_message("assistant", avatar="🚀"):
+            with st.chat_message("assistant", avatar=STEPFUN_LOGO_PATH):
                 # 思考内容用 expander
                 if msg.get("thinking"):
                     render_thinking_expander(msg["thinking"], is_streaming=False)
@@ -212,7 +230,7 @@ def main():
             st.markdown(prompt)
 
         # 助手回复
-        with st.chat_message("assistant", avatar="🚀"):
+        with st.chat_message("assistant", avatar=STEPFUN_LOGO_PATH):
             # 思考内容占位符
             thinking_placeholder = st.empty()
             # 回答内容占位符