added loading

app.py

@@ -51,6 +51,7 @@ with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; bo
     author_inp = gr.Dropdown(label="Autoren", choices=authors_list_de, multiselect=True)
     inp = gr.Textbox(label="Query", placeholder="Wie kann ich gesund leben?; Wie kann ich mich besser konzentrieren?; Was ist der Sinn des Lebens?; ...")
     btn = gr.Button("Suchen")
+    loading_indicator = gr.Markdown(visible=False)
     results = gr.State()
 
     #def update_authors(database):
@@ -62,22 +63,26 @@ with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; bo
     #    outputs=[author_inp]
     #)
 
-    def perform_query(queries, authors, database):
+    def perform_query(queries, authors):
         task = "Given a question, retrieve passages that answer the question"
         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
         results_data = []
         for query, embedding in zip(queries, embeddings):
             res = query_chroma(collection, embedding, authors)
             results_data.append((query, res))
-        return results_data
+        return results_data, ""
 
     btn.click(
+        fn=lambda: ("Searching... This may take up to 40 seconds.", gr.update(visible=True)),
+        inputs=None,
+        outputs=[loading_indicator, loading_indicator],
+        queue=False
+    ).then(
         perform_query,
         inputs=[inp, author_inp],
-        outputs=[results]
+        outputs=[results, loading_indicator]
     )
 
     @gr.render(inputs=[results])