Update app.py

app.py

@@ -54,10 +54,24 @@ def predict_stock_price(ticker, open_price, high_price, low_price, close_price,
         lstm_predictions = lstm_model.predict(x_test_lstm)
         lstm_predictions = scaler.inverse_transform(lstm_predictions)
         next_day_lstm_price = lstm_predictions[-1][0]
-
+        
+        # Plot the historical data and the predicted future price
+        plt.figure(figsize=(12, 6))
+        plt.plot(data.index, data['Close'], label='Historical Data')
+        plt.axhline(y=next_day_lstm_price, color='g', linestyle='--', label='Predicted Future Price')
+        plt.title(f'Historical Stock Price and Predicted Future Price for {ticker}')
+        plt.xlabel('Date')
+        plt.ylabel('Price (USD)')
+        plt.legend()
+        
+        # Save the plot to a file
+        plot_filename = 'stock_price_prediction.png'
+        plt.savefig(plot_filename)
+        plt.close()
+        
         result = f"Predicted future price for {ticker}: ${next_day_lstm_price:.2f}"
 
-        return result
+        return result, plot_filename
     except Exception as e:
         return str(e)
 
@@ -89,7 +103,7 @@ iface = gr.Interface(
         gr.Number(label="Adj Close"),
         gr.Number(label="Volume")
     ],
-    outputs=gr.Textbox(),
+    outputs=[gr.Textbox(),gr.Image()],
     title="Stock Price Predictor",
     description="Select the stock ticker and input the last recorded values to predict the closing price using the LSTM model."
 )