Update app.py

app.py

@@ -185,4 +185,20 @@ def plot_stock_data(ticker, start_date, end_date):
     plt.show()
     
     return plt.gcf()  # Return the current figure object for Gradio to display
+    print(f"Future dates length: {len(future_dates)}")
+print(f"Future prices length: {len(future_prices)}")
+def plot_stock_data(ticker, start_date, end_date):
+    # Your existing code to fetch data and make predictions...
+
+    # Check lengths of future_dates and future_prices
+    print(f"Future dates length: {len(future_dates)}")
+    print(f"Future prices length: {len(future_prices)}")
+
+    # Ensure they match
+    if len(future_dates) != len(future_prices):
+        raise ValueError("The length of future_dates and future_prices must match.")
+
+    plt.plot(future_dates, future_prices, label="Predicted Future Prices")
+    # Additional plotting code...
+