Update app.py

app.py

@@ -55,7 +55,18 @@ def forecast_stock(ticker, period, interval):
 
         true_predictions = scaler.inverse_transform(test_predictions)
 
+        # # Plot results
+        # plt.figure(figsize=(10, 6))
+        # plt.plot(ts.index, ts.values, label='Original Data')
+        # plt.plot(ts_test.index, true_predictions, label='Forecasted Data')
+        # plt.legend()
+        # plt.xlabel('Time')
+        # plt.ylabel('Value')
+        # plt.title('Stock Price Forecast')
+        # plt.switch_backend("TkAgg")
         
+
+        # return plt.show()  # Indicate successful execution
         fig = plt.figure(figsize=(10, 6))
         plt.plot(ts.index, ts.values, label='Original Data')
         plt.plot(ts_test.index, true_predictions, label='Forecasted Data')
@@ -69,7 +80,8 @@ def forecast_stock(ticker, period, interval):
 
     except Exception as e:
         print(f"An error occurred: {e}")
-        return gr.Markdown(f"An error occurred: {e}")  # Display error in interface
+        return False  # Indicate error
+
 
 tickers_info="""
         **Ticker Examples**
@@ -95,9 +107,6 @@ tickers_info="""
         - **QQQ:** Invesco QQQ Trust
         - **IWM:** iShares Russell 2000 ETF
         """
-
-
-
 with gr.Blocks() as demo:
     gr.Interface(
         forecast_stock,  # Function to execute on submit