jetzt aber

app.py

@@ -1,9 +1,7 @@
 import os
 import gradio as gr
 import chromadb
-
 from sentence_transformers import SentenceTransformer
-
 import spaces
 
 @spaces.GPU
@@ -15,12 +13,12 @@ def get_embeddings(query, task):
 
 # Initialize a persistent Chroma client and retrieve collection
 client = chromadb.PersistentClient(path="./chroma")
-collection = client.get_collection(name="phil_de")
-
-authors_list = ["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 = ["Friedrich Nietzsche", "Joscha Bach"]
+collection_de = client.get_collection(name="phil_de")
+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(embeddings, authors, num_results=10):
+def query_chroma(collection, embeddings, authors, num_results=10):
     try:
         where_filter = {"author": {"$in": authors}} if authors else {}
 
@@ -57,10 +55,11 @@ def query_chroma(embeddings, authors, num_results=10):
 
 
 # Main function
-def perform_query(query, authors, num_results):
+def perform_query(query, authors, num_results, database):
     task = "Given a question, retrieve passages that answer the question"
     embeddings = get_embeddings(query, task)
-    results = query_chroma(embeddings, authors, num_results)
+    collection = collection_de if database == "German" else collection_en
+    results = query_chroma(collection, embeddings, authors, num_results)
 
     if "error" in results:
         return [gr.update(visible=True, value=f"Error: {results['error']}") for _ in range(max_textboxes * 2)]
@@ -74,6 +73,9 @@ def perform_query(query, authors, num_results):
 
     return updates
 
+def update_authors(database):
+    return gr.update(choices=authors_list_de if database == "German" else authors_list_en)
+
 # Gradio interface
 max_textboxes = 30
 
@@ -81,25 +83,29 @@ with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; bo
     gr.Markdown("Enter your query, filter authors (default is all), click **Search** to search. Click **Flag** if a result is relevant to the query and interesting to you.")
     with gr.Row():
         with gr.Column():
-            inp = gr.Textbox(label="query", placeholder="Enter thought...")
-            author_inp = gr.Dropdown(label="authors", choices=authors_list, multiselect=True)
+            database_inp = gr.Dropdown(label="Database", choices=["English", "German"], value="German")
+            inp = gr.Textbox(label="query", placeholder="Enter question...")
+            author_inp = gr.Dropdown(label="authors", choices=authors_list_de, multiselect=True)
             num_results_inp = gr.Number(label="number of results", value=10, step=1, minimum=1, maximum=max_textboxes)
             btn = gr.Button("Search")
 
     components = []
-    textboxes = []
 
     for _ in range(max_textboxes):
         with gr.Column() as col:
             text_out = gr.Markdown(visible=False, elem_classes="custom-markdown")
             components.append(text_out)
-            textboxes.append(text_out)
 
     btn.click(
         fn=perform_query, 
-        inputs=[inp, author_inp, num_results_inp],
+        inputs=[inp, author_inp, num_results_inp, database_inp],
         outputs=components
     )
     
+    database_inp.change(
+    fn=update_authors,
+    inputs=database_inp,
+    outputs=author_inp
+    )
 
 demo.launch()