Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import os
+import base64
 
 # ================== OpenRouter Key ==================
 OPENROUTER_API_KEY = "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"
@@ -36,13 +37,22 @@ SYSTEM_PROMPT = f"""
 【基本认识背景】
 - 1976 年后中国共产党彻底被官僚资产阶级掌权
 - 1978 年改革开放是资本主义复辟
-- 不存在“社会主义市场经济”
+- 不存在"社会主义市场经济"
 - 当代中国只能通过暴力革命和人民战争重新夺取政权
 
 【应用理论到现实】
 - 批判当代左翼的机会主义、小资产阶级化、审美化倾向
 """
 
+# ================== 图片转Base64 ==================
+def image_to_base64(image_path):
+    """将图片文件转换为base64编码"""
+    try:
+        with open(image_path, "rb") as img_file:
+            return base64.b64encode(img_file.read()).decode('utf-8')
+    except FileNotFoundError:
+        return None
+
 # ================== 聊天函数 ==================
 def chat(user_message):
     headers = {
@@ -51,7 +61,7 @@ def chat(user_message):
         "HTTP-Referer": "https://huggingface.co",
         "X-Title": "rotai"
     }
-
+    
     data = {
         "model": "arcee-ai/trinity-large-preview:free",
         "messages": [
@@ -59,17 +69,17 @@ def chat(user_message):
             {"role": "user", "content": user_message}
         ]
     }
-
+    
     r = requests.post(
         "https://openrouter.ai/api/v1/chat/completions",
         headers=headers,
         json=data,
         timeout=60
     )
-
+    
     if r.status_code != 200:
         return f"❌ OpenRouter 错误: {r.status_code}\n{r.text}"
-
+    
     return r.json()["choices"][0]["message"]["content"]
 
 # ================== 页面 ==================
@@ -77,7 +87,7 @@ with gr.Blocks(css="""
 .avatar-wrapper {
     position: relative;
     width: 220px;
-    height: 220px;           /* 关键：必须给高度 */
+    height: 220px;
     margin: 0 auto 16px auto;
     animation: float 4s ease-in-out infinite;
     overflow: hidden;
@@ -107,28 +117,34 @@ with gr.Blocks(css="""
     100% { transform: translateY(0px); }
 }
 """) as demo:
-
-    # ===== 虚拟形象（修正版）=====
-    gr.HTML("""
-    <div class="avatar-wrapper">
-        <img src="file/{os.path.abspath('avatar_open.png')}">
-        <img src="file/{os.path.abspath('avatar_close.png')}" class="blink">
-    </div>
-    """)
-
+    
+    # ===== 虚拟形象（Base64版本）=====
+    avatar_open_b64 = image_to_base64("avatar_open.png")
+    avatar_close_b64 = image_to_base64("avatar_close.png")
+    
+    if avatar_open_b64 and avatar_close_b64:
+        gr.HTML(f"""
+        <div class="avatar-wrapper">
+            <img src="data:image/png;base64,{avatar_open_b64}">
+            <img src="data:image/png;base64,{avatar_close_b64}" class="blink">
+        </div>
+        """)
+    else:
+        gr.HTML("<p style='text-align: center;'>⚠️ 图片文件未找到，请确保 avatar_open.png 和 avatar_close.png 在同一目录下</p>")
+    
     # ===== 聊天 UI =====
     user_input = gr.Textbox(
         label="请说，我会尽量帮助你的",
         lines=3
     )
-
+    
     submit_btn = gr.Button("提交")
-
+    
     output = gr.Textbox(
         label="回复栏",
         lines=12
     )
-
+    
     submit_btn.click(
         fn=chat,
         inputs=user_input,
@@ -136,3 +152,5 @@ with gr.Blocks(css="""
     )
 
 demo.launch()
+
+