1

app.py

@@ -2,8 +2,30 @@ import streamlit as st
 from streamlit_chat import message
 import os
 import openai
-openai.api_key = os.environ.get('openai.api_key')
-# openai.api_key = ""
+
+#页面设置
+st.set_page_config(
+    page_title="小译AI对话单轮版",
+    page_icon=":robot:"
+)
+
+#检查账号登陆
+def get_text1():
+    if 'openai_key' not in st.session_state:
+        input_text1 = st.text_input("📫请请输入你的账号: ", key="input")
+        if st.button("发送", key="input3"):
+            st.session_state['openai_key'] = input_text1
+            return input_text1
+    else:
+        return st.session_state['openai_key']
+
+openai_key = get_text1()
+if openai_key:
+    openai.api_key = openai_key
+    st.write("")
+else:
+    st.write("请输入 OpenAI API Key")
+
 # openAI code
 def openai_create(prompt):
     try:
@@ -20,20 +42,29 @@ def openai_create(prompt):
         )
 
         return response.choices[0].text
-    except openai.error.RateLimitError:
-        Error="当前小译学长访问量过大，请过一段时间再试！开发者正在努力扩张服务能力，敬请期待！"
-        return Error
+
+    # except openai.error.UnauthorizedError:
+    #     return "API Key不正确，请输入正确的API Key。"
+    # except openai.error.RateLimitError:
+    #     Error="当前小译学长访问量过大，请过一段时间再试！开发者正在努力扩张服务能力，敬请期待！"
+    #     return Error
+
+    except Exception as e:
+        return "你的账号填写有误，请刷新页面重新填写正确的账号！"
 
 
 def chatgpt_clone(input):
     output = openai_create(input)
     return output
 
-# Streamlit App
-st.set_page_config(
-    page_title="Streamlit Chat - Demo",
-    page_icon=":robot:"
-)
+#  Streamlit App
+# st.set_page_config(
+#     page_title="小译AI对话单轮版",
+#     page_icon=":robot:"
+# )
+
+
+
 
 # st.header("🏅和小译学长聊聊天")
 
@@ -45,8 +76,8 @@ if 'past' not in st.session_state:
 
 
 def get_text():
-    input_text = st.text_input("📫请告诉小译学长你想说的吧: ", key="input")
-    if st.button("发送"):
+    input_text = st.text_input("📫请告诉小译学长你想说的吧: ", key="input1")
+    if st.button("发送", key="input2"):
         return input_text
     return None