Update app.py

app.py

@@ -21,16 +21,25 @@ def embed_query(text):
     return np.array(response.data[0].embedding, dtype=np.float32).reshape(1, -1)
 
 # Semantic search using FAISS (for older FAISS versions)
+
+# Semantic search with fallback handling
 def search(query, k=3):
     query_vec = embed_query(query).astype(np.float32)
 
-    distances = np.empty((1, k), dtype=np.float32)
-    labels = np.empty((1, k), dtype=np.int64)
+    try:
+        # Try modern FAISS API (common in recent versions)
+        distances, labels = index.search(query_vec, k)
+    except TypeError:
+        try:
+            # Try older FAISS API where distances and labels must be preallocated
+            distances = np.empty((1, k), dtype=np.float32)
+            labels = np.empty((1, k), dtype=np.int64)
+            index.search(query_vec, k, distances, labels)
+        except Exception as e:
+            raise RuntimeError(f"FAISS search failed: {e}")
 
-    index.search(query_vec, k, distances, labels)
     return [chunks[i] for i in labels[0]]
 
-# Chat modes
 def chat_no_rag(question):
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",