Refactor:examples.json and utils.py

app.py

@@ -1,20 +1,21 @@
-from os import write
-from typing import Sequence
 import streamlit as st
 from hf_model import classifier_zero,load_model
-from utils import plot_result
+from utils import plot_result,examples_load
 import json
+
 classifier=load_model()
-with open("examples.json") as f:
-    data=json.load(f)
+ex_text,ex_labels=examples_load()
+
+# with open("examples.json") as f:
+#     data=json.load(f)
 
 if __name__ == '__main__':
     st.header("Zero Shot Classification")
     st.write("This app allows you to classify any text into any categories you are interested in.")
 
     with st.form(key='my_form'):
-        text_input = st.text_area("Input Text",data['text'])
-        labels = st.text_input('Possible topics (separated by `,`)',data["labels"], max_chars=1000)
+        text_input = st.text_area("Input Text",ex_text)
+        labels = st.text_input('Possible topics (separated by `,`)',ex_labels, max_chars=1000)
         labels = list(set([x.strip() for x in labels.strip().split(',') if len(x.strip()) > 0]))
         radio = st.radio("Select Multiclass",('Only one topic can be corect at a time','Multiple topics can be correct at a time'),)
         multi_class= True if radio=="Multiple topics can be correct at a time" else False

examples.json

@@ -1,4 +1,4 @@
 {
 "text":"The Democratic president had just signed into law the most significant infrastructure package in generations. And he had done it by bringing Democrats and Republicans together.",
-"labels":["Economy", "Politics", "Environment", "Entertainment"]
+"labels":"Economy,Politics,Environment,Entertainment"
 }

utils.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import numpy as np
 import plotly.express as px
+import json
+
 def plot_result(top_topics, scores):
     top_topics = np.array(top_topics)
     scores = np.array(scores)
@@ -14,4 +16,10 @@ def plot_result(top_topics, scores):
                  color_continuous_scale='GnBu')
     fig.update(layout_coloraxis_showscale=False)
     fig.update_traces(texttemplate='%{text:0.1f}%', textposition='outside')
-    st.plotly_chart(fig)
+    st.plotly_chart(fig)
+
+def examples_load():
+    with open("examples.json") as f:
+        data=json.load(f)
+    
+    return data['text'],data['labels']