error handling updated

app.py

@@ -46,9 +46,9 @@ def prepare_data(df, forecast_col, forecast_out):
     y = np.array(label)  # Target array
 
     # Check if we have enough data for train-test split
-    if len(y) == 0 or len(X) == 0:
+    if len(X) < 2 or len(y) < 2:  # Need at least two samples to split
         st.error(
-            "Not enough data for training. Adjust the forecast period or date range."
+            "Not enough data for splitting into training and testing sets. Please adjust the date range or prediction period."
         )
         return None, None, None, None, None
 
@@ -61,10 +61,16 @@ def prepare_data(df, forecast_col, forecast_out):
 
 # Button to trigger model generation and prediction
 if st.button("Generate"):
-    # Fetch stock data
+    # Ensure the days are in proper format
     if daysago != "max":
-        daysago = str(daysago) + "d"
+        try:
+            # Convert daysago to the correct format recognized by yfinance
+            daysago = str(int(daysago)) + "d"
+        except ValueError:
+            st.error("Invalid time frame. Please enter a number or 'max'.")
+            st.stop()
 
+    # Fetch stock data
     ticker = yf.Ticker(stock)
     data = ticker.history(period=daysago)