rebuild the ternary plot w cliponaxis=False

Rock_fact_checker.py

@@ -4,8 +4,6 @@ import logging
 from json import JSONDecodeError
 
 import streamlit as st
-import pandas as pd
-import plotly.express as px
 
 from app_utils.backend_utils import load_statements, query
 from app_utils.frontend_utils import (
@@ -13,6 +11,7 @@ from app_utils.frontend_utils import (
     reset_results,
     entailment_html_messages,
     create_df_for_relevant_snippets,
+    create_ternary_plot,
 )
 from app_utils.config import RETRIEVER_TOP_K
 
@@ -59,12 +58,14 @@ def main():
         # Re-runs the script setting the random statement as the textbox value
         # Unfortunately necessary as the Random statement button is _below_ the textbox
         # Adapted for Streamlit>=1.12
-        if hasattr(st, 'scriptrunner'):
-            raise st.scriptrunner.script_runner.RerunException(st.scriptrunner.script_requests.RerunData(""))
+        if hasattr(st, "scriptrunner"):
+            raise st.scriptrunner.script_runner.RerunException(
+                st.scriptrunner.script_requests.RerunData("")
+            )
         else:
             raise st.runtime.scriptrunner.script_runner.RerunException(
-            st.runtime.scriptrunner.script_requests.RerunData("") 
-        )
+                st.runtime.scriptrunner.script_requests.RerunData("")
+            )
     else:
         st.session_state.random_statement_requested = False
     run_query = (
@@ -79,7 +80,7 @@ def main():
         with st.spinner("🧠    Performing neural search on documents..."):
             try:
                 st.session_state.results = query(statement, RETRIEVER_TOP_K)
-                print(query)
+                print(statement)
                 time_end = time.time()
                 print(time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime()))
                 print(f"elapsed time: {time_end - time_start}")
@@ -105,20 +106,14 @@ def main():
 
         st.markdown(f"###### Aggregate entailment information:")
         col1, col2 = st.columns([2, 1])
-        df_agg_entailment_info = pd.DataFrame([results["agg_entailment_info"]])
-        fig = px.scatter_ternary(
-            df_agg_entailment_info,
-            a="contradiction",
-            b="neutral",
-            c="entailment",
-            size="contradiction",
-        )
+        agg_entailment_info = results["agg_entailment_info"]
+        fig = create_ternary_plot(agg_entailment_info)
         with col1:
             st.plotly_chart(fig, use_container_width=True)
         with col2:
             st.write(results["agg_entailment_info"])
 
-        st.markdown(f"###### Relevant snippets:")
+        st.markdown(f"###### Most Relevant snippets:")
         df, urls = create_df_for_relevant_snippets(docs)
         st.dataframe(df)
 

app_utils/config.py

@@ -11,7 +11,8 @@ RETRIEVER_TOP_K = 5
 
 # In HF Space, we use microsoft/deberta-v2-xlarge-mnli
 # for local testing, a smaller model is better
-NLI_MODEL = "valhalla/distilbart-mnli-12-1"
-if hasattr(st, 'secrets'):
-    NLI_MODEL = st.secrets['NLI_MODEL']
-print(f'Used NLI model: {NLI_MODEL}')
+try:
+    NLI_MODEL = st.secrets["NLI_MODEL"]
+except:
+    NLI_MODEL = "valhalla/distilbart-mnli-12-1"
+print(f"Used NLI model: {NLI_MODEL}")

app_utils/frontend_utils.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import pandas as pd
+import plotly.graph_objects as go
+
 
 entailment_html_messages = {
     "entailment": 'The knowledge base seems to <span style="color:green">confirm</span> your statement',
@@ -20,6 +22,57 @@ def reset_results(*args):
     st.session_state.raw_json = None
 
 
+def create_ternary_plot(entailment_data):
+    hover_text = ""
+    for label, value in entailment_data.items():
+        hover_text += f"{label}: {value}<br>"
+
+    fig = go.Figure(
+        go.Scatterternary(
+            {
+                "cliponaxis": False,
+                "mode": "markers",
+                "a": [i for i in map(lambda x: x["entailment"], [entailment_data])],
+                "b": [i for i in map(lambda x: x["contradiction"], [entailment_data])],
+                "c": [i for i in map(lambda x: x["neutral"], [entailment_data])],
+                "hoverinfo": "text",
+                "text": hover_text,
+                "marker": {
+                    "color": "#636efa",
+                    "size": [0.01],
+                    "sizemode": "area",
+                    "sizeref": 2.5e-05,
+                    "symbol": "circle",
+                },
+            }
+        )
+    )
+
+    fig.update_layout(
+        {
+            "ternary": {
+                "sum": 1,
+                "aaxis": makeAxis("Entailment", 0),
+                "baxis": makeAxis("<br>Contradiction", 45),
+                "caxis": makeAxis("<br>Neutral", -45),
+            }
+        }
+    )
+    return fig
+
+
+def makeAxis(title, tickangle):
+    return {
+        "title": title,
+        "titlefont": {"size": 20},
+        "tickangle": tickangle,
+        "tickcolor": "rgba(0,0,0,0)",
+        "ticklen": 5,
+        "showline": False,
+        "showgrid": True,
+    }
+
+
 def highlight_cols(s):
     coldict = {"con": "#FFA07A", "neu": "#E5E4E2", "ent": "#a9d39e"}
     if s.name in coldict.keys():

data/statements.txt

@@ -29,7 +29,7 @@ Sum 41 were originally called Kaspir
 Bruce Springsteen has been named "the buzz"
 Talking Heads collaborated with Brian Eno
 Chris Cornell took part in Soundgarden and Audioslave
-Chris Cornell took part in Pearl Jam
+Chris Cornell was a member of Pearl Jam
 "There is a light that never goes out" is a song by The Smiths
 Guns N' Roses formed in 1987
 Izzy Stradlin took part in Guns N' Roses