[WIP]

app.py

@@ -11,6 +11,7 @@ collection_name = "docs"
 
 search_engine = searchengine(model_name, collection_name)
 
+
 def main():
     st.title("ChromaDB Search Engine")
     
@@ -34,10 +35,10 @@ def main():
         st.subheader("Search Results:")
         logging.info("result :")
         logging.info(results)
-        ids = results['ids']
-        distances = results['distances']
-        metadatas = results['metadatas']
-        documents = results['documents']
+        ids = results['ids'][0]
+        distances = results['distances'][0]
+        metadatas = results['metadatas'][0]
+        documents = results['documents'][0]
 
         for index, id in enumerate(ids):