Upload 2 files

app.py

@@ -8,13 +8,10 @@
     from matplotlib.font_manager import FontProperties
     myfont=FontProperties(fname='/Users/yunshi/Downloads/360Data/Data Center/Working-On Task/演讲与培训/2023ChatGPT/Coding/code_interpreter/rawdata/SimHei.ttf')
     sns.set_style('whitegrid',{'font.sans-serif':['simhei','Arial']})      
-1. app.py:
-    1. openai key, bing key.
-    1. proxy
-    1. 
+
 
 '''
-# TODO：1. remove the table display in the second round. 2. account system. 
+# TODO：1. Chinese display isssue. 2. account system.
 
 import requests
 from codeinterpreterapi import CodeInterpreterSession, File
@@ -33,34 +30,28 @@ from pathlib import Path
 from matplotlib.font_manager import FontProperties
 import seaborn as sns
 from time import sleep
+import streamlit_authenticator as stauth
 
 os.environ["OPENAI_API_KEY"] = os.environ['user_token']
 openai.api_key = os.environ['user_token']
+# os.environ["VERBOSE"] = "True"  # 可以看到具体的错误？
 
-# #* 如果碰到接口问题，可以启用如下设置。
+# # #* 如果碰到接口问题，可以启用如下设置。
 # openai.proxy = {
 #     "http": "http://127.0.0.1:7890",
 #             "https": "http://127.0.0.1:7890"
 # }
 
+
 # layout settings.
 st.title("个人大语言模型商业智能中心")
 st.subheader("Artificial Intelligence Backend Center for Individuals")
 
-# st.image("./知识星球收入.jpg")
-# tab1, tab2 = st.tabs(["传统模式", "数据模型"])
-# with tab1:
-col1, col2 = st.columns(spec=[1, 2])
-radio_1 = col1.radio(label='ChatGPT版本', options=[
-    'GPT-3.5', 'GPT-4.0'], horizontal=True, label_visibility='visible')
-radio_2 = col2.radio(label='模式选择', options=[
-    '核心模式', '联网模式', '数据模式'], horizontal=True, label_visibility='visible')
-
 # with tab2:
 def upload_file(uploaded_file):
     if uploaded_file is not None:
         filename = uploaded_file.name
-        st.write(filename) ## print out the whole file name to validate.
+        st.write(filename)  # print out the whole file name to validate.
         try:
             if '.csv' in filename:
                 csv_file = pd.read_csv(uploaded_file)
@@ -81,14 +72,15 @@ def upload_file(uploaded_file):
             # ! 必须用这种格式读入内容，然后才可以写入temporary文件夹中。
             # output_temporary_file.write(uploaded_file.getvalue())
             output_temporary_file.write(uploaded_file.getvalue())
+            # st.write(uploaded_file_path)  # * 可以查看文件是否真实存在，然后是否可以
+        # st.write('Now file saved successfully.')
 
         return None
 
 
-bing_search_api_key = os.environ['bing_api_key']
+bing_search_api_key = os.environ['user_token']
 bing_search_endpoint = 'https://api.bing.microsoft.com/v7.0/search'
 
-
 def search(query):
     # Construct a request
     # mkt = 'en-EN'
@@ -110,7 +102,6 @@ def search(query):
 
 # openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-# async def main():
 async def text_mode():
     # Set a default model
     if "openai_model" not in st.session_state:
@@ -148,7 +139,7 @@ async def text_mode():
             full_response = ""
 
             if radio_2 == '联网模式':
-                # print('联网模式入口，prompt:', prompt)
+                print('联网模式入口，prompt:', prompt)
                 input_message = prompt
                 internet_search_result = search(input_message)
                 search_prompt = [
@@ -175,6 +166,7 @@ async def text_mode():
                 message_placeholder.markdown(full_response)
                 st.session_state.messages.append(
                     {"role": "assistant", "content": full_response})
+                st.session_state.messages = []
 
             if radio_2 == '核心模式':
                 print('GPT only starts!!!')
@@ -185,7 +177,8 @@ async def text_mode():
                     #     {"role": m["role"], "content": m["content"]}
                     #     for m in st.session_state.messages
                     # ],
-                    messages=[{'role': 'system', 'content': 'you are ChatGPT'}, {'role': 'user', 'content': prompt}],
+                    messages=[{'role': 'system', 'content': 'you are ChatGPT'}, {
+                        'role': 'user', 'content': prompt}],
                     stream=True,
                 ):
                     full_response += response.choices[0].delta.get(
@@ -198,9 +191,9 @@ async def text_mode():
 
 
 async def data_mode():
-    # print('数据分析模式启动！')
+    print('数据分析模式启动！')
     uploaded_file_path = './upload.csv'
-    # st.write(f"passed file path in data_mode: {uploaded_file_path}")
+    # # st.write(f"passed file path in data_mode: {uploaded_file_path}")
     # tmp1 = pd.read_csv('./upload.csv')
     # st.write(tmp1[:5])
 
@@ -226,23 +219,22 @@ async def data_mode():
 
         with st.chat_message("assistant"):
             async with CodeInterpreterSession() as session:
-                myfont=FontProperties(fname='./SimHei.ttf')
-                print(myfont)
                 # user_request = "对于文件中的'SepalLengthCm’数据给我一个'直方图'，提供图表，并给出分析结果"
                 #! 可以用设定dpi=300来输出高质量的图表。（注：图的解析度dpi设定为300）
                 environ_settings = """【背景要求】如果我没有告诉你任何定制化的要求，那么请你按照以下的默认要求来回答：
                 -------------------------------------------------------------------------
                 1. 你需要用提问的语言来回答（如：中文提问你就用中文来回答，英文提问你就用英文来回答）。
-                2. 如果要求你输出图表，那么图的解析度dpi需要设定为600。图尽量使用seaborn库。seaborn库的参数设定：sns.set(rc={'axes.facecolor':'#FFF9ED','figure.facecolor':'#FFF9ED'}, palette='dark'。
-                3. 输出的图上所有的文字全部用[英文（English）]来表示。
+                2. 如果要求你输出图表，那么图的解析度dpi需要设定为300。图尽量使用seaborn库。seaborn库的参数设定：sns.set(rc={'axes.facecolor':'#FFF9ED','figure.facecolor':'#FFF9ED'}, palette='dark'。
+                3. 在图上显示中文时，需要按如下要求操作：
+                    myfont=FontProperties(fname='/Users/yunshi/Downloads/360Data/Data Center/Working-On Task/演讲与培训/2023ChatGPT/Coding/code_interpreter/SimHei.ttf')
+                    plt.rcParams['font.sans-serif'] = ['myfont']
+                    plt.rcParams['font.family']='sans-serif'
+                    sns.set_style({'font.sans-serif':['myfont']})
                 -------------------------------------------------------------------------
                 """  # seaborn中的palette参数可以设定图表的颜色,选项包括：deep, muted, pastel, bright, dark, colorblind，Spectral。更多参数可以参考：https://seaborn.pydata.org/generated/seaborn.color_palette.html。
 
                 # uploaded_file_path = upload_file()
 
-                ## 设置中文字体显示。
-                # sns.set_style({'font.sans-serif':['myfont']})
-                
                 user_request = environ_settings + "\n\n" + \
                     "你需要完成以下任务：\n\n" + prompt + "\n\n" \
                     f"注：文件位置在{uploaded_file_path}"
@@ -268,32 +260,134 @@ async def data_mode():
                         # st.image(file.get_image()) #! working.
                         # * 注意这里的设定，可以提高图片的精细程度。
                         st.image(file.get_image(), width=None,
-                                    output_format='PNG')
+                                 output_format='PNG')
 
                     # message_placeholder.markdown(full_response + "▌") ## orignal code.
                 # message_placeholder.markdown(full_response) ## orignal code.
                 st.write(full_response)
                 status.update(label='complete', state='complete')
-                # st.session_state.messages.append(
-                #     {"role": "assistant", "content": full_response})
+
+                # TODO: 确认是否要记录所有的full response。
+                st.session_state.messages.append(
+                    {"role": "assistant", "content": full_response})
 
                 await session.astop()  # ! 确认需要关闭。
-        st.session_state.messages.append({"role": "assistant", "content": full_response})
+        # st.session_state.messages.append({"role": "assistant", "content": full_response})
+
+
+# authentication.
+import yaml
+from yaml.loader import SafeLoader
+with open('/Users/yunshi/Downloads/360Data/Data Center/Working-On Task/演讲与培训/2023ChatGPT/Coding/code_interpreter/config.yaml') as file:
+    config = yaml.load(file, Loader=SafeLoader)
+authenticator = stauth.Authenticate(
+    config['credentials'],
+    config['cookie']['name'],
+    config['cookie']['key'],
+    config['cookie']['expiry_days'],
+    config['preauthorized']
+)
+
+name, authentication_status, username = authenticator.login('Login', 'sidebar')
+
+if authentication_status:
+    with st.sidebar:
+        #     st.markdown(
+        #     """
+        #     <style>
+        #     [data-testid="stSidebar"][aria-expanded="true"]{
+        #         min-width: 600x;
+        #         max-width: 600px;
+        #     }
+        #     """,
+        #     unsafe_allow_html=True,
+        # )
+        st.header(f'**欢迎 **{name}** 来到人工智能的世界** ♠')
+        st.write(f'_Welcome and Hope U Enjoy Staying Here！_')
+        authenticator.logout('登出', 'sidebar')
+        # with st.header(body="欢迎"):
+        #     st.markdown("# 欢迎使用大语言模型商业智能中心")
+        # with st.expander(label=("**重要的使用注意事项**"), expanded=True):
+        with st.container():
+            st.markdown("#### 重要的使用注意事项")
+            with st.text(body="说明"):
+                st.markdown("* 重启一轮新对话时，只需要刷新页面（按Ctrl/Command + R）即可。")
+            with st.text(body="说明"):
+                st.markdown("* 为了保护数据与隐私，所有对话均不会被保持，刷新页面立即删除。敬请放心。")
+            with st.text(body="说明"):
+                st.markdown("* 'GPT-4'回答质量极佳，但速度缓慢、且不支持长文。建议适当使用。")
+            with st.text(body="说明"):
+                st.markdown("* “联网模式”与搜索引擎一致，仅限一轮对话，不会保持之前的会话记录。")
+            with st.text(body="说明"):
+                st.markdown("* “数据模式”暂时只支持1000个单元格以内的数据分析，单元格中的内容不支持中文数据（表头也尽量使用中文）。一般运行时间在1-5分钟左右，期间需要保持网络畅通。")
+            with st.text(body="说明"):
+                st.markdown("* “数据模式”推荐上传csv格式的文件（即utf-8格式）；部分Excel文件容易出现数据不兼容的情况。")
+        
+        st.markdown("#### 参考资料")
+        with st.expander(label="**核心模式的专用提示词Prompt示例**", expanded=False):
+            # with st.subheader(body="提示词Prompt"):
+            st.code(body="继续用中文写一篇关于 [文章主题] 的文章，以下列句子开头：[文章开头］。", language='plaintext')
+            st.code(body="将以下文字概括为 100 个字，使其易于阅读和理解。避免使用复杂的句子结构或技术术语。", language='plaintext')
+            st.code(body="给我出一个迪奥2023春季发布会活动策划。", language='plaintext')
+            st.code(body="帮我按照正式会议结构写一个会邀：主题是xx手机游戏立项会议。", language='plaintext')
+            st.code(body="帮我写一个车内健康监测全场景落地的项目计划，用表格。", language='plaintext')
+            st.code(
+                body="同时掷两枚质地均匀的骰子，则两枚骰子向上的点数之和为 7 的概率是多少。", language='plaintext')
+            st.code(body="写一篇产品经理的演讲稿，注意使用以下词汇: 赋能,抓手,中台,闭环,落地,漏斗,沉淀,给到,同步,对齐,对标,迭代,拉通,打通,升级,交付,聚焦,倒逼,复盘,梳理,方案,联动,透传,咬合,洞察,渗透,兜底,解耦,耦合,复用,拆解。", language='plaintext')
+        
+        with st.expander(label="**数据模式的专用提示词Prompt示例**", expanded=False):
+            # with st.subheader(body="提示词Prompt"):
+            st.code(body="分析此数据集并绘制一些'有趣的图表'。", language='python')
+            st.code(body="对于这个文件中的数据，你需要要找出[X,Y]数据之间的寻找'相关性'。", language='python')
+            st.code(body="对于这个文件中的[xxx]数据给我一个'整体的分析'。", language='python')
+            st.code(body="对于[xxx]数据给我一个'直方图'，提供图表，并给出分析结果。", language='python')
+            st.code(body="对于[xxx]数据给我一个'小提琴图'，并给出分析结果。", language='python')
+            st.code(
+                body="对于[X,Y,Z]数据在一个'分布散点图 (stripplot)'，所有的数据在一张图上展现, 并给出分析结果。", language='python')
+            st.code(body="对于[X，Y]数据，进行'T检验'，你需要展示图表，并给出分析结果。", language='python')
+            st.code(body="对于[X，Y]数据给我一个3个类别的'聚类分析'，并给出分析结果。", language='python')
+
+    col1, col2 = st.columns(spec=[1, 2])
+    radio_2 = col2.radio(label='模式选择', options=[
+        '核心模式', '联网模式', '数据模式'], horizontal=True, label_visibility='visible')
+    # radio_1 = col1.selectbox(label='ChatGPT版本', options=[
+    #     'GPT-3.5', 'GPT-4.0'], label_visibility='visible')
+    radio_1 = col1.radio(label='ChatGPT版本', options=[
+        'GPT-3.5', 'GPT-4.0'], horizontal=True, label_visibility='visible')
+                            
+            # import asyncio
+            # if radio_2 != "数据模式":
+            #     # * 也可以用命令执行这个python文件。’streamlit run frontend/app.py‘
+            #     asyncio.run(text_mode())
+            #     # upload_file()
+            # elif radio_2 == "数据模式":
+            #     uploaded_file = st.file_uploader(
+            #         "选择一个文件", type=(["csv", "xlsx", "xls"]))
+            #     # 默认状态下没有上传文件，None，会报错。需要判断。
+            #     if uploaded_file is not None:
+            #         uploaded_file_path = upload_file(uploaded_file)
+            #         asyncio.run(data_mode())
+
+elif authentication_status == False:
+    st.error('⛔ 用户名或密码错误！')
+elif authentication_status == None:
+    st.warning(' ⬅️ 请先登录！')
+    
 
 if __name__ == "__main__":
     import asyncio
-    if radio_2 != "数据模式":
-        # * 也可以用命令执行这个python文件。’streamlit run frontend/app.py‘
-        asyncio.run(text_mode())
-        # upload_file()
-    elif radio_2 == "数据模式":
-        uploaded_file = st.file_uploader(
-        "选择一个文件", type=(["csv", "xlsx", "xls"]))
-        ## 默认状态下没有上传文件，None，会报错。需要判断。
-        if uploaded_file is not None:
-            uploaded_file_path = upload_file(uploaded_file)
-            asyncio.run(data_mode())
-        
-
-
-
+    try:
+        if radio_2 != "数据模式":
+            # * 也可以用命令执行这个python文件。’streamlit run frontend/app.py‘
+            asyncio.run(text_mode())
+            # upload_file()
+        elif radio_2 == "数据模式":
+            uploaded_file = st.file_uploader(
+                "选择一个文件", type=(["csv", "xlsx", "xls"]))
+            # 默认状态下没有上传文件，None，会报错。需要判断。
+            if uploaded_file is not None:
+                uploaded_file_path = upload_file(uploaded_file)
+                asyncio.run(data_mode())
+    except:
+        # st.markdown('**请先登录！**')
+        pass

config.yaml

@@ -0,0 +1,22 @@
+credentials:
+  usernames:
+    joeshi:
+      email: [email protected]
+      name: joeshi
+      password: '$2b$12$8Jbjy0Cvq4UnXksw2PUa8.BOmssCExwIS/SrSZX4JwGWu1zJJ5Jxu' # To be replaced with hashed password
+    cindyji:
+      email: [email protected]
+      name: cindyji
+      password: '$2b$12$CA3Kc0sL0jVax7Wx0bbC7Op3edhtxlFUJkHS6ABjZCBN/1kbS3yem' # To be replaced with hashed password
+    test:
+      email: [email protected]
+      name: test
+      password: '$2b$12$ZkS1cFhMN7L7em0QOwZ8zewAwvJrt1eA5/2bpikHEyXXFpXApVBwy' # To be replaced with hashed password, test
+
+cookie:
+  expiry_days: 30
+  key: random_signature_key # Must be string
+  name: random_cookie_name
+preauthorized:
+  emails:
+  - [email protected]