Update main.py

main.py

@@ -17,10 +17,19 @@ def initialize_system():
     # Load and process documents
     documents = load_single_document(Config.DOCUMENT_FILE)
     logger.info(f"Loaded {len(documents)} documents")
+    # ❗ Prevent crash if no document loaded
+    if documents.empty:
+        raise FileNotFoundError(f"❌ No document found at: {Config.DOCUMENT_FILE}. "
+                                f"Make sure the .txt file is uploaded in Hugging Face.")
+
 
     # Generate embeddings
     embeddings = generate_embeddings(documents)
     logger.info(f"Generated embeddings of shape {embeddings.shape}")
+    # ❗ Prevent FAISS error if embeddings are empty
+    if embeddings.shape[0] == 0:
+        raise ValueError("❌ Embeddings are empty! Document might be empty or path is wrong.")
+
 
     # Initialize search engine
     search_engine = SearchEngine(documents, embeddings)