Update app.py

app.py

@@ -69,7 +69,7 @@ def retrieve(query: str, index, category: str = None, k: int = 5) -> List[str]:
     if category:
         kwargs["filter"] = {"category": {"$eq": category}}
     res = index.query(vector=embed, **kwargs)
-    return [m["metadata"]["text"] for m in res["matches"] if "text" in m["metadata"]]
+    return [m["metadata"].get("text", "") for m in res["matches"] if "metadata" in m and m["metadata"].get("text")]
 
 # Generate final answer
 def generate_answer(query: str, docs: List[str]) -> str:
@@ -98,7 +98,8 @@ if __name__ == "__main__":
         documents = load_documents(".")
         index_documents(pinecone_index, documents)
 
-    categories = sorted({Path(p).parts[0] for p in glob.glob('*/*.txt')})
+    categories = sorted(set(Path(p).parts[0] for p in glob.glob('*/*.txt')) | set(recommended.keys()) - {"all"})
+
     recommended = {
         "all": [
             "Where can I find benefits information?",