Update app.py

app.py

@@ -1,3 +1,4 @@
+<<<<<<< HEAD
 import logging
 import sys
 
@@ -42,3 +43,62 @@ def main() -> None:
 if __name__ == "__main__":
     load_dotenv()
     main()
+=======
+import logging
+import sys
+
+import streamlit as st
+
+from xpipe_wiki.manager_factory import XPipeRobotManagerFactory, XPipeRobotRevision
+from xpipe_wiki.robot_manager import XPipeWikiRobot, AzureOpenAIXPipeWikiRobot
+
+logging.basicConfig(
+    stream=sys.stdout, level=logging.DEBUG
+)  # logging.DEBUG for more verbose output
+logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
+
+# Sidebar contents
+with st.sidebar:
+    st.title("🤗💬 LLM Chat App")
+    st.markdown(
+        """
+    ## About
+    This app is an LLM-powered chatbot built using:
+    - [Streamlit](https://streamlit.io/)
+    - [LangChain](https://python.langchain.com/)
+    - [X-Pipe](https://github.com/ctripcorp/x-pipe)
+    """
+    )
+    # add_vertical_space(5)
+    st.write("Made by Nick")
+
+
+def main() -> None:
+    st.header("X-Pipe Wiki 机器人 💬")
+    # robot_manager = XPipeRobotManagerFactory.get_or_create(XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0)
+    # robot = robot_manager.get_robot()
+    from llama_index import StorageContext
+    store = StorageContext.from_defaults(persist_dir="./dataset")
+    from langchain_manager.manager import LangChainAzureManager
+    lc_manager = LangChainAzureManager()
+    from llama_index import ServiceContext
+    from llama_index import LangchainEmbedding
+    service = ServiceContext.from_defaults(llm=lc_manager.get_llm(), embed_model=LangchainEmbedding(lc_manager.get_embedding()))
+
+    from llama_index import load_index_from_storage
+    index = load_index_from_storage(store)
+    query_engine = index.as_query_engine(service_context=service)
+    robot = AzureOpenAIXPipeWikiRobot(query_engine)
+    query = st.text_input("X-Pipe Wiki 问题:")
+    if query:
+        response = robot.ask(question=query)
+        st.write(response)
+    query = st.text_input("X-Pipe Wiki 问题:")
+    if query:
+        response = robot.ask(question=query)
+        st.write(response)
+
+
+if __name__ == "__main__":
+    main()
+>>>>>>> 2bf20e0 (Update app.py)