Update app.py

app.py

@@ -23,14 +23,13 @@ DEFAULT_QUESTION_AT_STARTUP = os.getenv("DEFAULT_QUESTION_AT_STARTUP", "What's t
 DEFAULT_ANSWER_AT_STARTUP = os.getenv("DEFAULT_ANSWER_AT_STARTUP", "Paris")
 
 def place_header_center(text, lottie_data):
-    img, title= st.columns([3,3])
+    img, title= st.columns([1,3])
     with img:
-        st_lottie(lottie_data, height=250)
+        st_lottie(lottie_data, height=150)
     with title:
         st.title(text)
 
 
-
 @st.experimental_memo
 def get_lottie(path):
     with open(path, 'r', errors='ignore') as f:
@@ -46,6 +45,36 @@ def load_and_write_data(document_store):
     document_store.write_documents(docs)
 
 
+def get_backlink(result):
+    if result.get("document", None):
+        doc = result["document"]
+        if isinstance(doc, dict):
+            if doc.get("meta", None):
+                if isinstance(doc["meta"], dict):
+                    if doc["meta"].get("url", None):
+                        return doc["meta"]["url"]
+    return None
+
+
+def get_doc_name(result):
+    if result.get("document", None):
+        doc = result["document"]
+        if isinstance(doc, dict):
+            if doc.get("meta", None):
+                if isinstance(doc["meta"], dict):
+                    if doc["meta"].get("name", None):
+                        return doc["meta"]["name"]
+    return None
+
+def get_doc_reg_id(result):
+    if result.get("document", None):
+        doc = result["document"]
+        if isinstance(doc, dict):
+            if doc.get("meta", None):
+                if isinstance(doc["meta"], dict):
+                    if doc["meta"].get("reg_id", None):
+                        return doc["meta"]["reg_id"]
+    return None
 # Haystack Components
 # @st.cache(allow_output_mutation=True)
 # def start_haystack():
@@ -94,6 +123,8 @@ def ask_question(question):
                     "context": "..." + answer["context"] + "...",
                     "answer": answer["answer"],
                     "relevance": round(answer["score"] * 100, 2),
+                    "document": [doc for doc in response["documents"] if doc["id"] == answer["document_id"]][0],
+                    # "_raw": answer,
                     "offset_start_in_doc": answer["offsets_in_document"][0]["start"],
                 }
             )
@@ -129,7 +160,21 @@ if st.session_state.results:
                 markdown(context[:start_idx] + str(annotation(body=answer, label="ANSWER", background="#ff700f", color='#ffffff')) + context[end_idx:]),
                 unsafe_allow_html=True,
             )
-            st.markdown(f"**Relevance:** {result['relevance']}")
+            source = ""
+            url = get_backlink(result)
+            name = get_doc_name(result)
+            reg_id = get_doc_reg_id(result)
+            if name:
+                source += f"[{result['document']['meta']['name']}]"
+            
+            if url:
+                source += f"({result['document']['meta']['url']})"
+
+            if reg_id:
+                source += f"({result['document']['meta']['reg_id']})"
+
+            st.markdown(f"**Relevance:** {result['relevance']} -  **Source:** {source}")
+            
         else:
             st.info(
                 "🤔    Haystack is unsure whether any of the documents contain an answer to your question. Try to reformulate it!"