Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 from llama_index.core import VectorStoreIndex, Settings
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.llms.huggingface import HuggingFaceLLM
-from llama_index.readers import SimpleDirectoryReader # Corrected import
+from llama_index import SimpleDirectoryReader  # Corrected import
 
 st.set_page_config(page_title="llama_index_demo", page_icon="🦜🔗")
 st.title("llama_index_demo")
@@ -24,16 +24,7 @@ def init_models():
     )
     Settings.llm = llm
 
-    # 获取 data 文件夹的绝对路径
-    data_dir_path = os.path.abspath("./data")
-
-    # 加载目录中的所有文件
-    try:
-        documents = SimpleDirectoryReader(data_dir_path).load_data()
-    except Exception as e:
-        st.error(f"Error loading data from directory {data_dir_path}: {e}")
-        return None
-    
+    documents = SimpleDirectoryReader("./").load_data()
     index = VectorStoreIndex.from_documents(documents)
     query_engine = index.as_query_engine()