update: Change database loading method to use a predefined path and add error handling

app.py

@@ -21,10 +21,10 @@ def main():
     # Load embedding model
     embedding_model = SentenceTransformer(model_name_or_path="all-mpnet-base-v2", device=device)
 
-    # File upload for the database
-    database_file = st.file_uploader("Upload the Parquet database file", type=["parquet"])
+    # Load database from predefined path
+    database_file = '[Embedded Chunks] The Alchemy of Happiness (Ghazzālī, Claud Field) (Z-Library).parquet'
 
-    if database_file is not None:
+    try:
         df, chunk_embeddings = load_data(database_file)
         st.success("Database loaded successfully!")
 
@@ -52,6 +52,8 @@ def main():
                 st.write(f"**Text:** {df.iloc[int(idx)]['ext']}")
                 st.write(f"**Number of tokens:** {df.iloc[int(idx)]['tokens']}")
                 st.write("---")
+    except Exception as e:
+        st.error(f"Failed to load database: {str(e)}")
 
 if __name__ == "__main__":
     main()