working

app.py

@@ -18,14 +18,12 @@ collection_en = client.get_collection(name="phil_en")
 authors_list_de = ["Ludwig Wittgenstein", "Sigmund Freud", "Marcus Aurelius", "Friedrich Nietzsche", "Epiktet", "Ernst Jünger", "Georg Christoph Lichtenberg", "Balthasar Gracian", "Hannah Arendt", "Erich Fromm", "Albert Camus"]
 authors_list_en = ["Friedrich Nietzsche", "Joscha Bach"]
 
-def query_chroma(collection, embeddings, authors):
+def query_chroma(collection, embedding, authors):
     try:
         where_filter = {"author": {"$in": authors}} if authors else {}
-
-        embeddings_list = embeddings[0].tolist()
-
+        # Directly use the embedding provided, already in list format suitable for the query
         results = collection.query(
-            query_embeddings=[embeddings_list],
+            query_embeddings=[embedding.tolist()],  # Ensure embedding is properly formatted
             n_results=10,
             where=where_filter,
             include=["documents", "metadatas", "distances"]
@@ -51,7 +49,7 @@ def query_chroma(collection, embeddings, authors):
 
         return formatted_results
     except Exception as e:
-        return {"error": str(e)}
+        return [{"error": str(e)}]
 
 def update_authors(database):
     return gr.update(choices=authors_list_de if database == "German" else authors_list_en)
@@ -64,26 +62,39 @@ with gr.Blocks() as demo:
     author_inp = gr.Dropdown(label="Authors", choices=authors_list_de, multiselect=True)
     inp = gr.Textbox(label="Query", placeholder="Enter questions separated by semicolons...")
     btn = gr.Button("Search")
+    results = gr.State()  # Store results in a State component
 
     def perform_query(queries, authors, database):
-        queries = queries.split(';')
         task = "Given a question, retrieve passages that answer the question"
+        queries = queries.split(';')
         embeddings = get_embeddings(queries, task)
         collection = collection_de if database == "German" else collection_en
-
-        results = [query_chroma(collection, embedding, authors) for embedding in embeddings]
-
-        for query, result in zip(queries, results):
-            with gr.Accordion(query):
-                markdown_contents = "\n".join(f"**{res['author']}, {res['book']}**\n\n{res['text']}" for res in result)
-                gr.Markdown(value=markdown_contents)
+        results_data = []
+        for query, embedding in zip(queries, embeddings):
+            res = query_chroma(collection, embedding, authors)
+            results_data.append((query, res))
+        return results_data
 
     btn.click(
         perform_query,
         inputs=[inp, author_inp, database_inp],
-        outputs=[]
+        outputs=[results]
     )
 
+    @gr.render(inputs=[results])
+    def display_accordion(data):
+        output_blocks = []
+        for query, res in data:
+            with gr.Accordion(query) as acc:
+                if not res:
+                    markdown_contents = "No results found."
+                elif "error" in res[0]:
+                    markdown_contents = f"Error retrieving data: {res[0]['error']}"
+                else:
+                    markdown_contents = "\n".join(f"**{r['author']}, {r['book']}**\n\n{r['text']}" for r in res)
+                gr.Markdown(markdown_contents)
+
+
     database_inp.change(
         fn=lambda database: update_authors(database),
         inputs=[database_inp],