Upload 3 files

htmlTemplates.py

@@ -0,0 +1,44 @@
+css = '''
+<style>
+.chat-message {
+    padding: 1.5rem; border-radius: 0.5rem; margin-bottom: 1rem; display: flex
+}
+.chat-message.user {
+    background-color: #2b313e
+}
+.chat-message.bot {
+    background-color: #475063
+}
+.chat-message .avatar {
+  width: 12%;
+}
+.chat-message .avatar img {
+  max-width: 60px;
+  max-height: 60px;
+  border-radius: 50%;
+  object-fit: cover;
+}
+.chat-message .message {
+  width: 90%;
+  padding: 0 .35rem;
+  color: #fff;
+}
+'''
+
+bot_template = '''
+<div class="chat-message bot">
+    <div class="avatar">
+        <img src="https://www.sideshow.com/storage/product-images/2171/c-3po_star-wars_square.jpg">
+    </div>
+    <div class="message">{{MSG}}</div>
+</div>
+'''
+
+user_template = '''
+<div class="chat-message user">
+    <div class="avatar">
+        <img src="https://resizing.flixster.com/ocuc8yjm8Fu5UK5Ze8lbdp58m9Y=/300x300/v2/https://flxt.tmsimg.com/assets/p11759522_i_h9_aa.jpg">
+    </div>    
+    <div class="message">{{MSG}}</div>
+</div>
+'''

main.py

@@ -0,0 +1,214 @@
+import streamlit as st
+import yfinance as yf
+import pandas as pd
+from langchain.agents import create_pandas_dataframe_agent
+import re
+import sqlite3
+from htmlTemplates import css, user_template, bot_template
+from typing import Optional, List, Mapping, Any
+from langchain.llms.base import LLM
+import g4f
+
+
+class FinLLM(LLM):
+
+    @property
+    def _llm_type(self) -> str:
+        return "custom"
+
+    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
+        out = g4f.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.5,  # You can adjust parameters as needed
+            max_tokens=350  # Adjust the token limit as needed
+        )  #
+        if stop:
+            stop_indexes = (out.find(s) for s in stop if s in out)
+            min_stop = min(stop_indexes, default=-1)
+            if min_stop > -1:
+                out = out[:min_stop]
+        return out
+
+
+llm = FinLLM()
+
+def create_users_db():
+    with sqlite3.connect('MASTER.db') as conn:
+        cursor = conn.cursor()
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS Users (
+                user_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                email TEXT UNIQUE,
+                password TEXT
+            )
+        """)
+
+def add_user_to_db(email, password):
+    with sqlite3.connect('MASTER.db') as conn:
+        cursor = conn.cursor()
+        try:
+            insert_query = "INSERT INTO Users (email, password) VALUES (?, ?)"
+            cursor.execute(insert_query, (email, password))
+        except sqlite3.IntegrityError as e:
+            # Handle specific error if the email already exists
+            print(f"Error: {e}")
+            return False
+        return True
+
+def authenticate_user(email, password):
+    with sqlite3.connect('MASTER.db') as conn:
+        cursor = conn.cursor()
+        select_query = "SELECT * FROM Users WHERE email = ? AND password = ?"
+        cursor.execute(select_query, (email, password))
+        user = cursor.fetchone()
+    return user is not None
+
+
+def init_ses_states():
+    st.session_state.setdefault('chat_history', [])
+    st.session_state.setdefault('user_authenticated', False)
+
+
+def relative_returns(df):
+    rel = df.pct_change()
+    cumret = ((1 + rel).cumprod() - 1).fillna(0)
+    return cumret
+
+
+def display_convo():
+    with st.container():
+        for i, message in enumerate(reversed(st.session_state.chat_history)):
+            if i % 2 == 0:
+                 st.markdown(bot_template.replace("{{MSG}}", message), unsafe_allow_html=True)
+            else:
+                st.markdown(user_template.replace("{{MSG}}", message), unsafe_allow_html=True)
+
+
+def approve_password(password):
+    if len(password) >= 8 and re.search(r"(?=.*[a-z])(?=.*[A-Z])(?=.*[0-9])(?=.*[_@$#!?&*%])", password):
+        return True
+    return False
+    
+
+def approve_email(email):
+    email_regex = '^[a-zA-Z0-9]+[\._]?[a-zA-Z0-9]+[@]\w+[.]\w{2,3}$'
+    if re.search(email_regex, email):
+        return True
+    else:
+        return False
+    
+
+def user_authentication_tab():
+    with st.expander("User Authentication", expanded=True):
+        login_tab, create_account_tab = st.tabs(["Login", "Create Account"])
+
+        with login_tab:
+            email = st.text_input("Email:") 
+            password = st.text_input("Password:", type='password')
+            if st.button("Login"):
+                if authenticate_user(email=email,password=password):
+                    st.session_state.user_authenticated = True
+                else:
+                    st.caption('Incorrect Username or Password.')
+
+            if st.session_state.user_authenticated:
+                st.caption("User Authenticated")
+
+        with create_account_tab:
+            new_email = st.text_input("New Email:")
+            new_password = st.text_input("New Password:", type='password')
+            confirm_password = st.text_input("Confirm Password:", type='password')
+            if st.button("Create Account"):
+                if not approve_email(new_email):
+                    st.caption("Invalid Email")
+                    return
+                if not approve_password(new_password):
+                    st.caption("Invalid Password")
+                    return
+                if new_password != confirm_password:
+                    st.caption("Passwords do not match")
+                    return
+                add_user_to_db(email=new_email, password=new_password)
+                st.caption(f"{new_email} Successfully Added")
+
+
+def main():
+    st.set_page_config(page_title="Stock Price AI Bot", page_icon=":chart:")
+    st.write(css, unsafe_allow_html=True)
+    create_users_db()
+    init_ses_states()
+    st.title("Stock Price AI Bot")
+    st.caption("Visualizations and OpenAI Chatbot for Multiple Stocks Over A Specified Period")
+
+
+    with st.sidebar:
+        user_authentication_tab()
+    
+    
+    if st.session_state.user_authenticated:
+        with st.sidebar:
+            with st.expander("Settings",expanded=True):
+                asset_tickers = sorted(['DOW','NVDA','TSL','GOOGL','AMZN','AI','NIO','LCID','F','LYFY','AAPL', 'MSFT', 'BTC-USD', 'ETH-USD'])
+                asset_dropdown = st.multiselect('Pick Assets:', asset_tickers)
+
+                metric_tickers = ['Adj. Close', 'Relative Returns']
+                metric_dropdown = st.selectbox("Metric", metric_tickers)
+
+                viz_tickers = ['Line Chart', 'Area Chart']
+                viz_dropdown = st.multiselect("Pick Charts:", viz_tickers)
+
+                start = st.date_input('Start', value=pd.to_datetime('2023-01-01'))
+                end = st.date_input('End', value=pd.to_datetime('today'))
+
+                chatbot_temp = st.slider("Chat Bot Temperature",0.0,1.0,0.5)
+
+        # Only when a stock is selected
+        if len(asset_dropdown) > 0:
+            df = yf.download(asset_dropdown,start,end)['Adj Close']
+            if metric_dropdown == 'Relative Returns':
+                df = relative_returns(df)
+            if len(viz_dropdown) > 0:
+                with st.expander("Data Visualizations for {} of {}".format(metric_dropdown,asset_dropdown), expanded=True):
+                    if "Line Chart" in viz_dropdown:
+                        st.subheader("Line Chart")
+                        st.line_chart(df)
+                    if "Area Chart" in viz_dropdown:
+                        st.subheader("Area Chart")
+                        st.area_chart(df)
+            st.header("Chat with your Data") 
+
+            query = st.text_input("Enter a query:")
+
+            chat_prompt = f'''
+                You are an AI ChatBot intended to help with user stock data.
+                \nYou have access to a pandas dataframe with the following specifications 
+                \nDATA MODE: {metric_dropdown}
+                \nSTOCKS: {asset_dropdown} 
+                \nTIME PERIOD: {start} to {end}
+                \nCHAT HISTORY: {st.session_state.chat_history}
+                \nUSER MESSAGE: {query}
+                \nAI RESPONSE HERE:
+            '''
+
+            if st.button("Execute") and query:
+                with st.spinner('Generating response...'):
+                    try:
+                        agent = create_pandas_dataframe_agent(
+                            llm,
+                            pd.DataFrame(df),
+                            verbose=True
+                        )
+
+                        answer = agent.run(chat_prompt)
+                        st.session_state.chat_history.append(f"USER: {query}\n")
+                        st.session_state.chat_history.append(f"AI: {answer}\n")
+                        display_convo()
+
+                    except Exception as e:
+                        st.error(f"An error occurred: {str(e)}")
+
+
+if __name__ == '__main__':
+    main()
+

requirements.txt

@@ -0,0 +1,90 @@
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==5.0.1
+appdirs==1.4.4
+async-timeout==4.0.2
+attrs==23.1.0
+beautifulsoup4==4.12.2
+blinker==1.6.2
+cachetools==5.3.1
+certifi==2023.5.7
+cffi==1.15.1
+charset-normalizer==3.1.0
+click==8.1.3
+colorama==0.4.6
+contourpy==1.0.7
+cryptography==41.0.1
+cycler==0.11.0
+dataclasses-json==0.5.7
+decorator==5.1.1
+fonttools==4.39.4
+frozendict==2.3.8
+frozenlist==1.3.3
+gitdb==4.0.10
+GitPython==3.1.31
+gpt-index==0.4.24
+greenlet==2.0.2
+html5lib==1.1
+idna==3.4
+importlib-metadata==6.6.0
+Jinja2==3.1.2
+jsonschema==4.17.3
+kiwisolver==1.4.4
+langchain==0.0.188
+lxml==4.9.2
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+marshmallow==3.19.0
+marshmallow-enum==1.5.1
+matplotlib==3.7.1
+mdurl==0.1.2
+multidict==6.0.4
+multitasking==0.0.11
+mypy-extensions==1.0.0
+numexpr==2.8.4
+numpy==1.24.3
+openai==0.27.7
+openapi-schema-pydantic==1.2.4
+packaging==23.1
+pandas==2.0.2
+Pillow==9.5.0
+protobuf==4.23.2
+pyarrow==12.0.0
+pycparser==2.21
+pydantic==1.10.8
+pydeck==0.8.1b0
+Pygments==2.15.1
+Pympler==1.0.1
+pyparsing==3.0.9
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+pytz==2023.3
+pytz-deprecation-shim==0.1.0.post0
+PyYAML==6.0
+regex==2023.6.3
+requests==2.31.0
+rich==13.4.1
+six==1.16.0
+smmap==5.0.0
+soupsieve==2.4.1
+SQLAlchemy==1.4.48
+streamlit==1.23.0
+tabulate==0.9.0
+tenacity==8.2.2
+tiktoken==0.4.0
+toml==0.10.2
+toolz==0.12.0
+tornado==6.3.2
+tqdm==4.65.0
+typing-inspect==0.9.0
+typing_extensions==4.6.2
+tzdata==2023.3
+tzlocal==4.3
+urllib3==2.0.2
+validators==0.20.0
+watchdog==3.0.0
+webencodings==0.5.1
+yarl==1.9.2
+yfinance==0.2.18
+zipp==3.15.0