update streamlit and other libs; better caching; fix graphical appearance

README.md

@@ -4,7 +4,7 @@ emoji: 🎸
 colorFrom: purple
 colorTo: blue
 sdk: streamlit
-sdk_version: 1.10.0
+sdk_version: 1.19.0
 app_file: Rock_fact_checker.py
 pinned: true
 models: [sentence-transformers/msmarco-distilbert-base-tas-b, microsoft/deberta-v2-xlarge-mnli, google/flan-t5-large]

Rock_fact_checker.py

@@ -109,7 +109,8 @@ def main():
         col1, col2 = st.columns([2, 1])
         fig = create_ternary_plot(agg_entailment_info)
         with col1:
-            st.plotly_chart(fig, use_container_width=True)
+            # theme=None helps to preserve default plotly colors
+            st.plotly_chart(fig, use_container_width=True, theme=None)
         with col2:
             st.write(agg_entailment_info)
 

app_utils/backend_utils.py

@@ -18,7 +18,7 @@ from app_utils.config import (
 )
 
 
-@st.cache()
+@st.cache_data
 def load_statements():
     """Load statements from file"""
     with open(STATEMENTS_PATH) as fin:
@@ -29,9 +29,7 @@ def load_statements():
 
 
 # cached to make index and models load only at start
-@st.cache(
-    hash_funcs={"builtins.SwigPyObject": lambda _: None}, allow_output_mutation=True
-)
+@st.cache_resource
 def start_haystack():
     """
     load document store, retriever, entailment checker and create pipeline
@@ -66,16 +64,14 @@ pipe, prompt_node = start_haystack()
 
 # the pipeline is not included as parameter of the following function,
 # because it is difficult to cache
-@st.cache(allow_output_mutation=True, ttl=36000)
+@st.cache_resource
 def check_statement(statement: str, retriever_top_k: int = 5):
     """Run query and verify statement"""
     params = {"retriever": {"top_k": retriever_top_k}}
     return pipe.run(statement, params=params)
 
 
-@st.cache(
-    hash_funcs={"tokenizers.Tokenizer": lambda _: None}, allow_output_mutation=True
-)
+@st.cache_resource
 def explain_using_llm(
     statement: str, documents: List[Document], entailment_or_contradiction: str
 ) -> str:

app_utils/frontend_utils.py

@@ -1,7 +1,10 @@
 import streamlit as st
 import pandas as pd
 import plotly.graph_objects as go
+import plotly.io as pio
 
+# this helps to preserve default plotly colors
+pio.templates.default = "plotly"
 
 entailment_html_messages = {
     "entailment": 'The knowledge base seems to <span style="color:green">confirm</span> your statement',