Update app.py

app.py

@@ -43,39 +43,9 @@ def main():
 if __name__ == "__main__":
     main()
 
-def main():
-    st.title("Strategic Trading Signals")
-
-    # Input for stock symbol
-    symbol = st.text_input("Enter stock symbol (e.g., AAPL):", "AAPL")
-
-    # Display strategic trading signals
-    strategic_signals = StrategicSignals(symbol=symbol)
-
-    st.subheader("Bollinger Bands Signals:")
-    bollinger_bands_signals = strategic_signals.get_bollinger_bands_signals()
-    st.write(bollinger_bands_signals)
-
-    st.subheader("Breakout Signals:")
-    breakout_signals = strategic_signals.get_breakout_signals()
-    st.write(breakout_signals)
-
-    st.subheader("Crossover Signals:")
-    crossover_signals = strategic_signals.get_crossover_signals()
-    st.write(crossover_signals)
-
-    st.subheader("MACD Signals:")
-    macd_signals = strategic_signals.get_macd_signals()
-    st.write(macd_signals)
-
-    st.subheader("RSI Signals:")
-    rsi_signals = strategic_signals.get_rsi_signals()
-    st.write(rsi_signals)
-
-if __name__ == "__main__":
-    main()
+# Hard-coded API key for demonstration purposes
+API_KEY = "QR8F9B7T6R2SWTAT"
 
-    
 def fetch_alpha_vantage_data(api_key, symbol):
     url = f'https://www.alphavantage.co/query?function=TIME_SERIES_INTRADAY&symbol={symbol}&interval=5min&apikey={api_key}'
     response = requests.get(url)
@@ -91,15 +61,18 @@ def calculate_indicators(data):
     data['Inside'] = (data['High'] < data['High'].shift(1)) & (data['Low'] > data['Low'].shift(1))
     return data
 
-def main():
+def display_signals(signal_type, signals):
+    st.subheader(f"{signal_type} Signals:")
+    st.write(signals)
 
+def main():
     st.title("AI Stock Trend Predictor")
-    
-    # Use the hard-coded API key
-    api_key = API_KEY
+
+    # Input for stock symbol
+    symbol = st.text_input("Enter stock symbol (e.g., AAPL):", "AAPL")
 
     # Fetch Alpha Vantage data
-    alpha_vantage_data = fetch_alpha_vantage_data(api_key, symbol)
+    alpha_vantage_data = fetch_alpha_vantage_data(API_KEY, symbol)
 
     # Extract relevant data from Alpha Vantage response
     alpha_vantage_time_series = alpha_vantage_data.get('Time Series (5min)', {})
@@ -113,6 +86,15 @@ def main():
     # Calculate indicators
     df = calculate_indicators(df)
 
+    # Display stock trading signals
+    strategic_signals = StrategicSignals(symbol=symbol)
+
+    display_signals("Bollinger Bands", strategic_signals.get_bollinger_bands_signals())
+    display_signals("Breakout", strategic_signals.get_breakout_signals())
+    display_signals("Crossover", strategic_signals.get_crossover_signals())
+    display_signals("MACD", strategic_signals.get_macd_signals())
+    display_signals("RSI", strategic_signals.get_rsi_signals())
+
     # Create predictor
     my_market_predictor = Pandas_Market_Predictor(df)
 
@@ -120,11 +102,11 @@ def main():
     indicators = ["Doji", "Inside"]
     trend = my_market_predictor.Trend_Detection(indicators, 10)
 
-    # Display results
+    # Display trend predictions
     st.subheader("Predicted Trend:")
     st.write("Buy Trend :", trend['BUY'])
     st.write("Sell Trend :", trend['SELL'])
-    
+
     # Delete the DataFrame to release memory
     del df