english only now

app.py

@@ -5,9 +5,9 @@ from sentence_transformers import SentenceTransformer
 import spaces
 
 client = chromadb.PersistentClient(path="./chroma")
-collection_de = client.get_collection(name="phil_de")
+#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_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"]
 
 @spaces.GPU
@@ -47,26 +47,27 @@ def query_chroma(collection, embedding, authors):
 
 with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; border-radius: 5px; }")  as demo:
     gr.Markdown("Enter your query, filter authors (default is all), click **Search** to search. Delimit multiple queries with semicola; since there is a quota for each user (based on IP) it makes sense to query in batches. The search takes around 40 seconds, regardless of the number of queries, because the embedding model needs to be loaded to a GPU each time.")
-    database_inp = gr.Dropdown(label="Database", choices=["German", "English"], value="German")
-    author_inp = gr.Dropdown(label="Authors", choices=authors_list_de, multiselect=True)
-    inp = gr.Textbox(label="Query", placeholder="Wie kann ich gesund leben und bedeutet Gesundheit für jeden das gleiche?; Was ist der Sinn des Lebens?; ...")
+    #database_inp = gr.Dropdown(label="Database", choices=["German", "English"], value="German")
+    author_inp = gr.Dropdown(label="Authors", choices=authors_list_en, multiselect=True)
+    inp = gr.Textbox(label="Query", placeholder="How can I live a healthy life?; How can I improve my ability to focus?; What is the meaning of life?; ...")
     btn = gr.Button("Search")
     results = gr.State()
 
-    def update_authors(database):
-        return gr.update(choices=authors_list_de if database == "German" else authors_list_en)
+    #def update_authors(database):
+    #    return gr.update(choices=authors_list_de if database == "German" else authors_list_en)
 
-    database_inp.change(
-        fn=lambda database: update_authors(database),
-        inputs=[database_inp],
-        outputs=[author_inp]
-    )
+    #database_inp.change(
+    #    fn=lambda database: update_authors(database),
+    #    inputs=[database_inp],
+    #    outputs=[author_inp]
+    #)
 
     def perform_query(queries, authors, database):
         task = "Given a question, retrieve passages that answer the question"
-        queries = queries.split(';')
+        queries = [query.strip() for query in queries.split(';')]
         embeddings = get_embeddings(queries, task)
-        collection = collection_de if database == "German" else collection_en
+        #collection = collection_de if database == "German" else collection_en
+        collection = collection_en
         results_data = []
         for query, embedding in zip(queries, embeddings):
             res = query_chroma(collection, embedding, authors)
@@ -75,7 +76,7 @@ with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; bo
 
     btn.click(
         perform_query,
-        inputs=[inp, author_inp, database_inp],
+        inputs=[inp, author_inp],
         outputs=[results]
     )