Update app.py

app.py

@@ -1,32 +1,27 @@
 import streamlit as st
-from thronetrader import RealTimeSignals
+from thronetrader import RealTimeSignals, Predictions
 
 def main():
-    st.title("Real-Time Trading Signals")
+    st.title("Stock Market Analysis")
 
-    # Create an instance of RealTimeSignals with the desired symbol
+    # Input for stock symbol
     symbol = st.text_input("Enter stock symbol (e.g., AAPL):", "AAPL")
-    realtime_signals = RealTimeSignals(symbol=symbol)
-
-    # Display financial signals
-    st.subheader("Financial Signals:")
-    financial_signals = realtime_signals.get_financial_signals()
-    st.write(financial_signals)
-
-    # Display insider signals
-    st.subheader("Insider Signals:")
-    insider_signals = realtime_signals.get_insider_signals()
-    st.write(insider_signals)
 
-    # Display trading volume
-    st.subheader("Trading Volume:")
-    series1, series2 = realtime_signals.get_trading_volume()
-
-    st.write(f"Series 1 ({series1.name}):")
-    st.write(series1.to_dict())
-
-    st.write(f"Series 2 ({series2.name}):")
-    st.write(series2.to_dict())
+    # Display real-time trading signals
+    realtime_signals = RealTimeSignals(symbol=symbol)
+    st.subheader("Real-Time Trading Signals:")
+    st.write("Financial Signals:")
+    st.write(realtime_signals.get_financial_signals())
+    st.write("Insider Signals:")
+    st.write(realtime_signals.get_insider_signals())
+
+    # Display stock price predictions
+    predictions = Predictions(symbol=symbol)
+    st.subheader("Stock Price Predictions:")
+    st.write("Linear Regression Prediction:")
+    st.write(predictions.linear_regression_prediction())
+    st.write("Gradient Boosting Prediction:")
+    st.write(predictions.gradient_boosting_prediction())
 
 if __name__ == "__main__":
     main()