Update app.py

app.py

@@ -71,5 +71,68 @@ def main():
     rsi_signals = strategic_signals.get_rsi_signals()
     st.write(rsi_signals)
 
+if __name__ == "__main__":
+    main()
+
+
+
+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)
+    alpha_vantage_data = response.json()
+    return alpha_vantage_data
+
+def calculate_indicators(data):
+    # Convert all columns to numeric
+    data = data.apply(pd.to_numeric, errors='coerce')
+
+    # Example: Simple condition for doji and inside
+    data['Doji'] = abs(data['close'] - data['open']) <= 0.01 * (data['high'] - data['low'])
+    data['Inside'] = (data['high'] < data['high'].shift(1)) & (data['low'] > data['low'].shift(1))
+    return data
+
+def main():
+    st.title("AI Stock Trend Predictor")
+
+    # User input for stock symbol
+    symbol = st.text_input("Enter Stock Symbol (e.g., IBM):")
+
+    if not symbol:
+        st.warning("Please enter a valid stock symbol.")
+        st.stop()
+
+    # Use the hard-coded API key
+    api_key = API_KEY
+
+    # Fetch Alpha Vantage data
+    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)', {})
+    df = pd.DataFrame(alpha_vantage_time_series).T
+    df.index = pd.to_datetime(df.index)
+    df = df.dropna(axis=0)
+
+    # Rename columns
+    df = df.rename(columns={'1. open': 'open', '2. high': 'high', '3. low': 'low', '4. close': 'close', '5. volume': 'volume'})
+
+    # Calculate indicators
+    df = calculate_indicators(df)
+
+    # Create predictor
+    my_market_predictor = Pandas_Market_Predictor(df)
+
+    # Predict Trend
+    indicators = ["Doji", "Inside"]
+    trend = my_market_predictor.Trend_Detection(indicators, 10)
+
+    # Display results
+    st.subheader("Predicted Trend:")
+    st.write("Buy Trend :", trend['BUY'])
+    st.write("Sell Trend :", trend['SELL'])
+    
+    # Delete the DataFrame to release memory
+    del df
+
 if __name__ == "__main__":
     main()