update

pages/chatbase.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : chatpdf
+# @Time         : 2023/4/25 17:01
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from chatllm.applications import ChatBase
+from chatllm.utils import llm_load4chat
+
+import streamlit as st
+from appzoo.streamlit_app.utils import display_pdf, reply4input
+
+st.set_page_config('🔥ChatLLM', layout='centered', initial_sidebar_state='collapsed')
+
+
+@st.experimental_singleton
+def get_chat_func():
+    chat_func = llm_load4chat()
+    return chat_func
+
+
+chat_func = get_chat_func()
+
+qa = ChatBase(chat_func=chat_func)
+
+
+def reply_func(query):
+    for response, _ in qa(query=query):
+        yield response
+
+
+# def reply_func(x):
+#     for i in range(10):
+#         time.sleep(1)
+#         x += str(i)
+#         yield x
+
+
+container = st.container()  # 占位符
+text = st.text_area(label="用户输入", height=100, placeholder="请在这儿输入您的问题")
+# knowledge_base = st.sidebar.text_area(label="知识库", height=100, placeholder="请在这儿输入您的问题")
+
+if st.button("发送", key="predict"):
+    with st.spinner("AI正在思考，请稍等........"):
+        history = st.session_state.get('state')
+        st.session_state["state"] = reply4input(text, history, container=container, reply_func=reply_func)

pages/多轮对话.py

@@ -6,25 +6,32 @@
 # @Author       : yuanjie
 # @WeChat       : meutils
 # @Software     : PyCharm
-# @Description  : 
+# @Description  :
+import time
+import types
+
 import streamlit as st
-from streamlit_chat import message
+from meutils.pipe import *
 
 from appzoo.streamlit_app.utils import reply4input
 
 if __name__ == '__main__':
-    def display_previous_message(texts=None):
-        if texts:
-            for msg in texts:
-                message(msg, avatar_style="bottts")  # display all the previous message
-
 
-    display_previous_message(["你好！我是你的电影小助手，很高兴为您服务。", "你可以向我提问。"])
+    previous_messages = ["你好！我是你的电影小助手，很高兴为您服务。", "你可以向我提问。"]
 
     container = st.container()  # 占位符
     text = st.text_area(label="用户输入", height=100, placeholder="请在这儿输入您的问题")
 
+
+    def reply_func(query):
+        for i in range(10):
+            time.sleep(0.5)
+            yield query
+            query += str(i)
+
+
     if st.button("发送", key="predict"):
         with st.spinner("AI正在思考，请稍等........"):
             history = st.session_state.get('state')
-            st.session_state["state"] = reply4input(text, history, container=container)
+            st.session_state["state"] = reply4input(text, history, container=container, reply_func=reply_func,
+                                                    previous_messages=previous_messages)

run.sh

@@ -6,5 +6,5 @@
 # @Software     : PyCharm
 # @Description  : ${DESCRIPTION}
 
-streamlit run _👋_.py --server.port 9955
+streamlit run _👋_.py --server.port 7766