Spaces:
Runtime error
Runtime error
| import zipfile | |
| import gradio as gr | |
| from ChatHaruhi import ChatHaruhi | |
| import wget | |
| import os | |
| import copy | |
| NAME_DICT = {'汤师爷': 'tangshiye', '慕容复': 'murongfu', '李云龙': 'liyunlong', 'Luna': 'Luna', '王多鱼': 'wangduoyu', | |
| 'Ron': 'Ron', '鸠摩智': 'jiumozhi', 'Snape': 'Snape', | |
| '凉宫春日': 'haruhi', 'Malfoy': 'Malfoy', '虚竹': 'xuzhu', '萧峰': 'xiaofeng', '段誉': 'duanyu', | |
| 'Hermione': 'Hermione', 'Dumbledore': 'Dumbledore', '王语嫣': 'wangyuyan', | |
| 'Harry': 'Harry', 'McGonagall': 'McGonagall', '白展堂': 'baizhantang', '佟湘玉': 'tongxiangyu', | |
| '郭芙蓉': 'guofurong', '旅行者': 'wanderer', '钟离': 'zhongli', | |
| '胡桃': 'hutao', 'Sheldon': 'Sheldon', 'Raj': 'Raj', 'Penny': 'Penny', '韦小宝': 'weixiaobao', | |
| '乔峰': 'qiaofeng', '神里绫华': 'ayaka', '雷电将军': 'raidenShogun', '于谦': 'yuqian'} | |
| os.makedirs("characters_zip", exist_ok=True) | |
| os.makedirs("characters", exist_ok=True) | |
| ai_roles_obj = {} | |
| for ai_role_en in NAME_DICT.values(): | |
| file_url = f"https://github.com/LC1332/Haruhi-2-Dev/raw/main/data/character_in_zip/{ai_role_en}.zip" | |
| os.makedirs(f"characters/{ai_role_en}", exist_ok=True) | |
| if f"{ai_role_en}.zip" not in os.listdir(f"characters_zip"): | |
| destination_file = f"characters_zip/{ai_role_en}.zip" | |
| wget.download(file_url, destination_file) | |
| destination_folder = f"characters/{ai_role_en}" | |
| with zipfile.ZipFile(destination_file, 'r') as zip_ref: | |
| zip_ref.extractall(destination_folder) | |
| db_folder = f"./characters/{ai_role_en}/content/{ai_role_en}" | |
| system_prompt = f"./characters/{ai_role_en}/content/system_prompt.txt" | |
| ai_roles_obj[ai_role_en] = ChatHaruhi( | |
| system_prompt=system_prompt, | |
| llm="BaiChuan2GPT", | |
| story_db=db_folder, | |
| verbose=True | |
| ) | |
| async def get_response(user_role, user_text, ai_role, chatbot): | |
| role_en = NAME_DICT[ai_role] | |
| ai_roles_obj[role_en].dialogue_history = copy.deepcopy(chatbot) | |
| response = ai_roles_obj[role_en].chat(role=user_role, text=user_text) | |
| user_msg = user_role + ':「' + user_text + '」' | |
| latest_msg = (user_msg, response) | |
| print(latest_msg) | |
| chatbot.append(latest_msg) | |
| return chatbot | |
| async def respond(user_role, user_text, ai_role, chatbot): | |
| return await get_response(user_role, user_text, ai_role, chatbot), None | |
| with gr.Blocks() as demo: | |
| gr.Markdown( | |
| """ | |
| # Chat凉宫春日 ChatHaruhi | |
| ## Reviving Anime Character in Reality via Large Language Model | |
| ChatHaruhi2.0的 BaiChuan-13B 版本 Demo Implemented by [Sirly](https://github.com/SirlyDreamer) | |
| 更多信息见项目GitHub链接 [https://github.com/LC1332/Chat-Haruhi-Suzumiya](https://github.com/LC1332/Chat-Haruhi-Suzumiya) | |
| 如果觉得有趣请拜托为我们点上Star. | |
| If you find it interesting, please be kind enough to give us a Star. | |
| user_role 为用户扮演的人物 请尽量设置为与剧情相关的人物 且不要与主角同名 | |
| """ | |
| ) | |
| with gr.Row(): | |
| chatbot = gr.Chatbot() | |
| with gr.Row(): | |
| user_role = gr.Textbox(label="user_role") | |
| user_text = gr.Textbox(label="user_text") | |
| with gr.Row(): | |
| submit = gr.Button("Submit") | |
| clean = gr.ClearButton(value="Clear") | |
| ai_role = gr.Radio(['汤师爷', '慕容复', '李云龙', | |
| 'Luna', '王多鱼', 'Ron', '鸠摩智', | |
| 'Snape', '凉宫春日', 'Malfoy', '虚竹', | |
| '萧峰', '段誉', 'Hermione', 'Dumbledore', | |
| '王语嫣', | |
| 'Harry', 'McGonagall', | |
| '白展堂', '佟湘玉', '郭芙蓉', | |
| '旅行者', '钟离', '胡桃', | |
| 'Sheldon', 'Raj', 'Penny', | |
| '韦小宝', '乔峰', '神里绫华', | |
| '雷电将军', '于谦'], label="characters", value='凉宫春日') | |
| ai_role.change(lambda x: (None, None, []), ai_role, [user_role, user_text, chatbot]) | |
| user_text.submit(fn=respond, inputs=[user_role, user_text, ai_role, chatbot], outputs=[chatbot, user_text]) | |
| submit.click(fn=respond, inputs=[user_role, user_text, ai_role, chatbot], outputs=[chatbot, user_text]) | |
| clean.click(lambda x, y, z: (None, None, []), [user_role, user_text, chatbot], [user_role, user_text, chatbot]) | |
| demo.launch(debug=True, share=False, server_name='0.0.0.0') |