import gradio as gr import pandas as pd import numpy as np from CharacterStatistics import CharacterStatistics from WorldSimulation import WorldSimulation # 创建一个世界模拟对象 simulation = None num_rounds = 0 def initialize_world(world_spiritual_energy=1000000, resources = 100000, resources_threshold = 10): global simulation simulation = WorldSimulation(world_spiritual_energy=world_spiritual_energy, resources=resources, resources_threshold=resources_threshold) return f"世界初始化成功：\n\n世界灵气能量：{world_spiritual_energy}" def add_custom_character(name, gender, special_constitution, spiritual_roots, clan=None): special_constitution = [1 if i in special_constitution else 0 for i in range(4)] spiritual_roots = [1 if i in spiritual_roots else 0 for i in range(5)] simulation.add_custom_character(name, gender, special_constitution, spiritual_roots, clan) return str(simulation.characters[-1]) def start_simulation(c1, c2, c3, c4, r1, r2, r3, r4, r5, initial_population=3000): special_constitution_ratio = [c1, c2, c3, c4] spiritual_roots_ratio = [r1, r2, r3, r4, r5] initial_population = int(initial_population) simulation.create_population(special_constitution_ratio=special_constitution_ratio, spiritual_roots_ratio=spiritual_roots_ratio, initial_population=initial_population) return f"世界人口数量达到{len(simulation.characters)}，可以运行模拟了。" def run_simulation(rounds): global num_rounds num_rounds = rounds if simulation is None: return "请先初始化世界" return simulation.run_simulation(num_rounds) def find_character_by_clan(name): characters = simulation.find_characters_by_clan(name) if len(characters) == 0: return f"没有找到名字为{name}的宗族" else: return "\n\n".join([str(c) for c in characters]) def get_character_info(name): characters = simulation.find_characters_by_name(name) if len(characters) == 0: return f"没有找到名字为{name}的角色" else: return "\n\n".join([str(c) for c in characters]) def get_dead_character_info(character_index): if character_index < len(simulation.dead_characters): return str(simulation.dead_characters[character_index]) else: return f"没有第{character_index}个死亡角色，目前共有{len(simulation.dead_characters)}个死亡角色" def get_world_stats(): stats = CharacterStatistics(simulation.characters) return stats.summarize_markdown() def plot(): y = simulation.history_population df = pd.DataFrame({"x": np.arange(len(y)), "y": y, "spi_count": simulation.history_spi_energy}) return df with gr.Blocks() as demo: # 初始化世界 gr.Markdown("## 初始化世界") with gr.Row(): world_spiritual_energy_input = gr.Number(label="世界灵气能量", value=1000000) initialize_button = gr.Button("初始化世界") output_text = gr.Markdown(label="初始化结果") initialize_button.click(fn=initialize_world, inputs=[ world_spiritual_energy_input], outputs=output_text) # 添加角色 with gr.Accordion("添加角色 (可选)", open=False): gr.Markdown("每次添加一个，添加完可以继续添加。") name_input = gr.Textbox(label="姓名") gender_input = gr.Radio(["男", "女"], label="性别") clan_input = gr.Textbox(label="宗族") special_constitution_input = gr.CheckboxGroup(["战斗体质", "合欢体质", "灵龟体质", "蜉蝣体质"], label="特殊体质", type="index") spiritual_roots_input = gr.CheckboxGroup(["金", "木", "水", "火", "土"], label="灵根", type="index") add_character_button = gr.Button("添加角色") new_character_info_output = gr.Textbox(label="新角色信息") add_character_button.click(fn=add_custom_character, inputs=[name_input, gender_input, special_constitution_input, spiritual_roots_input, clan_input], outputs=new_character_info_output) # 生成初始人口 gr.Markdown("## 生成初始人口") initial_population_input = gr.Number(label="初始人口", value=3000, precision=0) # 特殊体质 gr.Markdown("特殊体质比例") with gr.Row(): special_constitution_ratio_input = [ gr.Number(label="战斗体质", minimum=0, maximum=1, value=0.001), gr.Number(label="合欢体质", minimum=0, maximum=1, value=0.001), gr.Number(label="灵龟体质", minimum=0, maximum=1, value=0.001), gr.Number(label="蜉蝣体质", minimum=0, maximum=1, value=0.001)] # 灵根 gr.Markdown("灵根比例") with gr.Row(): spiritual_roots_ratio_input = [ gr.Number(label="金", minimum=0, maximum=1, value=0.04), gr.Number(label="木", minimum=0, maximum=1, value=0.04), gr.Number(label="水", minimum=0, maximum=1, value=0.04), gr.Number(label="火", minimum=0, maximum=1, value=0.04), gr.Number(label="土", minimum=0, maximum=1, value=0.04)] start_simulation_button = gr.Button("生成初始人口") output_text = gr.Markdown("") start_simulation_button.click(fn=start_simulation, inputs=[*special_constitution_ratio_input, *spiritual_roots_ratio_input, initial_population_input], outputs=output_text) # 运行模拟 gr.Markdown("## 运行模拟\n\n输入轮数，运行模拟，可以多次运行。") with gr.Row(): rounds_input = gr.Number(label="轮数", precision=0, value=1) run_simulation_button = gr.Button("运行模拟") simulation_result_output = gr.Textbox(label="模拟结果") run_simulation_button.click(fn=run_simulation, inputs=[rounds_input], outputs=[simulation_result_output]) # 查看统计信息 get_stats_button = gr.Button("查看统计信息") with gr.Accordion("详细文字统计信息", open=False): with gr.Box(): world_stats_output = gr.Markdown("") plot1 = gr.Plot() with gr.Row(): plot2 = gr.Plot() plot3 = gr.Plot() plot4 = gr.Plot() get_stats_button.click(fn=get_world_stats, inputs=[], outputs=[world_stats_output, plot1, plot2, plot3, plot4]) # 查看角色信息 with gr.Accordion("查看角色信息", open=False): with gr.Row(): character_info_output = gr.Textbox(label="角色信息", info="输入角色名字") get_character_info_button = gr.Button("查看角色信息") character_info_output_display = gr.Textbox(label="角色信息") get_character_info_button.click(fn=get_character_info, inputs=[character_info_output], outputs=[character_info_output_display]) # 查看宗族信息 with gr.Accordion("查看宗族信息", open=False): with gr.Row(): clan_name_input = gr.Textbox(label="宗族名字", info="输入宗族名字") get_clan_characters_button = gr.Button("查看宗族角色") characters_output_display = gr.Textbox(label="角色信息") get_clan_characters_button.click(fn=find_character_by_clan, inputs=[clan_name_input], outputs=[characters_output_display]) with gr.Accordion("查看死亡角色", open=False): with gr.Row(): dead_character_info_output = gr.Number(label="死亡角色信息", info="输入死亡角色序号（按死亡顺序排列）", precision=0) get_dead_character_info_button = gr.Button("查看死亡角色信息") dead_character_info_output_display = gr.Textbox(label="死亡角色信息") get_dead_character_info_button.click(fn=get_dead_character_info, inputs=[dead_character_info_output], outputs=[dead_character_info_output_display]) # 可视化 btn = gr.Button(value="查看人口图表") plot_output = gr.LinePlot( x="x", y="y", title="人口随时间变化图", x_title="年份", y_title="人口数量", overlay_point=True, tooltip=["x", "y"], width=350, height=300, ) btn2 = gr.Button(value="查看灵气图表") plot_output2 = gr.LinePlot( x="x", y="spi_count", title="灵气随时间变化图", x_title="年份", y_title="灵气数量", overlay_point=True, tooltip=["x", "spi_count"], width=350, height=300, ) btn.click(plot, outputs=plot_output) btn2.click(plot, outputs=plot_output2) demo.queue().launch(debug=True)