Spaces:
Running
Running
import json | |
import logging | |
import gradio as gr | |
from theme import Seafoam | |
from utils.mes_player_model import Player | |
from utils.mes_achievements import Achievement | |
from utils.mes_player_activity_model import PlayerActivity | |
from utils.utils import ( | |
get_content, | |
render_player_data, | |
save_latest_player_data, | |
render_finished, | |
) | |
from utils.completion_reward import CompletionReward | |
from utils.completion_reward_utils import ( | |
set_player_name, | |
set_player_selected_character, | |
create_certificate, | |
complete_reward, | |
check_is_in_completion_reward, | |
get_llm_response_once, | |
) | |
seafoam = Seafoam() | |
def get_player_info(player_backend_user_id): | |
try: | |
with open("latest_player_data.json", "r", encoding="utf-8") as file: | |
player_info = json.load(file) | |
except FileNotFoundError: | |
logging.error("No player data found. Loading new player data.") | |
save_latest_player_data() | |
with open("latest_player_data.json", "r", encoding="utf-8") as file: | |
player_info = json.load(file) | |
with open("latest_player_data.json", "r") as f: | |
datas = json.load(f) | |
processed_datas = {} | |
for k, v in datas.items(): | |
log = v["adventure_logs"] | |
filtered_log = [record for record in log if not record.startswith("恭喜")] | |
processed_log = [record.rsplit(",", 1)[0] for record in filtered_log] | |
concat_log = " ".join(processed_log) | |
processed_datas[k] = concat_log | |
temp_log = """你開始了星空探險隊的冒險! 你在離家 2 天後,找到了迷失已久的星際繪圖師,並在他的引導下進入了下一個未知的領域。 在完成了 41 關學習內容後,你獲得了遙遠星球上的神秘遺產。 在某天,你在冰晶極地找到了一顆具有神秘力量的寒冰寶石。 你在奇特的花園中找到了一種新的能源。 在雲端星球中,你遇見了一位雲之精靈,並從她那裡獲得了一個神祕的能量水球。 132 天之後,你們獲得了一個神秘的地圖,這將引導你們找到隱藏的神秘能量源。\n你們在一個古老的地下洞穴中找到了一個巨大的藍色能量晶體,這將極大地援助你們的旅程。 你在混亂的黑洞中找到了一片完好無損的石碑,上面記載著光束寶石的由來和使用方式。 你用寶石的力量召喚傳說中的先知,獲得了狐狸貓族的力量, 向黑洞發動了終極一擊。 你在繼續活動了 7 天後,你的行為感動了星際,你獲得了一個可以轉換成任何形象的能量護盾。 你在一場巨大的黑洞爆炸後,成功找回了去往家鄉的方向。 連日交戰後達成了學習 209 個內容的紀錄,獲得了星系長老的讚賞,並贈與你們神秘的光護符。 你成功與神秘的月亮寶石建立連結,從而獲得月亮之力。 你在遙遠的星海之上發現了一顆寂靜的星體,星體的中心藏有一個帶著神秘能量的寶藏。 你在神祕的狐狸貓族祖先墳場找到了一塊古老的碑文,裡面記載了如何使用和提煉能量的方法。 在經歷了若干日月後,你終於在 229 號洞穴找到了第一塊結晶。 你們在探索神秘洞穴的過程中,發現了一個神秘寶箱,裡面有 229 個神秘道具。 你們在冬季島上找到了冰晶葉,並獲得了 678 點的能量。 你剛剛結束了 14 天的劍橋學堂,你們獲得了 726 點的能量供給。 你剛剛結束了 14 天的劍橋學堂,你們獲得了 726 點的能量供給。 你在星空探險隊的冒險期間和狐貍貓一起度過了聖誕佳節 🎄 你從超空間黑洞脫離時發現了一串神秘的密碼,這個密碼包含著黑洞能量的謎題。 你們在一個神秘的星球上探險,在那裡你們發現了新的能量源。 你們在一片未知的領域中尋找到了狐狸貓族的古老遺蹟,並獲得了 780 的能量點數。 你收集了 836 個痕跡之星,並用風暴龍的力量打開了神秘之門。 你收集了 941 個痕跡之星,並用風暴龍的力量打開了神秘之門。 你結束了 2023 與狐貍貓的星空探險!""" | |
processed_datas["alvin.lau@junyiacademy.org"] = temp_log | |
processed_datas[ | |
"http://googleid.junyiacademy.org/115084976189396533674" | |
] = temp_log | |
processed_datas[ | |
"http://googleid.junyiacademy.org/106428943548495187296" | |
] = temp_log | |
with open("processed_adventure_logs.json", "w") as f: | |
json.dump(processed_datas, f) | |
if player_backend_user_id in player_info: | |
return player_info[player_backend_user_id] | |
else: | |
logging.info( | |
f"No data found for player ID {player_backend_user_id}. Initializing new player data." | |
) | |
new_player = Player( | |
player_backend_user_id=player_backend_user_id, | |
init=True, | |
available_achievements=Achievement.get_available_achievements(), | |
) | |
return new_player.to_dict() | |
def create_new_player_activity(): | |
return PlayerActivity() | |
def get_player_logs(player_backend_user_id): | |
with open("processed_adventure_logs.json") as file: | |
player_logs = json.load(file) | |
try: | |
player_logs = player_logs[player_backend_user_id] | |
return player_logs | |
except KeyError: | |
logging.error( | |
f"Player {player_backend_user_id} not found in processed_player_adventure_logs.json" | |
) | |
def init_reward(): | |
return CompletionReward() | |
# start of gradio interface | |
with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo: | |
player_info = gr.State() | |
player_logs = gr.State() | |
completion_reward = gr.State(init_reward) | |
player_activity_tracker = gr.State(create_new_player_activity) | |
with gr.Tab("個人化戰報", elem_id="personal_report"): | |
with gr.Row(): | |
with gr.Column( | |
scale=1, | |
elem_classes="gallery_container", | |
): | |
pet_description = gr.Markdown("# 夥伴", elem_id="pet_avatar_description") | |
pet_gallery = gr.Gallery( | |
[], | |
label="夥伴", | |
preview=False, | |
elem_id="pet_gallery", | |
columns=30, | |
height=200, | |
) | |
badge_description = gr.Markdown( | |
"# 徽章", elem_id="badge_avatar_description" | |
) | |
badge_gallery = gr.Gallery( | |
[], | |
label="徽章", | |
preview=False, | |
elem_id="badge_gallery", | |
columns=30, | |
height=200, | |
) | |
with gr.Column(scale=1, elem_id="player_avatar_container"): | |
avatar = gr.Image( | |
"avatar/blank_avatar.png", | |
elem_id="player_avatar", | |
) | |
avatar_description = gr.Markdown( | |
"# 光束守護者", elem_id="player_avatar_description" | |
) | |
with gr.Column(scale=1): | |
adventure_description = gr.Markdown( | |
"# 冒險階段", elem_id="adventure_description" | |
) | |
adventure = gr.Slider( | |
value=0, | |
show_label=False, | |
interactive=False, | |
elem_id="adventure_slider", | |
info="", | |
) | |
achievements_description = gr.Markdown( | |
"# 達成成就", elem_id="achievements_description" | |
) | |
achievements = gr.HighlightedText( | |
value=[], | |
elem_classes="achievements", | |
color_map={ | |
"完成": "green", | |
"未完成": "red", | |
}, | |
) | |
with gr.Row(): | |
html = ( | |
"<div style='max-width:100%; max-height:360px; overflow:auto'>" | |
+ get_content("htmls/adventure_blank.html") | |
) | |
adventure_log = gr.HTML( | |
html, label="Adventure Log", elem_id="adventure_log" | |
) | |
# handling player info | |
with gr.Row(): | |
player_backend_id = gr.Textbox( | |
"", elem_id="player_backend_id", visible=False | |
) | |
player_info_query_btn = gr.Button( | |
"Query", elem_id="trigger_button", visible=False | |
) | |
pull_newest_player_data = gr.Textbox("", visible=False) | |
update_status = gr.Textbox("", visible=False) | |
with gr.Tab("完賽獎勵", elem_id="completion_reward"): | |
with gr.Row(): | |
start_make_reward = gr.Button( | |
"開始製作完賽獎勵!", visible=False, elem_id="start_make_reward" | |
) | |
with gr.Row(): | |
not_participate = gr.Markdown( | |
"# 同學並未參與 2023 的星空探險隊唷!", visible=False, elem_id="not_participate" | |
) | |
with gr.Row(): | |
not_start = gr.Markdown( | |
"# 完賽獎勵還沒有開放申請,但是就快了,請敬請期待!", visible=False, elem_id="not_start" | |
) | |
with gr.Row(): | |
already_issued = gr.Image(visible=False) | |
with gr.Row(): | |
player_name_title = gr.Markdown( | |
"# 選擇玩家暱稱", visible=False, elem_id="player_name_title" | |
) | |
with gr.Row(): | |
player_name = gr.Textbox( | |
label="玩家暱稱", | |
info="請輸入想要用在完賽獎勵上的玩家暱稱,上限為 10 字,且像是 🤩🌟🎉😊 等 emoji 也會無法顯示唷;最後,獎勵發送後無法更改也無法補發,還請同學們謹慎填寫。", | |
interactive=True, | |
elem_id="player_name", | |
visible=False, | |
placeholder="請在這裡輸入暱稱", | |
) | |
with gr.Row(): | |
player_name_too_long = gr.Markdown( | |
"# 暱稱過長,請重新輸入", visible=False, elem_id="player_name_too_long" | |
) | |
with gr.Row(): | |
confirm_player_name = gr.Button( | |
"確認暱稱", elem_id="confirm_player_name", visible=False | |
) | |
cancel_player_name = gr.Button( | |
"取消", elem_id="cancel_player_name", visible=False | |
) | |
with gr.Row(): | |
player_name_next_step = gr.Button( | |
"下一步", visible=False, elem_id="player_name_next_step" | |
) | |
with gr.Row(): | |
story_title = gr.Markdown("# 選擇冒險故事", visible=False, elem_id="story_title") | |
with gr.Row(): | |
story_description = gr.Markdown( | |
"有五位星際夥伴來為你撰寫專屬於你的冒險故事,每位星際夥伴都會得到同學冒險週記的內容,並寫成一段故事,請同學選擇自己最喜歡的一段故事,這段故事將會被用來印在獎狀上<br><br>小叮嚀:請同學選擇喜歡的故事內容,而不是星際夥伴唷!星際夥伴可是不會和同學們一起冒險的!最後,撰寫故事的過程大概需要幾分鐘的時間,請同學耐心等待", | |
visible=False, | |
elem_id="story_description", | |
) | |
with gr.Row(): | |
openai_description = gr.Markdown( | |
"# 露米娜", elem_id="openai_description", visible=False | |
) | |
aws_description = gr.Markdown( | |
"# 索拉拉", elem_id="aws_description", visible=False | |
) | |
google_description = gr.Markdown( | |
"# 薇丹特", elem_id="google_description", visible=False | |
) | |
mtk_description = gr.Markdown( | |
"# 蔚藍", elem_id="mtk_description", visible=False | |
) | |
ntu_description = gr.Markdown( | |
"# 紅寶石", elem_id="ntu_description", visible=False | |
) | |
with gr.Row(): | |
openai_img = gr.Image( | |
"medias/lumina.png", | |
visible=False, | |
elem_id="openai_img", | |
interactive=False, | |
show_download_button=False, | |
) | |
aws_img = gr.Image( | |
"medias/solara.png", | |
visible=False, | |
elem_id="aws_img", | |
interactive=False, | |
show_download_button=False, | |
) | |
google_img = gr.Image( | |
"medias/verdant.png", | |
visible=False, | |
elem_id="google_img", | |
interactive=False, | |
show_download_button=False, | |
) | |
mtk_img = gr.Image( | |
"medias/azure.png", | |
visible=False, | |
elem_id="mtk_img", | |
interactive=False, | |
show_download_button=False, | |
) | |
ntu_img = gr.Image( | |
"medias/ruby.png", | |
visible=False, | |
elem_id="ntu_img", | |
interactive=False, | |
show_download_button=False, | |
) | |
with gr.Row(): | |
start_generate_story = gr.Button( | |
"開始寫作故事", visible=False, elem_id="start_generate_story" | |
) | |
weaving = gr.Button( | |
"星際夥伴努力撰寫故事中...", | |
visible=False, | |
elem_id="weaving", | |
size="lg", | |
interactive=False, | |
) | |
with gr.Row(): | |
bot1 = gr.Chatbot(visible=False, elem_id="bot1") | |
bot2 = gr.Chatbot(visible=False, elem_id="bot2") | |
bot3 = gr.Chatbot(visible=False, elem_id="bot3") | |
bot4 = gr.Chatbot(visible=False, elem_id="bot4") | |
bot5 = gr.Chatbot(visible=False, elem_id="bot5") | |
with gr.Row(): | |
select_story = gr.Radio( | |
["露米娜", "索拉拉", "薇丹特", "蔚藍", "紅寶石"], | |
interactive=True, | |
label="選擇故事", | |
visible=False, | |
elem_id="select_story", | |
) | |
with gr.Row(): | |
confirm_story = gr.Button("確認故事", visible=False, elem_id="confirm_story") | |
cancel_story = gr.Button("取消", visible=False, elem_id="cancel_story") | |
with gr.Row(): | |
start_generate_certificate = gr.Button( | |
"開始製作完賽獎勵!", visible=False, elem_id="start_generate_certificate" | |
) | |
processing = gr.Button( | |
"製作中...", | |
visible=False, | |
elem_id="processing", | |
size="lg", | |
interactive=False, | |
) | |
complete = gr.Markdown("# 完成!", visible=False, elem_id="complete") | |
with gr.Row(): | |
reward_result = gr.Image(visible=False) | |
# actions when player login | |
# define args | |
send_player_login_info = dict( | |
fn=render_finished, | |
inputs=[player_activity_tracker, player_info], | |
outputs=None, | |
queue=False, | |
) | |
check_is_in_completion_reward_args = dict( | |
fn=check_is_in_completion_reward, | |
inputs=player_backend_id, | |
outputs=[ | |
player_name_title, | |
player_name, | |
confirm_player_name, | |
not_participate, | |
not_start, | |
already_issued, | |
], | |
queue=False, | |
) | |
player_info_query_btn.click( | |
get_player_info, player_backend_id, player_info, queue=False | |
).then(get_player_logs, player_backend_id, player_logs, queue=False).then( | |
**check_is_in_completion_reward_args | |
).then( | |
render_player_data, | |
player_info, | |
[avatar, pet_gallery, badge_gallery, adventure_log, achievements, adventure], | |
queue=False, | |
).then( | |
**send_player_login_info | |
) | |
pull_newest_player_data.submit( | |
save_latest_player_data, | |
None, | |
update_status, | |
api_name="pull_newest_player_data", | |
) | |
def create_visibility_updates(visible, count): | |
return tuple(gr.update(visible=visible) for _ in range(count)) | |
start_make_reward.click( | |
check_is_in_completion_reward, | |
player_backend_id, | |
[ | |
start_make_reward, | |
player_name_title, | |
player_name, | |
confirm_player_name, | |
not_participate, | |
not_start, | |
already_issued, | |
], | |
queue=False, | |
) | |
set_player_name_args = dict( | |
fn=set_player_name, | |
inputs=[completion_reward, player_name, player_backend_id], | |
outputs=None, | |
queue=False, | |
) | |
def check_plyer_name_length(player_name): | |
if len(player_name) > 10: | |
return ( | |
gr.update(visible=True), | |
gr.update(visible=False), | |
gr.update(visible=True), | |
) | |
else: | |
return ( | |
gr.update(visible=True), | |
gr.update(visible=True), | |
gr.update(visible=False), | |
) | |
confirm_player_name.click( | |
lambda: (gr.update(interactive=False), gr.update(visible=False)), | |
None, | |
[player_name, confirm_player_name], | |
queue=False, | |
).then( | |
check_plyer_name_length, | |
player_name, | |
[cancel_player_name, player_name_next_step, player_name_too_long], | |
queue=False, | |
).then( | |
**set_player_name_args | |
) | |
cancel_player_name.click( | |
lambda: ( | |
gr.update(interactive=True), | |
gr.update(visible=True), | |
gr.update(visible=False), | |
), | |
None, | |
[player_name, confirm_player_name, player_name_too_long], | |
queue=False, | |
).then( | |
lambda: create_visibility_updates(False, 2), | |
None, | |
[cancel_player_name, player_name_next_step], | |
queue=False, | |
) | |
player_name_next_step.click( | |
lambda: create_visibility_updates(False, 5), | |
None, | |
[ | |
player_name, | |
player_name_next_step, | |
confirm_player_name, | |
player_name_title, | |
cancel_player_name, | |
], | |
queue=False, | |
).then( | |
lambda: create_visibility_updates(True, 13), | |
None, | |
[ | |
openai_img, | |
aws_img, | |
google_img, | |
mtk_img, | |
ntu_img, | |
story_title, | |
story_description, | |
openai_description, | |
aws_description, | |
google_description, | |
mtk_description, | |
ntu_description, | |
start_generate_story, | |
], | |
queue=False, | |
) | |
get_first_llm_response_args = dict( | |
fn=get_llm_response_once, | |
inputs=[completion_reward, player_logs], | |
outputs=bot1, | |
queue=False, | |
) | |
get_second_llm_response_args = dict( | |
fn=get_llm_response_once, | |
inputs=[completion_reward, player_logs], | |
outputs=bot2, | |
queue=False, | |
) | |
get_third_llm_response_args = dict( | |
fn=get_llm_response_once, | |
inputs=[completion_reward, player_logs], | |
outputs=bot3, | |
queue=False, | |
) | |
get_fourth_llm_response_args = dict( | |
fn=get_llm_response_once, | |
inputs=[completion_reward, player_logs], | |
outputs=bot4, | |
queue=False, | |
) | |
get_fifth_llm_response_args = dict( | |
fn=get_llm_response_once, | |
inputs=[completion_reward, player_logs], | |
outputs=bot5, | |
queue=False, | |
) | |
start_generate_story.click( | |
lambda: gr.update(visible=False), None, start_generate_story, queue=False | |
).then( | |
lambda: create_visibility_updates(True, 6), | |
None, | |
[bot1, bot2, bot3, bot4, bot5, weaving], | |
queue=False, | |
).then(lambda: [(None, "露米娜正埋頭撰寫故事中...請耐心等候")], None, bot1, queue=False).then( | |
**get_first_llm_response_args | |
).then(lambda: [(None, "索拉拉正埋頭撰寫故事中...請耐心等候")], None, bot2, queue=False).then( | |
**get_second_llm_response_args | |
).then(lambda: [(None, "薇丹特正埋頭撰寫故事中...請耐心等候")], None, bot3, queue=False).then( | |
**get_third_llm_response_args | |
).then(lambda: [(None, "蔚藍正埋頭撰寫故事中...請耐心等候")], None, bot4, queue=False).then( | |
**get_fourth_llm_response_args | |
).then(lambda: [(None, "紅寶石正埋頭撰寫故事中...請耐心等候")], None, bot5, queue=False).then( | |
**get_fifth_llm_response_args | |
).then( | |
lambda: gr.update(visible=True), None, [select_story], queue=False | |
).then( | |
lambda: gr.update(visible=False), None, [weaving], queue=False | |
) | |
select_story.select( | |
lambda: gr.update(visible=True), None, confirm_story, queue=False | |
) | |
set_player_selected_character_args = dict( | |
fn=set_player_selected_character, | |
inputs=[completion_reward, select_story], | |
outputs=None, | |
queue=False, | |
) | |
confirm_story.click( | |
lambda: gr.update(interactive=False), None, [select_story], queue=False | |
).then(lambda: gr.update(visible=False), None, [confirm_story], queue=False).then( | |
lambda: (gr.update(visible=True), gr.update(visible=True)), | |
None, | |
[start_generate_certificate, cancel_story], | |
queue=False, | |
).then( | |
**set_player_selected_character_args | |
) | |
cancel_story.click( | |
lambda: gr.update(interactive=True), None, [select_story], queue=False | |
).then(lambda: gr.update(visible=False), None, [cancel_story], queue=False).then( | |
lambda: (gr.update(visible=False), gr.update(visible=True)), | |
None, | |
[start_generate_certificate, confirm_story], | |
queue=False, | |
) | |
create_certificate_args = dict( | |
fn=create_certificate, | |
inputs=[completion_reward], | |
outputs=reward_result, | |
queue=False, | |
) | |
complete_reward_args = dict( | |
fn=complete_reward, | |
inputs=[completion_reward], | |
outputs=None, | |
queue=False, | |
) | |
start_generate_certificate.click( | |
lambda: create_visibility_updates(False, 21), | |
None, | |
[ | |
openai_img, | |
aws_img, | |
google_img, | |
mtk_img, | |
ntu_img, | |
story_title, | |
story_description, | |
openai_description, | |
aws_description, | |
google_description, | |
mtk_description, | |
ntu_description, | |
bot1, | |
bot2, | |
bot3, | |
bot4, | |
bot5, | |
select_story, | |
processing, | |
cancel_story, | |
start_generate_certificate, | |
], | |
queue=False, | |
).then(lambda: gr.update(visible=True), None, [processing], queue=False).then( | |
**create_certificate_args | |
).then( | |
lambda: (gr.update(visible=True), gr.update(visible=False)), | |
None, | |
[complete, processing], | |
queue=False, | |
).then( | |
**complete_reward_args | |
) | |
if __name__ == "__main__": | |
demo.launch() | |