Update jsinvn.py with OpenAI client integration and eliminated player flow

This commit is contained in:
st2411020210 2026-01-09 09:18:48 +08:00
parent 8a3939534d
commit c6572eecee

517
jsinvn.py
View File

@ -1,5 +1,15 @@
import streamlit as st
import random
import os
import requests
from dotenv import load_dotenv
# 加载环境变量
load_dotenv()
# 获取Deepseek API配置
DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
DEEPSEEK_BASE_URL = os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com")
# 页面配置
st.set_page_config(
@ -63,250 +73,45 @@ if "runoff" not in st.session_state:
st.session_state.runoff = False
if "tied_players" not in st.session_state:
st.session_state.tied_players = []
if "max_rounds" not in st.session_state:
st.session_state.max_rounds = 4
if "runoff_speeches" not in st.session_state:
st.session_state.runoff_speeches = []
if "player_runoff_speech" not in st.session_state:
st.session_state.player_runoff_speech = ""
if "player_runoff_speech_saved" not in st.session_state:
st.session_state.player_runoff_speech_saved = False
# 词语描述库
word_descriptions = {
"苹果": [
"这是一种常见的东西",
"有特定的颜色",
"圆形的,口感不错",
"可以直接食用,也可以加工",
"富含营养",
"通常生长在高处",
"有外皮和内部结构",
"是一种健康的选择",
"在中国很常见",
"有些品种味道比较特别"
],
"橘子": [
"这是一种可以食用的东西",
"有特定的颜色,味道有点特别",
"圆形的,有外皮",
"内部结构是一瓣一瓣的",
"富含营养",
"通常生长在温暖的地方",
"需要剥皮后食用",
"有不同的品种",
"汁水很多",
"在某些季节比较常见"
],
"电脑": [
"这是一种电子设备",
"可以用来做很多事情",
"有屏幕和输入设备",
"需要电源才能使用",
"可以连接网络",
"有主机和显示部分",
"可以运行各种程序",
"现在很多家庭都有",
"有不同的类型",
"需要定期维护"
],
"手机": [
"这是一种便携式电子设备",
"可以用来沟通和获取信息",
"触摸操作,便于携带",
"需要充电才能使用",
"可以安装各种应用",
"有拍照功能",
"现在几乎人人都有",
"有不同的品牌和型号",
"可以用来娱乐",
"需要特定卡片才能通话"
],
"篮球": [
"这是一种球类用品",
"圆形的,有特定材质",
"需要用手操作",
"有特定的颜色",
"在特定的场地上使用",
"有不同的规格",
"可以用来进行比赛",
"需要充气才能使用",
"很多年轻人喜欢",
"有专业的赛事"
],
"足球": [
"这是一种球类用品",
"圆形的,有特定的颜色搭配",
"需要用脚操作",
"在特定的场地上使用",
"可以用来进行比赛",
"需要充气才能使用",
"是世界上很受欢迎的活动",
"有不同的规格",
"很多人喜欢观看相关比赛",
"有专业的赛事"
],
"牛奶": [
"这是一种饮品",
"白色的,有营养",
"来自动物",
"可以直接饮用",
"富含蛋白质和钙",
"有不同的品牌",
"可以作为早餐的一部分",
"需要特定条件保存",
"有不同的类型",
"很多人每天都会喝"
],
"豆浆": [
"这是一种饮品",
"乳白色的,有特殊香味",
"用植物制作",
"可以直接饮用",
"富含植物蛋白",
"有不同的口味",
"可以作为早餐的一部分",
"是传统饮品",
"适合特定人群",
"很多早餐店都有"
],
"米饭": [
"这是一种主食",
"白色的,颗粒状",
"用特定食材制作",
"很多人经常吃",
"可以搭配各种食物",
"需要加水烹饪",
"有不同的品种",
"可以有不同的做法",
"吃了会有饱腹感",
"很多地方的人都吃"
],
"面条": [
"这是一种主食",
"长条状的,需要烹饪",
"用特定食材制作",
"可以有各种口味",
"有不同的形状",
"需要水煮",
"可以有不同的做法",
"是传统食物",
"烹饪时间不长",
"很多人喜欢吃"
],
"": [
"这是一种动物",
"毛茸茸的,会发出特定的声音",
"喜欢抓小东西",
"可以作为宠物",
"有不同的品种和颜色",
"喜欢休息",
"比较独立",
"会自己清洁",
"有爪子和尾巴",
"很多人喜欢养"
],
"": [
"这是一种动物",
"毛茸茸的,会发出特定的声音",
"可以作为宠物",
"很忠诚",
"有不同的品种和大小",
"喜欢活动和玩耍",
"需要定期外出",
"可以训练各种技能",
"有爪子和尾巴",
"被很多人视为朋友"
],
"夏天": [
"这是一个季节",
"天气比较热",
"有很多新鲜的东西",
"适合在水上进行活动",
"白天的时间比较长",
"需要穿轻薄的衣服",
"可能会有雷雨天气",
"学生们都有假期",
"适合吃冷的东西",
"阳光比较强烈"
],
"冬天": [
"这是一个季节",
"天气比较冷",
"可能会有白色的东西从天上掉下来",
"需要穿厚厚的衣服",
"白天的时间比较短",
"适合在雪地上活动",
"很多节日在这个时候",
"需要注意保暖",
"水面可能会变硬",
"适合喝热的饮料"
],
"可乐": [
"这是一种饮料",
"有特定的颜色,有气泡",
"味道甜甜的",
"含有咖啡因",
"可以解渴",
"有不同的品牌",
"可以加冰饮用",
"很多人喜欢喝",
"适合搭配食物",
"开瓶后会有气体冒出"
],
"雪碧": [
"这是一种饮料",
"透明的,有气泡",
"味道甜甜的",
"有柠檬味",
"可以解渴",
"有不同的品牌",
"可以加冰饮用",
"很多人喜欢喝",
"适合搭配食物",
"开瓶后会有气体冒出"
],
"手机": [
"这是一种便携式电子设备",
"可以用来沟通和获取信息",
"触摸操作,便于携带",
"需要充电才能使用",
"可以安装各种应用",
"有拍照功能",
"现在几乎人人都有",
"有不同的品牌和型号",
"可以用来娱乐",
"需要特定卡片才能通话"
],
"平板": [
"这是一种便携式电子设备",
"介于两种电子设备之间",
"触摸操作,比手机大",
"需要充电才能使用",
"可以用来娱乐和工作",
"有拍照功能",
"可以安装各种应用",
"适合观看视频",
"比电脑轻便",
"有不同的品牌和型号"
],
"老师": [
"这是一种职业",
"在特定的场所工作",
"教授知识",
"需要准备课程",
"有不同的专业领域",
"需要批改作业",
"对他人有指导作用",
"需要有耐心",
"很多人尊敬",
"可以影响他人的成长"
],
"学生": [
"这是一种身份",
"在特定的场所学习",
"需要上课和完成作业",
"有不同的阶段",
"可以学到各种知识",
"需要参加考试",
"有同学和老师",
"可以参加各种活动",
"是未来的希望",
"很多人都很努力"
]
}
# 调用Deepseek API生成AI发言
def call_deepseek_api(prompt):
try:
url = f"{DEEPSEEK_BASE_URL}/chat/completions"
headers = {
"Authorization": f"Bearer {DEEPSEEK_API_KEY}",
"Content-Type": "application/json"
}
data = {
"model": "deepseek-chat",
"messages": [
{
"role": "system",
"content": "你是一个谁是卧底游戏的玩家,需要根据你拿到的词语生成自然的发言,不能直接说出词语本身,发言要符合当前轮数的特点。"
},
{
"role": "user",
"content": prompt
}
],
"max_tokens": 100,
"temperature": 0.8
}
response = requests.post(url, headers=headers, json=data, timeout=10)
response.raise_for_status()
result = response.json()
return result["choices"][0]["message"]["content"].strip()
except Exception as e:
st.error(f"API调用失败: {str(e)}")
return None
# 获取AI发言
def get_ai_speech(player_index, round_num, used_speeches):
@ -314,71 +119,89 @@ def get_ai_speech(player_index, round_num, used_speeches):
player_word = st.session_state.player_words[player_index]
num_players = st.session_state.num_players
max_attempts = 100
max_attempts = 5
attempt = 0
while attempt < max_attempts:
# 按轮数递进的描述逻辑
# 每轮选择不同深度的描述
if player_word in word_descriptions:
all_descriptions = word_descriptions[player_word]
# 根据轮数选择描述数量,轮数越多描述越多
max_descriptions = min(round_num, len(all_descriptions))
# 随机选择描述,确保多样性
if len(all_descriptions) >= max_descriptions:
selected_descriptions = random.sample(all_descriptions, max_descriptions)
else:
selected_descriptions = all_descriptions[:max_descriptions]
else:
# 如果词语不在描述库中,使用通用描述
base_descriptions = [
f"这是一个{len(player_word)}字的词语",
"这个词很常见",
"和日常生活相关",
"大家应该都知道"
]
max_descriptions = min(round_num, len(base_descriptions))
selected_descriptions = random.sample(base_descriptions, max_descriptions)
# 构建发言
description_part = " ".join(selected_descriptions)
# 第一轮:基础描述
# 构建API提示词
if round_num == 1:
openings = ["我抽到的词", "我的词", "这个词", "我拿到的词", "这个东西"]
speech = f"{random.choice(openings)}{description_part}"
# 第二轮:基础描述 + 简单分析
# 第一轮:基础描述,不要太详细
prompt = f"你在玩谁是卧底游戏,你拿到的词语是 '{player_word}'。请生成第一轮的发言,只做基础描述,不要直接说出词语,发言要自然简洁。"
if is_spy:
prompt += " 你是卧底,但不要暴露自己的身份。"
elif round_num == 2:
analysis_phrases = [
"这个东西大家应该都接触过",
"日常生活中经常能看到",
"很多地方都能找到",
"比较常见的一种",
"大家应该都不陌生",
"随处可见的一种"
]
speech = f"{description_part}{random.choice(analysis_phrases)}"
# 第三轮及以后:基础描述 + 更多细节 + 怀疑
# 第二轮:更详细的描述,加一些分析
prompt = f"你在玩谁是卧底游戏,你拿到的词语是 '{player_word}'。请生成第二轮的发言,在第一轮的基础上增加一些细节分析,不要直接说出词语,发言要自然。"
if is_spy:
prompt += " 你是卧底,但不要暴露自己的身份。"
else:
detail_phrases = [
"仔细想想,它还有一些特点",
"其实它还有其他方面",
"从不同角度看,它有更多特征",
"再补充一点关于它的信息",
"进一步说,它还有",
"另外,它还有一个特点"
]
suspicion_phrases = [
f"我发现玩家{random.randint(1, num_players)}的描述好像不太对",
f"玩家{random.randint(1, num_players)}的发言有点可疑",
"对比大家的发言,有些人的描述和我的不太一样",
"根据这些特点,我觉得有人在隐藏什么",
"这些特征应该很明显,不符合的人可能有问题",
"我注意到有些人的描述和这些特征不符"
]
speech = f"{description_part}{random.choice(detail_phrases)}{random.choice(suspicion_phrases)}"
# 第三轮及以后:详细描述 + 合理怀疑
suspect_player = random.randint(1, num_players) if num_players > 1 else 1
prompt = f"你在玩谁是卧底游戏,你拿到的词语是 '{player_word}'。请生成第{round_num}轮的发言,包括详细描述和对其他玩家的合理怀疑(比如玩家{suspect_player}),不要直接说出词语,发言要自然。"
if is_spy:
prompt += " 你是卧底,要隐藏自己的身份,同时可以适当引导怀疑其他玩家。"
# 调用API生成发言
speech = call_deepseek_api(prompt)
# 如果API调用失败或返回空使用传统方式生成
if not speech:
# 备用方案:使用传统方式生成发言
if player_word in word_descriptions:
all_descriptions = word_descriptions[player_word]
max_descriptions = min(round_num, len(all_descriptions))
if len(all_descriptions) >= max_descriptions:
selected_descriptions = random.sample(all_descriptions, max_descriptions)
else:
selected_descriptions = all_descriptions[:max_descriptions]
else:
base_descriptions = [
f"这是一个{len(player_word)}字的词语",
"这个词很常见",
"和日常生活相关",
"大家应该都知道"
]
max_descriptions = min(round_num, len(base_descriptions))
selected_descriptions = random.sample(base_descriptions, max_descriptions)
description_part = " ".join(selected_descriptions)
if round_num == 1:
openings = ["我抽到的词", "我的词", "这个词", "我拿到的词", "这个东西"]
speech = f"{random.choice(openings)}{description_part}"
elif round_num == 2:
analysis_phrases = [
"这个东西大家应该都接触过",
"日常生活中经常能看到",
"很多地方都能找到",
"比较常见的一种",
"大家应该都不陌生",
"随处可见的一种"
]
speech = f"{description_part}{random.choice(analysis_phrases)}"
else:
detail_phrases = [
"仔细想想,它还有一些特点",
"其实它还有其他方面",
"从不同角度看,它有更多特征",
"再补充一点关于它的信息",
"进一步说,它还有",
"另外,它还有一个特点"
]
suspicion_phrases = [
f"我发现玩家{random.randint(1, num_players)}的描述好像不太对",
f"玩家{random.randint(1, num_players)}的发言有点可疑",
"对比大家的发言,有些人的描述和我的不太一样",
"根据这些特点,我觉得有人在隐藏什么",
"这些特征应该很明显,不符合的人可能有问题",
"我注意到有些人的描述和这些特征不符"
]
speech = f"{description_part}{random.choice(detail_phrases)}{random.choice(suspicion_phrases)}"
# 确保发言不直接包含词语本身
if player_word in speech:
# 如果直接说出了词语,尝试替换或修改
speech = speech.replace(player_word, "这个词")
# 检查发言是否唯一
if speech not in used_speeches:
@ -463,6 +286,56 @@ elif st.session_state.game_state == "playing":
for name in tied_player_names:
st.write(f"- {name}")
# 生成AI玩家第二轮发言
current_runoff_speeches = []
# 检查是否已经为当前第二轮投票生成了发言
runoff_speeches = st.session_state.runoff_speeches
if not runoff_speeches:
# 为当前第二轮投票创建一个空的已使用发言集合
used_speeches = set()
# 为每个AI玩家生成发言
for i in range(st.session_state.num_players):
if i != player_index and i not in st.session_state.eliminated_players:
# 生成唯一的发言,使用当前轮数+0.5表示第二轮投票
speech = get_ai_speech(i, st.session_state.current_round + 0.5, used_speeches)
# 将生成的发言添加到已使用集合中
used_speeches.add(speech)
current_runoff_speeches.append({
'player': i + 1,
'speech': speech,
'type': 'ai'
})
# 保存当前第二轮投票的发言
st.session_state.runoff_speeches = current_runoff_speeches
else:
# 使用已有的发言
current_runoff_speeches = runoff_speeches
# 显示AI玩家第二轮发言
st.subheader("第二轮发言")
for speech in current_runoff_speeches:
st.write(f"**{st.session_state.player_names[speech['player']-1]}**{speech['speech']}")
# 用户第二轮发言输入
if player_index not in st.session_state.eliminated_players:
st.write(f"\n### 你的第二轮发言({st.session_state.player_names[player_index]}")
st.session_state.player_runoff_speech = st.text_area(
"请输入你的第二轮发言",
value=st.session_state.player_runoff_speech,
height=100,
placeholder="在这里输入你的第二轮发言..."
)
# 保存用户第二轮发言
if st.button("确认第二轮发言"):
if st.session_state.player_runoff_speech.strip():
st.session_state.player_runoff_speech_saved = True
st.success("第二轮发言已保存!")
else:
st.error("请输入第二轮发言内容!")
# 投票输入(只能在平票玩家中选择)
vote_name = st.selectbox(
"选择要投票的玩家",
@ -473,9 +346,13 @@ elif st.session_state.game_state == "playing":
# 提交第二轮投票
if st.button("提交第二轮投票"):
# 模拟所有玩家的第二轮投票
st.session_state.votes = {}
st.session_state.votes[vote] = 1
# 检查用户是否已完成第二轮发言
if player_index not in st.session_state.eliminated_players and not st.session_state.player_runoff_speech_saved:
st.error("请先完成第二轮发言!")
else:
# 模拟所有玩家的第二轮投票
st.session_state.votes = {}
st.session_state.votes[vote] = 1
# 模拟AI玩家的第二轮投票
for ai_player in [i for i in range(st.session_state.num_players) if i not in st.session_state.eliminated_players]:
@ -516,16 +393,25 @@ elif st.session_state.game_state == "playing":
st.session_state.game_state = "ended"
st.success(st.session_state.game_result)
else:
# 进入下一轮
st.write("\n游戏继续,进入下一轮!")
st.session_state.current_round += 1
st.session_state.votes = {}
st.session_state.player_speech = ""
st.session_state.player_speech_saved = False
# 检查是否超过最大轮数
if st.session_state.current_round >= st.session_state.max_rounds:
st.session_state.game_result = "游戏结束已达到最大投票次数4次"
st.session_state.game_state = "ended"
st.success(st.session_state.game_result)
else:
# 进入下一轮
st.write("\n游戏继续,进入下一轮!")
st.session_state.current_round += 1
st.session_state.votes = {}
st.session_state.player_speech = ""
st.session_state.player_speech_saved = False
# 重置 runoff 状态
# 重置 runoff 状态和第二轮发言相关变量
st.session_state.runoff = False
st.session_state.tied_players = []
st.session_state.runoff_speeches = []
st.session_state.player_runoff_speech = ""
st.session_state.player_runoff_speech_saved = False
# 等待用户确认后再刷新
if st.button("进入下一轮"):
@ -714,6 +600,9 @@ elif st.session_state.game_state == "ended":
st.session_state.speeches = []
st.session_state.player_speech = ""
st.session_state.player_speech_saved = False
st.session_state.runoff_speeches = []
st.session_state.player_runoff_speech = ""
st.session_state.player_runoff_speech_saved = False
st.rerun()
# 游戏规则
@ -724,4 +613,6 @@ with st.sidebar:
st.write("3. 玩家通过讨论判断谁是卧底")
st.write("4. 每轮结束后进行投票,得票最多的玩家被淘汰")
st.write("5. 如果淘汰的是卧底,平民胜利")
st.write("6. 如果最后只剩下卧底,卧底胜利")
st.write("6. 如果最后只剩下卧底,卧底胜利")
st.write("7. 游戏最多进行4轮投票超过次数则游戏结束")