Config

app.py

@@ -5,6 +5,8 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import matplotlib
 import os
+import plotly.graph_objects as go
+import pandas as pd
 matplotlib.use('Agg')
 
 class TradeArenaEnv_Deterministic:
@@ -238,35 +240,80 @@ def create_news_display(obs):
 #     ax.grid(True, alpha=0.3)
 #     return fig
 
+# def create_price_chart():
+#     """Create individual price chart for each stock"""
+#     if len(history) <= 1:
+#         fig, axs = plt.subplots(1, 1, figsize=(10, 6))
+#         axs.text(0.5, 0.5, 'Trade to see price history', 
+#                  ha='center', va='center', fontsize=14, color='gray')
+#         axs.axis('off')
+#         return fig
+
+#     df = pd.DataFrame(history)
+#     num_stocks = len(env.stocks)
+#     fig, axs = plt.subplots(num_stocks, 1, figsize=(10, 4*num_stocks), sharex=True)
+    
+#     # 如果只有一个股票，axs不是数组，需要处理
+#     if num_stocks == 1:
+#         axs = [axs]
+
+#     colors = ['#3b82f6', '#10b981', '#f59e0b', '#ef4444', '#8b5cf6']
+    
+#     for i, stock in enumerate(env.stocks):
+#         ax = axs[i]
+#         ax.plot(df['day'], df[stock], marker='o', linewidth=2, color=colors[i % len(colors)], label=stock)
+#         ax.set_ylabel(f'{stock} ($)')
+#         ax.set_title(f'{stock} Price History')
+#         ax.legend(loc='best', framealpha=0.8)
+#         ax.grid(True, alpha=0.3)
+    
+#     axs[-1].set_xlabel('Day')
+#     plt.tight_layout()
+#     return fig
+
+
 def create_price_chart():
-    """Create individual price chart for each stock"""
+    """Create stock price chart using Plotly"""
     if len(history) <= 1:
-        fig, axs = plt.subplots(1, 1, figsize=(10, 6))
-        axs.text(0.5, 0.5, 'Trade to see price history', 
-                 ha='center', va='center', fontsize=14, color='gray')
-        axs.axis('off')
+        # 没有交易历史时，返回空白图
+        fig = go.Figure()
+        fig.add_annotation(
+            text="Trade to see price history",
+            xref="paper", yref="paper",
+            showarrow=False,
+            font=dict(size=16, color="gray")
+        )
+        fig.update_layout(
+            xaxis=dict(visible=False),
+            yaxis=dict(visible=False),
+            template="plotly_white",
+            height=400
+        )
         return fig
 
     df = pd.DataFrame(history)
-    num_stocks = len(env.stocks)
-    fig, axs = plt.subplots(num_stocks, 1, figsize=(10, 4*num_stocks), sharex=True)
-    
-    # 如果只有一个股票，axs不是数组，需要处理
-    if num_stocks == 1:
-        axs = [axs]
-
+    fig = go.Figure()
     colors = ['#3b82f6', '#10b981', '#f59e0b', '#ef4444', '#8b5cf6']
-    
+
     for i, stock in enumerate(env.stocks):
-        ax = axs[i]
-        ax.plot(df['day'], df[stock], marker='o', linewidth=2, color=colors[i % len(colors)], label=stock)
-        ax.set_ylabel(f'{stock} ($)')
-        ax.set_title(f'{stock} Price History')
-        ax.legend(loc='best', framealpha=0.8)
-        ax.grid(True, alpha=0.3)
-    
-    axs[-1].set_xlabel('Day')
-    plt.tight_layout()
+        fig.add_trace(go.Scatter(
+            x=df['day'],
+            y=df[stock],
+            mode='lines+markers',
+            name=stock,
+            line=dict(color=colors[i % len(colors)], width=2),
+            marker=dict(size=6)
+        ))
+
+    fig.update_layout(
+        title="Stock Price History",
+        xaxis_title="Day",
+        yaxis_title="Price ($)",
+        template="plotly_white",
+        legend=dict(title="Stocks", orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1),
+        height=400 + 50 * len(env.stocks)
+    )
+
     return fig
 
 

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 numpy
 pandas
-matplotlib
+matplotlib
+plotly