Update app.py

app.py

@@ -47,14 +47,11 @@ def plot_stock_data(df, forecast):
     # Plot historical data
     plt.plot(df['Date'], df['Close'], label='Historical', color='blue')
     
-    # Align predicted data by combining historical and future dates
-    combined_dates = pd.concat([df['Date'], forecast['ds']]).drop_duplicates()
+    # Plot predicted data only for the future dates (after the last date of historical data)
+    forecast_future = forecast[forecast['ds'] > df['Date'].max()]
     
-    # Append the predicted values starting from the last historical date onwards
-    predicted_values = pd.concat([pd.Series([None] * len(df)), forecast['yhat1'].iloc[len(df):]]).reset_index(drop=True)
-    
-    # Plot predicted data
-    plt.plot(combined_dates, predicted_values, label='Predicted', linestyle='--', color='red')
+    # Plot predicted future data
+    plt.plot(forecast_future['ds'], forecast_future['yhat1'], label='Predicted', linestyle='--', color='red')
     
     plt.xlabel('Date')
     plt.ylabel('Stock Price')