Update app.py

app.py

@@ -79,20 +79,28 @@ def get_market_minutes():
 # TRAIN MODEL
 @app.get('/model/train/{symbol}/{p_step}')
 def train(symbol: str, p_step: int):
-    time_step = 60
+    import pytz
+    import yfinance as yf
+
+    predict_steps = p_step
+    time_steps = 60
     training_period = '5d'
-    epochs = 10
+    epochs = 206
 
     tz = pytz.timezone('Asia/Kolkata')
     now = datetime.now(tz)
     today = now.date()
 
-    data = yf.download(tickers=symbol, interval='1m', period=training_period, progress=False)
-    data = data.tz_convert('Asia/Kolkata')
+    data = yf.download(tickers=symbol, interval='5m', period=training_period, progress=False)
+    if data.empty:
+        return {"error": "No data received from Yahoo Finance. Try another symbol or interval."}
 
+    data = data.tz_convert('Asia/Kolkata')
     data['Date'] = data.index.date
     available_dates = sorted(list(set(data['Date'])))
     if today not in available_dates:
+        if not available_dates:
+            return {"error": "No available trading dates in the data."}
         today = available_dates[-1]
 
     data_today = data[data['Date'] == today].between_time("09:15", "15:30")