formatted metadata

app.py

@@ -34,10 +34,10 @@ def query_chroma(collection, embedding, authors):
     for id_, metadata, document_text, distance in zip(ids, metadatas, documents, distances):
         result_dict = {
             "id": id_,
-            "author": metadata.get('author', 'Unknown author'),
-            "book": metadata.get('book', 'Unknown book'),
-            "section": metadata.get('section', 'Unknown section'),
-            "title": metadata.get('title', 'Untitled'),
+            "author": metadata.get('author', ''),
+            "book": metadata.get('book', ''),
+            "section": metadata.get('section', ''),
+            "title": metadata.get('title', ''),
             "text": document_text,
             "distance": distance
         }
@@ -84,10 +84,10 @@ custom_css = """
 """
 
 with gr.Blocks(theme=theme, css=custom_css) as demo:
-    gr.Markdown("Geben Sie ein, wonach Sie suchen möchten (Query), filtern Sie nach Autoren (ohne Auswahl werden alle durchsucht) und klicken Sie auf **Suchen**, um zu suchen. Trennen Sie mehrere Fragen durch Semikola; die Suche dauert, unabhängig von der Anzahl der Abfragen, etwa 40 Sekunden, da das Embeddingmodell jedes Mal auf eine GPU geladen werden muss.")
+    gr.Markdown("Geben Sie ein, wonach Sie suchen möchten (Query), trennen Sie mehrere Suchanfragen durch Semikola; filtern Sie nach Autoren (ohne Auswahl werden alle durchsucht) und klicken Sie auf **Suchen**, um zu suchen.")
     #database_inp = gr.Dropdown(label="Database", choices=["German", "English"], value="German")
     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?; ...")
+    inp = gr.Textbox(label="Query", lines=3, 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()
@@ -102,7 +102,7 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
     #)
 
     def perform_query(queries, authors):
-        task = "Given a question, retrieve passages that answer the question"
+        task = "Suche den zur Frage passenden Text"
         queries = [query.strip() for query in queries.split(';')]
         embeddings = get_embeddings(queries, task)
         collection = collection_de
@@ -113,7 +113,7 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
         return results_data, ""
 
     btn.click(
-        fn=lambda: ("Searching... This may take up to 40 seconds.", gr.update(visible=True)),
+        fn=lambda: ("Suche... Das kann bis zu 40 Sekunden dauern...", gr.update(visible=True)),
         inputs=None,
         outputs=[loading_indicator, loading_indicator],
         queue=False
@@ -129,10 +129,24 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
             with gr.Accordion(query, open=False) as acc:
                 for result in res:
                     with gr.Column():
-                        author = result.get('author', 'Unknown author')
-                        book = result.get('book', 'Unknown book')
-                        text = result.get('text')
-                        markdown_contents = f"**{author}, {book}**\n\n{text}"
+                        author = str(result.get('author', ''))
+                        book = str(result.get('book', ''))
+                        section = str(result.get('section', ''))
+                        title = str(result.get('title', ''))
+                        text = str(result.get('text', ''))
+
+                        header_parts = []
+                        if author and author != "Unknown":
+                            header_parts.append(author)
+                        if book and book != "Unknown":
+                            header_parts.append(book)
+                        if section and section != "Unknown":
+                            header_parts.append(section)
+                        if title and title != "Unknown":
+                            header_parts.append(title)
+                        
+                        header = ", ".join(header_parts)
+                        markdown_contents = f"**{header}**\n\n{text}"
                         gr.Markdown(value=markdown_contents, elem_classes="custom-markdown")
 
 demo.launch()