Update charts.py

charts.py

@@ -3,21 +3,28 @@ import matplotlib.dates as mdates
 from io import BytesIO
 
 def plot_candlestick(df):
-    fig, ax = plt.subplots(figsize=(8, 4))
-    df = df.tail(50)
+    try:
+        df = df.tail(50).copy()
+        fig, ax = plt.subplots(figsize=(8, 4))
 
-    for idx, row in df.iterrows():
-        color = 'green' if row['close'] >= row['open'] else 'red'
-        ax.plot([row['timestamp'], row['timestamp']], [row['low'], row['high']], color='black')
-        ax.plot([row['timestamp'], row['timestamp']], [row['open'], row['close']], color=color, linewidth=5)
+        df['timestamp'] = pd.to_datetime(df['timestamp'])
 
-    ax.xaxis.set_major_formatter(mdates.DateFormatter('%H:%M'))
-    plt.xticks(rotation=45)
-    plt.title('Latest Candlestick Chart')
-    plt.tight_layout()
+        for idx, row in df.iterrows():
+            color = 'green' if row['close'] >= row['open'] else 'red'
+            ax.plot([row['timestamp'], row['timestamp']], [row['low'], row['high']], color='black')
+            ax.plot([row['timestamp'], row['timestamp']], [row['open'], row['close']], color=color, linewidth=5)
 
-    buf = BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
-    plt.close(fig)
-    return buf
+        ax.xaxis.set_major_formatter(mdates.DateFormatter('%H:%M'))
+        ax.xaxis.set_major_locator(mdates.AutoDateLocator())
+        plt.xticks(rotation=45)
+        plt.title('Latest Candlestick Chart')
+        plt.tight_layout()
+
+        buf = BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        plt.close(fig)
+        return buf
+    except Exception as e:
+        print("Chart error:", e)
+        raise e