Update app.py

app.py

@@ -75,27 +75,14 @@ with st.sidebar:
 
 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(
-        service_context=service,
-        storage_context=store,
+    robot_manager = XPipeRobotManagerFactory.get_or_create(
+        XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0
     )
-    query_engine = index.as_query_engine(service_context=service)
-    # robot = AzureOpenAIXPipeWikiRobot(query_engine)
+    robot = robot_manager.get_robot()
     query = st.text_input("X-Pipe Wiki 问题:")
     if query:
-        # response = robot.ask(question=query)
-        st.write(query_engine.query(query))
+        response = robot.ask(question=query)
+        st.write(response)
 
 
 if __name__ == "__main__":