Update app.py

app.py

@@ -19,8 +19,8 @@ DOCS_PATH = os.path.join(DATA_DIR, 'all_docs_36838.pkl')
 LOTTIE_PATH = './img/108423-search-for-documents.json'
 PROG_TITLE = "QA project Demo"
 # Adjust to a question that you would like users to see in the search bar when they load the UI:
-DEFAULT_QUESTION_AT_STARTUP = os.getenv("DEFAULT_QUESTION_AT_STARTUP", "What's the capital of France?")
-DEFAULT_ANSWER_AT_STARTUP = os.getenv("DEFAULT_ANSWER_AT_STARTUP", "Paris")
+DEFAULT_QUESTION_AT_STARTUP = os.getenv("DEFAULT_QUESTION_AT_STARTUP", "Что делает Домашняя бухгалтерия?")
+DEFAULT_ANSWER_AT_STARTUP = os.getenv("DEFAULT_ANSWER_AT_STARTUP", "Домашняя бухгалтерия позволяет вести счета в разных валютах")
 
 def place_header_center(text, lottie_data):
     img, title= st.columns([1,3])
@@ -94,7 +94,7 @@ def set_state_if_absent(key, value):
 set_state_if_absent("question", DEFAULT_QUESTION_AT_STARTUP)
 set_state_if_absent("answer", DEFAULT_ANSWER_AT_STARTUP)
 set_state_if_absent("results", None)
-
+set_state_if_absent("predictions", None)
 
 def reset_results(*args):
     st.session_state.results = None
@@ -114,18 +114,20 @@ question = st.text_input("", value=st.session_state.question, max_chars=100, on_
 
 def ask_question(question):
     prediction = pipeline.run(query=question, params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}})
+    answers = prediction["answers"]
     results = []
-    for answer in prediction["answers"]:
+    for answer in answers:
         answer = answer.to_dict()
-        if answer["answer"]:
+        if answer.get("answer", None):
             results.append(
                 {
                     "context": "..." + answer["context"] + "...",
-                    "answer": answer["answer"],
+                    "answer": answer.get("answer", None),
+                    "source": answer["meta"]["name"],
                     "relevance": round(answer["score"] * 100, 2),
-                    "document": [doc for doc in prediction["documents"] if doc["id"] == answer["document_id"]][0],
-                    # "_raw": answer,
+                    "document": [doc for doc in response["documents"] if doc["id"] == answer["document_id"]][0],
                     "offset_start_in_doc": answer["offsets_in_document"][0]["start"],
+                    "_raw": answer,
                 }
             )
         else:
@@ -133,10 +135,12 @@ def ask_question(question):
                 {
                     "context": None,
                     "answer": None,
+                    "document": None,
                     "relevance": round(answer["score"] * 100, 2),
+                    "_raw": answer,
                 }
             )
-    return results
+    return results, prediction
 
 
 if question:
@@ -144,7 +148,7 @@ if question:
         try:
             msg = 'Asked ' + question
             logging.info(msg)
-            st.session_state.results = ask_question(question)
+            st.session_state.results, st.session_state.predictions = ask_question(question)
         except Exception as e:
             logging.exception(e)
 
@@ -172,8 +176,10 @@ if st.session_state.results:
 
             if reg_id:
                 source += f"({result['document']['meta']['reg_id']})"
-
-            st.markdown(f"**Relevance:** {result['relevance']} -  **Source:** {source}")
+            if source:
+                st.markdown(f"**Relevance:** {result['relevance']} -  **Source:** {source}")
+            else:
+                st.markdown(f"**Relevance:** {result['relevance']}")
             
         else:
             st.info(