Upload main.py

main.py

@@ -0,0 +1,91 @@
+from stocks import *
+from functions import *
+from datetime import datetime
+import streamlit as st
+
+st.set_page_config(layout="wide")
+
+st.title("Tech Stocks Trading Assistant")
+
+left_column, right_column = st.columns(2)
+
+with left_column:
+
+    all_tickers = {
+                "Apple":"AAPL", 
+                "Microsoft":"MSFT", 
+                "Nvidia":"NVDA", 
+                "Paypal":"PYPL",
+                "Amazon":"AMZN",
+                "Spotify":"SPOT",
+                "Twitter":"TWTR",
+                "AirBnB":"ABNB",
+                "Uber":"UBER",
+                "Google":"GOOG"
+                }
+
+    st.subheader("Technical Analysis Methods")
+    option_name = st.selectbox('Choose a stock:', all_tickers.keys())
+    option_ticker = all_tickers[option_name]
+    execution_timestamp = datetime.now()
+    'You selected: ', option_name, "(",option_ticker,")"
+    'Last execution:', execution_timestamp
+
+    s = Stock_Data()
+    t = s.Ticker(tick=option_ticker)  
+
+    m = Models()
+
+    with st.spinner('Loading stock data...'):
+
+        technical_analysis_methods_outputs = {
+            'Technical Analysis Method': [
+                'Bollinger Bands (20 days & 2 stand. deviations)', 
+                'Bollinger Bands (10 days & 1.5 stand. deviations)', 
+                'Bollinger Bands (50 days & 3 stand. deviations)', 
+                'Moving Average Convergence Divergence (MACD)'
+                ],
+                'Outlook': [
+                    m.bollinger_bands_20d_2std(t),         
+                    m.bollinger_bands_10d_1point5std(t), 
+                    m.bollinger_bands_50d_3std(t), 
+                    m.MACD(t)
+                    ],
+                'Timeframe of Method': [
+                    "Medium-term",         
+                    "Short-term", 
+                    "Long-term", 
+                    "Short-term"
+                    ]
+                }
+
+        df = pd.DataFrame(technical_analysis_methods_outputs)
+
+
+    def color_survived(val):
+        color = ""
+        if (val=="Sell" or val=="Downtrend and sell signal" or val=="Downtrend and no signal"):
+            color="#EE3B3B"
+        elif (val=="Buy" or val=="Uptrend and buy signal" or val=="Uptrend and no signal"):
+            color="#3D9140"
+        else:
+            color="#CD950C"
+        return f'background-color: {color}'
+    
+    
+    st.table(df.sort_values(['Timeframe of Method'], ascending=False).
+               reset_index(drop=True).style.applymap(color_survived, subset=['Outlook']))
+    
+with right_column:
+
+    st.subheader("FinBERT-based Sentiment Analysis")
+
+    with st.spinner("Connecting with www.marketwatch.com..."):
+        st.plotly_chart(m.finbert_headlines_sentiment(t)["fig"])
+        "Current sentiment:", m.finbert_headlines_sentiment(t)["current_sentiment"], "%"
+
+    st.subheader("LSTM-based 7-day stock price prediction model")
+
+    with st.spinner("Compiling LSTM model.."):
+        st.plotly_chart(m.LSTM_7_days_price_predictor(t))
+