Add application file

app.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from utils import get_chat_response
+from langchain.memory import ConversationBufferMemory
+
+st.title("聊天小助手")
+
+with st.sidebar:
+    openai_api_key = st.text_input("请输入OpenAI API密钥：", type= "password")
+    st.markdown("[获取 OpenAI API 密钥](https://platform.openai.com/docs/examples)")
+
+if "memory" not in st.session_state:
+    st.session_state["memory"] = ConversationBufferMemory(return_messages=True)
+    st.session_state["messages"] = [{"role":"ai",
+                                      "content":"你好，我是你的AI助手，有什么可以帮你的吗？" }]
+    
+for message in st.session_state["messages"]:
+    st.chat_message(message["role"]).write(message["content"])
+
+prompt = st.chat_input()
+
+if prompt:
+    if not openai_api_key:
+        st.info("请输入你的OpenAI API Key")
+        st.stop
+    st.session_state["messages"].append({"role":"human", "content":prompt})
+    st.chat_message("human").write(prompt)
+
+    with st.spinner("AI正在思考，请稍等..."):
+        response = get_chat_response(prompt, st.session_state["memory"], openai_api_key)
+    
+    msg = {"role": "ai", "content": response}
+    st.session_state["messages"].append(msg)
+    st.chat_message("ai").write(response)
+
+submit = st.button("开启新一轮对话")
+if submit:
+    st.session_state["memory"] = ConversationBufferMemory(return_messages=True)
+    st.session_state["messages"] = [{"role":"ai", "content":"你好，我是你的AI助手，有什么可以帮你的吗？"}]
+
+    st.experimental_rerun()
+
+

requirements.txt

@@ -0,0 +1,2 @@
+langchain_openai
+langchain

utils.py

@@ -0,0 +1,14 @@
+from langchain_openai import ChatOpenAI
+from langchain.chains import ConversationChain
+from langchain.memory import ConversationBufferMemory
+
+def get_chat_response(prompt, memory,openai_api_key):
+    model = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key = openai_api_key, base_url= "https://api.aigc369.com/v1")
+    chain = ConversationChain(llm=model, memory=memory)
+
+    response = chain.invoke({"input":prompt})
+    return response["response"]
+
+# memory = ConversationBufferMemory(return_messages=True)
+# print(get_chat_response("北京有谁在呀？",memory, openai_api_key="sk-YWPVrZ0KM3B3BFVk40Ee827a2dE34124Af34F881Bf4d79B6"))
+# print(get_chat_response("我的上一个问题是什么？",memory, openai_api_key="sk-YWPVrZ0KM3B3BFVk40Ee827a2dE34124Af34F881Bf4d79B6"))