Update app.py

app.py

@@ -66,7 +66,6 @@ model = None
 def initialize_model():
     """
     Initialize the sentence transformer model.
-
     Returns:
         bool: Whether the model was successfully initialized.
     """
@@ -195,6 +194,13 @@ def semantic_search(query, top_k=5):
         # Load saved embeddings and chunks from OUTPUTS_DIR
         embeddings_file = os.path.join(OUTPUTS_DIR, "embeddings.npy")
         chunks_file = os.path.join(OUTPUTS_DIR, "chunks.txt")
+
+        logger.info(f"Checking for embeddings file: {embeddings_file}")
+        logger.info(f"Checking for chunks file: {chunks_file}")
+
+        if not os.path.exists(embeddings_file) or not os.path.exists(chunks_file):
+            return "Embeddings or chunks not found. Please generate embeddings first."
+
         stored_embeddings = np.load(embeddings_file)
         with open(chunks_file, "r", encoding="utf-8") as f:
             chunks = f.read().split("\n===CHUNK_SEPARATOR===\n")