Update pages/candlestick_patterns.py

pages/candlestick_patterns.py

@@ -1,13 +1,13 @@
 import streamlit as st
 import matplotlib.pyplot as plt
-import mplfinance as mpf
 import pandas as pd
 import numpy as np
 import yfinance as yf
 from datetime import datetime, timedelta
-from sklearn.ensemble import RandomForestClassifier
+from mplfinance.original_flavor import candlestick_ohlc
+import matplotlib.dates as mdates
 
-# ✅ Import local utility functions
+# ✅ Import local functions
 from utils.patterns import identify_patterns, calculate_technical_indicators
 from utils.predictions import predict_movement
 from utils.trading import fetch_market_data, is_market_open
@@ -28,19 +28,14 @@ status_color = "🟢" if market_open else "🔴"
 market_status = "Market Open" if market_open else "Market Closed"
 st.sidebar.write(f"{status_color} {market_status}")
 
-# ✅ **User Inputs for Stock Symbol & Timeframe**
-symbol = st.sidebar.text_input("Enter Stock Symbol", value="AAPL", help="Example: AAPL, MSFT, TSLA")
-timeframe = st.sidebar.selectbox(
-    "Select Timeframe",
-    ["30m", "1h", "2h", "4h"],
-    index=0
-)
+# ✅ **User Inputs**
+symbol = st.sidebar.text_input("Enter Stock Symbol", value="AAPL")
+timeframe = st.sidebar.selectbox("Select Timeframe", ["30m", "1h", "2h", "4h"])
 
-# ✅ **Auto-refresh Option**
+# ✅ **Auto-refresh**
 auto_refresh = st.sidebar.checkbox("Auto-refresh Data", value=True)
 if auto_refresh:
-    st.sidebar.write("🔄 Data refreshes automatically every minute")
-    st.rerun()  # ✅ Corrected (Replaces deprecated `st.experimental_rerun()`)
+    st.rerun()
 
 # 🏆 **Main Title**
 st.title("📈 Candlestick Pattern & Trading Analysis")
@@ -58,15 +53,23 @@ try:
             st.subheader(f"Candlestick Chart for {symbol}")
 
             fig, ax = plt.subplots(figsize=(12, 6))
-            mpf.plot(
-                df.tail(50),
-                type='candle',
-                style='charles',
-                title=f"{symbol} - {timeframe} Candlestick Chart",
-                ylabel='Price',
-                volume=True,
-                ax=ax
+            df_ohlc = df.copy()
+            df_ohlc['Date'] = df_ohlc.index.map(mdates.date2num)
+
+            candlestick_ohlc(
+                ax,
+                df_ohlc[['Date', 'Open', 'High', 'Low', 'Close']].values,
+                width=0.0005,  # Adjust candle width
+                colorup='g',
+                colordown='r'
             )
+
+            ax.xaxis.set_major_formatter(mdates.DateFormatter('%H:%M'))
+            plt.xlabel("Time")
+            plt.ylabel("Price")
+            plt.title(f"{symbol} - {timeframe} Candlestick Chart")
+            plt.grid(True)
+
             st.pyplot(fig)
 
             # ✅ **Pattern Analysis**