Update app.py

app.py

@@ -116,29 +116,54 @@ def hybrid_retrieval(query, corpus_docs, faiss_index, embedding_model, k=3, kw_w
     return [doc for _, doc in scored_docs[:k]]
 
 # ========= Step 6: 聊天函数 =========
+import random
+
+# 新增：用于主动引导话题的 fallback 逻辑
+def choose_fallback_topic(user_input, knowledge_docs):
+    if len(user_input.strip()) < 5:
+        candidates = [doc.page_content for doc in knowledge_docs if "？" in doc.page_content]
+        if not candidates:
+            candidates = [doc.page_content for doc in knowledge_docs]
+        if candidates:
+            return f"{user_input}，{random.choice(candidates)}"
+    return user_input
+
+
 def chat(user_input, history):
     history = history or []
-    context_text = "\n".join([f"用户：{msg['content']}" if msg["role"] == "user" else f"sophia：{msg['content']}" for msg in history])
 
-    # 检索合并后的 corpus 和 knowledge 文档
+    # ✅ 限制历史轮数，避免太长（最后 4 轮 = 8 条 message）
+    history = history[-8:]
+
+    # ✅ 把用户输入和引导语拼起来
+    prompt_question = choose_fallback_topic(user_input, knowledge_docs)
+
+    # ✅ 格式化历史记录
+    context_text = "\n".join([
+        f"用户：{msg['content']}" if msg['role'] == "user" else f"sophia：{msg['content']}"
+        for msg in history
+    ])
+
+    # ✅ 语义+关键词混合检索，增强语境
     retrieved_docs = hybrid_retrieval(
-        query=user_input,
-        corpus_docs=all_docs,  # 使用 all_docs（包括 knowledge）进行检索
+        query=prompt_question,
+        corpus_docs=all_docs,
         faiss_index=index,
         embedding_model=embedding_model,
         k=3
     )
     retrieved_context = "\n".join([doc.page_content for doc in retrieved_docs])
 
-    # 从对话集中取风格示例
+    # ✅ 随机选几条风格示例拼到 Prompt
     example_pairs = dialog_data[:5]
     example_text = "\n".join([f"user：{pair['user']}\nsophia：{pair['sophia']}" for pair in example_pairs])
 
+    # ✅ 构造 Prompt
     prompt = prompt_template.format(
         system=system_prompt,
         examples=example_text,
         context=retrieved_context + "\n" + context_text,
-        question=user_input
+        question=prompt_question
     )
 
     try:
@@ -146,12 +171,17 @@ def chat(user_input, history):
     except Exception as e:
         reply = f"哎呀出错了：{str(e)}"
 
+    # ✅ 更新对话历史（按 OpenAI 风格）
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": reply})
+
     return history, history
 
-import gradio as gr
 
+
+import gradio as gr
+import gradio as gr
+import os
 # === 背景图与音乐 URL ===
 background_images = [
     "https://huggingface.co/spaces/Ronaldo1111/Sophia/resolve/main/family.jpg",
@@ -168,7 +198,9 @@ background_images = [
 avatar_url = "https://huggingface.co/spaces/Ronaldo1111/Sophia/resolve/main/bean.jpg"
 music1 = "https://huggingface.co/spaces/Ronaldo1111/Sophia/resolve/main/FNG.mp3"
 music2 = "https://huggingface.co/spaces/Ronaldo1111/Sophia/resolve/main/PGY.mp3"
+cake_url = "https://huggingface.co/spaces/Ronaldo1111/Sophia/resolve/main/birthday.jpg"
 
+# ==== Gradio 页面 ====
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🎀Horse And 7 Agent🎀")
     gr.Markdown("欢迎来到 **豌豆的家**，相信你也是马＋7大家庭中的一员。我能在家里拉粑粑吗！💬")
@@ -178,64 +210,64 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     state = gr.State([{"role": "assistant", "content": "勾巴，我是豌豆，你想聊啥？"}])
     btn = gr.Button("投喂")
 
-    # 插入 CSS、背景图轮播、头像动画、聊天框美化、背景音乐播放
+    # === 页面美化部分 ===
     gr.HTML(f"""
     <style>
         body {{
             margin: 0;
-            animation: backgroundCycle 30s infinite;
+            animation: backgroundCycle 60s infinite;
             background-size: cover;
             background-position: center;
             transition: background-image 1s ease-in-out;
         }}
         @keyframes backgroundCycle {{
-            0% {{ background-image: url({background_images[0]}); }}
-            10% {{ background-image: url({background_images[1]}); }}
-            20% {{ background-image: url({background_images[2]}); }}
-            30% {{ background-image: url({background_images[3]}); }}
-            40% {{ background-image: url({background_images[4]}); }}
-            50% {{ background-image: url({background_images[5]}); }}
-            60% {{ background-image: url({background_images[6]}); }}
-            70% {{ background-image: url({background_images[7]}); }}
-            80% {{ background-image: url({background_images[8]}); }}
-            90% {{ background-image: url({background_images[9]}); }}
+            {''.join([f"{i*10}% {{ background-image: url({img}); }}\n" for i, img in enumerate(background_images)])}
         }}
 
         .gr-chatbot {{
-            background: linear-gradient(135deg, #fbd3e9, #bb377d);
-            border-radius: 12px;
+            background: rgba(255, 255, 255, 0.5); /* 半透明 */
+            border-radius: 16px;
             padding: 10px;
         }}
 
         #sophia-avatar {{
             position: fixed;
-            top: 10px;
-            left: 10px;
-            width: 60px;
-            height: 60px;
+            top: 60px;
+            left: 30px;
+            width: 80px;
+            height: 80px;
             border-radius: 50%;
-            animation: bounceSpin 4s infinite;
+            z-index: 9999;
+            cursor: move;
+        }}
+
+        #birthday-cake {{
+            position: fixed;
+            bottom: 10px;
+            right: 10px;
+            width: 80px;
+            animation: bounce 2s infinite ease-in-out alternate;
             z-index: 9999;
         }}
 
-        @keyframes bounceSpin {{
-            0%, 100% {{ transform: translateY(0) rotate(0deg); }}
-            25% {{ transform: translateY(-10px) rotate(90deg); }}
-            50% {{ transform: translateY(0) rotate(180deg); }}
-            75% {{ transform: translateY(10px) rotate(270deg); }}
+        @keyframes bounce {{
+            0% {{ transform: translateY(0px); }}
+            100% {{ transform: translateY(-20px); }}
         }}
     </style>
 
     <img id="sophia-avatar" src="{avatar_url}" />
+    <img id="birthday-cake" src="{cake_url}" />
+
+    <canvas id="fireworks" style="position:fixed; top:0; left:0; width:100vw; height:100vh; pointer-events:none; z-index:999;"></canvas>
 
     <audio id="bg-music" autoplay loop>
         <source src="{music1}" type="audio/mpeg">
         <source src="{music2}" type="audio/mpeg">
-        Your browser does not support the audio tag.
     </audio>
 
     <script>
-        // 自动循环播放多首歌曲
+        // 音乐自动切换
         const musicList = ["{music1}", "{music2}"];
         let currentTrack = 0;
         const audio = document.getElementById("bg-music");
@@ -245,12 +277,46 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             audio.src = musicList[currentTrack];
             audio.play();
         }});
+
+        // 拖动头像
+        const avatar = document.getElementById("sophia-avatar");
+        avatar.onmousedown = function(e) {{
+            const offsetX = e.clientX - avatar.offsetLeft;
+            const offsetY = e.clientY - avatar.offsetTop;
+            function move(e) {{
+                avatar.style.left = (e.clientX - offsetX) + 'px';
+                avatar.style.top = (e.clientY - offsetY) + 'px';
+            }}
+            document.addEventListener('mousemove', move);
+            document.addEventListener('mouseup', () => {{
+                document.removeEventListener('mousemove', move);
+            }}, {{ once: true }});
+        }};
+
+        // 简易烟花（伪特效）
+        const canvas = document.getElementById("fireworks");
+        const ctx = canvas.getContext("2d");
+        canvas.width = window.innerWidth;
+        canvas.height = window.innerHeight;
+
+        function firework() {{
+            const x = Math.random() * canvas.width;
+            const y = Math.random() * canvas.height;
+            for (let i = 0; i < 30; i++) {{
+                setTimeout(() => {{
+                    ctx.beginPath();
+                    ctx.arc(x, y, Math.random() * 3 + 2, 0, Math.PI * 2);
+                    ctx.fillStyle = `hsl(${Math.random() * 360}, 100%, 60%)`;
+                    ctx.fill();
+                }}, i * 20);
+            }}
+        }}
+        setInterval(firework, 1500);
     </script>
     """)
-
-    # 聊天逻辑按钮触发
-    btn.click(chat, inputs=[msg, state], outputs=[chatbot, state])
-    msg.submit(chat, inputs=[msg, state], outputs=[chatbot, state])
+    # === 按钮绑定 ===
+    btn.click(chat_with_log, inputs=[msg, state], outputs=[chatbot, state])
+    msg.submit(chat_with_log, inputs=[msg, state], outputs=[chatbot, state])
 
 demo.launch()