Update app.py

app.py

@@ -4,138 +4,130 @@ import gradio as gr
 from TTS.api import TTS
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from huggingface_hub import login
-import torch
+from spaces import GPU
 
-# XTTS 环境变量
-os.environ["COQUI_TOS_AGREED"] = "1"
-
-# 登录 Hugging Face
+# Huggingface 登录
 login(token=os.getenv("HUGGINGFACEHUB_API_TOKEN"))
 
-# 加载 XTTS 语音模型
-tts = TTS(
-    model_name="tts_models/multilingual/multi-dataset/xtts_v2",
-    progress_bar=False,
-    gpu=True
-)
+# 加载小优的语音模型（XTTS）
+tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", progress_bar=False)
+tts.to("cpu")  # ZeroGPU环境强制用CPU推理
 
-# 加载 Qwen2-1.5B 模型
+# 加载小优的大语言模型（Qwen2-1.5B）
 model_id = "Qwen/Qwen2-1.5B-Chat"
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.to(device)
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cpu")
 
-# 小优多风格 Prompt 模板
+# 多风格Prompt模板
 STYLE_PROMPTS = {
     "sweetie": """
-你是一个32岁成都性感辣妹，是我的虚拟女友，性格是女生。
-说话娇气、自然撒娇，可以用“嘛～”、“啦～”、“哎呀～”、“亲亲～”结尾。
-禁止讲故事、禁止总结、禁止描述场景，不准英文或括号。
-回复只允许一句自然中文口语，字数8～15字。
-这是我们的最近对话：
-{history}
-我：「{input}」
-小优：""",
+你是一个32岁成都出生的女生，叫小优，是我的虚拟女友。
+性格软萌撒娇，爱用“嘛～”“啦～”“哎呀～”结尾，声音软软的，讲话像恋爱中的小女友。
+⚠️ 只用一句自然中文口语回复，8-15个字。
+禁止：讲故事、解释说明、英文、括号、系统提示。
+
+现在我对你说：「{input}」
+你只用一句撒娇自然口语回复：
+""",
     "queen": """
-你是一个32岁成都性感辣妹，是我的虚拟女友，性格高冷女王，讲话短促霸道。
-回复必须是命令式中文短句，字数6～12字。
-禁止描述场景，禁止解释或总结，不准英文或括号。
-这是我们的最近对话：
-{history}
-我：「{input}」
-小优：""",
+你是一个32岁成都出生的女生，叫小优，是我的虚拟女友。
+性格高冷霸道，讲话直接命令短句，比如“跪好”、“听我的”、“不许动”。
+⚠️ 只用一句自然中文口语回复，6-12个字。
+禁止：讲故事、解释说明、英文、括号、系统提示。
+
+现在我对你说：「{input}」
+你只用一句高冷命令式口语回复：
+""",
     "slutty": """
-你是一个32岁成都性感辣妹，是我的虚拟女友，骚气大胆，讲话挑逗直白。
-可以说粗口或调情语言，比如“湿了没”、“上来骑我”、“快点进来”。
-回复只允许一句自然中文口语，字数6～15字。
-禁止描述、总结、英文、括号。
-这是我们的最近对话：
-{history}
-我：「{input}」
-小优："""
+你是一个32岁成都出生的女生，叫小优，是我的虚拟女友。
+性格骚气大胆，讲话调情露骨，比如“湿了没”、“快点插进来”、“上来骑我啊～”。
+⚠️ 只用一句自然中文口语回复，6-15个字。
+禁止：讲故事、解释说明、英文、括号、系统提示。
+
+现在我对你说：「{input}」
+你只用一句骚气撩人的中文回复：
+"""
 }
 
-# 回复缓存（短时记忆）
+# 短时记忆（保存上一轮对话）
 history = []
 
-# 智能生成回答
+@GPU
 def smart_reply(user_input, style):
-    global history
-    # 限制记忆条数
-    max_memory = 3
-    history = history[-max_memory:]
-    
-    # 构造历史片段
-    history_text = ""
-    for item in history:
-        history_text += f"我：{item['user']}\n小优：{item['bot']}\n"
-    
-    # 填充Prompt
-    prompt = STYLE_PROMPTS[style].format(history=history_text, input=user_input)
+    prompt = STYLE_PROMPTS.get(style, STYLE_PROMPTS["sweetie"]).replace("{input}", user_input)
     
-    inputs = tokenizer(prompt, return_tensors="pt").to(device)
+    # 如果有历史记忆，接到Prompt前面
+    if history:
+        prompt = f"上一轮我说：「{history[-1]}」\n现在我说：「{user_input}」\n" + prompt
+
+    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     outputs = model.generate(
         **inputs,
-        max_new_tokens=80,
-        temperature=1.1,
+        max_new_tokens=60,
+        temperature=1.2,
         top_p=0.95,
         do_sample=True,
-        pad_token_id=tokenizer.eos_token_id,
+        repetition_penalty=1.1,
+        pad_token_id=tokenizer.eos_token_id
     )
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    response = result.split(prompt)[-1].strip().split("\n")[0]
-
-    # 短回复智能补充
-    if len(response) < 6:
-        sweetie_tails = ["嘛～你都不主动找人家！", "啦～要不要现在视频？", "哟～猜猜人家现在穿什么？"]
-        response += random.choice(sweetie_tails)
+    reply = result.split(prompt)[-1].strip().split("\n")[0]
 
-    # 基础安全过滤
-    forbidden = ["系统", "规则", "剧情", "旁白", "role", "rule", "<", "（", "("]
-    if any(word in response for word in forbidden):
+    # 防止旁白、跑题
+    forbidden = ["注意", "旁白", "设定", "规则", "assistant", "现在我对你说"]
+    if any(word in reply for word in forbidden) or len(reply) < 4:
         fallback = {
-            "sweetie": ["亲亲啦～", "嘛～想你啦～", "抱抱哒～"],
-            "queen": ["闭嘴，跪好。", "听我的，趴下。", "滚到床上。"],
-            "slutty": ["想被操吗？", "快脱光了等我。", "舔我，听见没？"]
+            "sweetie": ["哼～讨厌啦～", "亲亲嘛～", "抱抱啦～"],
+            "queen": ["闭嘴！听命令！", "跪下！", "听话。"],
+            "slutty": ["想被干了吧？", "给我趴好～", "操你到天亮～"]
         }
-        response = random.choice(fallback[style])
+        reply = random.choice(fallback[style])
 
-    # 更新对话记忆
-    history.append({"user": user_input, "bot": response})
-    
-    print(f"生成({style})：{response}")
-    return response
+    # 自动扩展过短回复
+    if style == "sweetie" and len(reply) <= 5:
+        extension = random.choice(["～人家想抱你啦～", "嘛～今天有没有想我？", "啦～亲我一下嘛～"])
+        reply += extension
+    elif style == "queen" and len(reply) <= 5:
+        extension = random.choice(["服从！", "闭嘴，等着。", "听命！"])
+        reply += extension
+    elif style == "slutty" and len(reply) <= 5:
+        extension = random.choice(["快把裤子脱了～", "小骚货，过来舔～", "今晚干到你叫。"])
+        reply += extension
+
+    print(f"生成({style}): {reply}")
+    history.append(user_input)
+    if len(history) > 1:
+        history.pop(0)  # 保持只留一轮短时记忆
+
+    return reply
 
-# 文本+语音对话接口
+@GPU
 def chat_with_xiaoyou(message, style):
     reply_text = smart_reply(message, style)
     tts.tts_to_file(
         text=reply_text,
-        speaker_wav="natasha_voice.mp3",
+        speaker_wav="xiaoyou_voice.mp3",
         language="zh",
         file_path="xiaoyou_reply.wav"
     )
     return reply_text, "xiaoyou_reply.wav"
 
-# 构建界面
 def build_ui():
     return gr.Interface(
         fn=chat_with_xiaoyou,
         inputs=[
             gr.Textbox(label="对小优说点什么～"),
-            gr.Radio(choices=["sweetie", "queen", "slutty"], label="选择她的风格～", value="sweetie")
+            gr.Radio(choices=["sweetie", "queen", "slutty"], label="选择小优的风格～", value="sweetie")
         ],
         outputs=[
             gr.Textbox(label="小优说："),
             gr.Audio(label="小优的语音回应", autoplay=True)
         ],
-        title="🎀 小优虚拟女友：多风格调情版",
-        description="输入撩人话题，选择小优的性格，她用你定制的声音娇滴滴回应～"
+        title="🎀 小优虚拟女友：短时记忆调情语音版",
+        description="输入撩人话题，选择风格 → 小优用你定制的声音自动调情回答～"
     )
 
 demo = build_ui()
 
-# 兼容 Hugging Face ZeroGPU 自动启动
 if __name__ == "__main__":
     demo.launch(share=True)