Config

app.py

@@ -314,6 +314,7 @@ def reset_env():
 custom_css = """
 .gradio-container { font-family: 'Arial', sans-serif; }
 """
+
 with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="AI Trading Arena") as demo:
     gr.Markdown("# 🚀 AI Trading Arena\n### Interactive Stock Trading Simulator")
 
@@ -339,11 +340,10 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="AI Trading Arena")
                 init_btn = gr.Button("🚀 Load Config", variant="primary", size="lg")
 
             with gr.Accordion("💹 Trading Operations", open=True):
-                # ✅ 这里是修改点：先初始化为空
                 stock_dropdown = gr.Dropdown(
-                    choices=env.stocks if hasattr(env, "stocks") else [],
+                    choices=["S0", "S1", "S2", "S3", "S4", "S5"],
                     label="Select Stock",
-                    value=None
+                    value=100
                 )
                 action_radio = gr.Radio(choices=["Buy", "Sell"], label="Action", value="Buy")
                 amount_input = gr.Number(label="Amount (shares)", value=10, minimum=1, step=1)
@@ -364,50 +364,18 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="AI Trading Arena")
             with gr.Tab("💰 Portfolio Value"):
                 value_chart = gr.Plot(label="Total Portfolio Value")
 
-    # ✅ 修改 initialize_env，使其返回 stocks 列表用于更新 dropdown
-    def initialize_env_with_stocks(config_json=None):
-        status, portfolio, news, price_fig, value_fig = initialize_env(config_json)
-        stocks = getattr(env, "stocks", [])
-        return status, portfolio, news, price_fig, value_fig, gr.update(choices=stocks, value=stocks[0] if stocks else None)
-
     # 绑定逻辑
-    load_file_btn.click(
-        fn=load_config_from_file,
-        inputs=[config_file_dropdown],
-        outputs=[config_input]
-    )
-
-    init_btn.click(
-        fn=initialize_env_with_stocks,
-        inputs=[config_input],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart, stock_dropdown]
-    )
-
-    reset_btn.click(
-        fn=reset_env,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-
-    trade_btn.click(
-        fn=execute_trade,
-        inputs=[stock_dropdown, action_radio, amount_input],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-
-    advance_btn.click(
-        fn=advance_day,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart]
-    )
-
-    # ✅ 初始化时也更新 dropdown 选项
-    demo.load(
-        fn=initialize_env_with_stocks,
-        inputs=[],
-        outputs=[status_output, portfolio_table, news_display, price_chart, value_chart, stock_dropdown]
-    )
-
+    load_file_btn.click(fn=load_config_from_file, inputs=[config_file_dropdown], outputs=[config_input])
+    init_btn.click(fn=initialize_env, inputs=[config_input],
+                   outputs=[status_output, portfolio_table, news_display, price_chart, value_chart])
+    reset_btn.click(fn=reset_env, inputs=[], 
+                    outputs=[status_output, portfolio_table, news_display, price_chart, value_chart])
+    trade_btn.click(fn=execute_trade, inputs=[stock_dropdown, action_radio, amount_input],
+                    outputs=[status_output, portfolio_table, news_display, price_chart, value_chart])
+    advance_btn.click(fn=advance_day, inputs=[], 
+                      outputs=[status_output, portfolio_table, news_display, price_chart, value_chart])
+    demo.load(fn=initialize_env, inputs=[], 
+              outputs=[status_output, portfolio_table, news_display, price_chart, value_chart])
 
 if __name__ == "__main__":
     demo.launch()