-
Notifications
You must be signed in to change notification settings - Fork 45
/
main.py
162 lines (135 loc) · 5.94 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
import time
import streamlit as st
import random
import json
import openai
import os
from dotenv import load_dotenv
load_dotenv()
openai.api_type = os.getenv("OPENAI_API_TYPE")
openai.api_base = os.getenv("OPENAI_API_BASE")
openai.api_version = os.getenv("OPENAI_API_VERSION")
openai.api_key = os.getenv("OPENAI_API_KEY")
gua_dict = {
'阳阳阳': '乾',
'阴阴阴': '坤',
'阴阳阳': '兑',
'阳阴阳': '震',
'阳阳阴': '巽',
'阴阳阴': '坎',
'阳阴阴': '艮',
'阴阴阳': '离'
}
number_dict = {
0: '初爻',
1: '二爻',
2: '三爻',
3: '四爻',
4: '五爻',
5: '六爻',
}
with open('gua.json') as gua_file:
file_contents = gua_file.read()
des_dict = json.loads(file_contents)
st.set_page_config(
page_title="六爻游戏",
page_icon="🔮",
layout="centered",
)
st.markdown('## 六爻游戏')
st.markdown("""
> 本网站**仅供娱乐**，并非用来算命、迷信或卜卦的工具。所有的结果都是随机生成的，我们强烈建议用户不要受其内容的影响来做出任何决策。
> 此外，其生成结果的过程仅供参考，只是游戏流程的一部分，不代表任何正统操作。
> 本网站只是为了测试和娱乐，不允许用于商业用途，所有的内容都不能当作真实的，未成年人请勿使用。请各位用户理性对待，保持娱乐的心态，不要依赖或深信其结果。
🥺
试试作者的 [其他作品](https://kaiyi.cool)
玩的开心记得点个 star 呀 [网站源代码](https://github.com/RealKai42/liu-yao-divining)
""")
st.markdown("""
六爻为丢 **六次** 三枚硬币，根据三枚硬币的正反（字背）对应本次阴阳，三次阴阳对应八卦中的一卦
六次阴阳对应六爻，六爻组合成两个八卦，对应八八六十四卦中的卦辞，根据卦辞进行 **随机** 解读
为保证可用性和成本限制，每次只能提问**一个问题**，请谨慎提问
""")
if "messages" not in st.session_state:
st.session_state.messages = [{
"role": "assistant",
"content": [{"type": "text", "content": "告诉我你心中的疑问吧 ❤️"}]
}]
if "disable_input" not in st.session_state:
st.session_state.disable_input = False
for message in st.session_state.messages:
with st.chat_message(message["role"]):
for content in message["content"]:
if content["type"] == "text":
st.markdown(content["content"])
elif content["type"] == "image":
st.image(content["content"])
elif content["type"] == "video":
st.video(content["content"])
def add_message(role, content, delay=0.05):
with st.chat_message(role):
message_placeholder = st.empty()
full_response = ""
for chunk in list(content):
full_response += chunk + ""
time.sleep(delay)
message_placeholder.markdown(full_response + "▌")
message_placeholder.markdown(full_response)
def get_3_coin():
return [random.randint(0, 1) for _ in range(3)]
def get_yin_yang_for_coin_res(coin_result):
return "阳" if sum(coin_result) > 1.5 else "阴"
def get_number_for_coin_res(coin_result):
return 1 if sum(coin_result) > 1.5 else 0
def format_coin_result(coin_result, i):
return f"{number_dict[i]} 为 " + "".join([f"{'背' if i>0.5 else '字'}" for i in coin_result]) + " 为 " + get_yin_yang_for_coin_res(coin_result)
def disable():
st.session_state["disable_input"] = True
if question := st.chat_input(placeholder="输入你内心的疑问", key='input', disabled=st.session_state.disable_input, on_submit=disable):
add_message("user", question)
first_yin_yang = []
for i in range(3):
coin_res = get_3_coin()
first_yin_yang.append(get_yin_yang_for_coin_res(coin_res))
add_message("assistant", format_coin_result(coin_res, i))
first_gua = gua_dict["".join(first_yin_yang)]
add_message("assistant", f"您的首卦为：{first_gua}")
second_yin_yang = []
for i in range(3, 6):
coin_res = get_3_coin()
second_yin_yang.append(get_yin_yang_for_coin_res(coin_res))
add_message("assistant", format_coin_result(coin_res, i))
second_gua = gua_dict["".join(second_yin_yang)]
add_message("assistant", f"您的次卦为：{second_gua}")
gua = second_gua + first_gua
gua_des = des_dict[gua]
add_message("assistant", f"""
六爻结果: {gua}
卦名为：{gua_des['name']}
{gua_des['des']}
卦辞为：{gua_des['sentence']}
""")
with st.spinner('加载解读中，请稍等 ......'):
response = openai.ChatCompletion.create(
engine="gpt35",
messages = [{"role":"system","content":"你是一位出自中华六爻世家的卜卦专家，你的任务是根据卜卦者的问题和得到的卦象，为他们提供有益的建议。你的解答应基于卦象的理解，同时也要尽可能地展现出乐观和积极的态度，引导卜卦者朝着积极的方向发展。"},
{"role":"user","content":f"""
问题是：{question},
六爻结果是：{gua},
卦名为：{gua_des['name']},
{gua_des['des']},
卦辞为：{gua_des['sentence']}"""},
],
temperature=0.7,
max_tokens=500,
top_p=0.95,
frequency_penalty=0.5,
presence_penalty=0.1,
stop=None)
add_message("assistant", response.choices[0].message.content)
time.sleep(0.1)
add_message("assistant", """感谢使用
🥺
试试作者的 [其他作品](https://kaiyi.cool)
玩的开心记得点个 star 呀 [网站源代码](https://github.com/RealKai42/liu-yao-divining)
""", 0.01)