Update app.py

app.py

@@ -4,55 +4,43 @@ import json
 from normalizer import process_df
 from process_tags import list_ents, no_of_tags, color_creator, scatter_document
 
-tags_data = {
-"_dtags_": [],
-"DEP": [],
-"dep_explain": {},
-"_ptags_": [],
-"POS": [],
-"pos_explain": {},
-"_ntags_": ["concept", "tool", "framework", "crew", "position", "measurement", "artifact", "datasheet", "briefing", "standard", "insight", "client", "output", "output_component", "system", "challenge", "environment", "process", "process_component", "activity", "business_need", "third_party", "agreement", "raw_data", "authority", "resource_pool", "field_of_study", "geo_loc", "TBD"],
-"NER": ["concept", "#e3648a", "tool", "#d56b2e", "framework", "#ff0000", "crew", "#ffafcc", "position", "#a2cd5a", "measurement", "#779be7", "artifact", "#bee1e6", "datasheet", "#7e7d61", "briefing", "#4b8d76", "standard", "#9381ff", "insight", "#ffc125", "client", "#ffc6ff", "output", "#ffcfd2", "output_component", "#fde4cf", "system", "#656e1b", "challenge", "#ff5400", "environment", "#affc41", "process", "#adc178", "process_component", "#dde5b6", "activity", "#cfbaf0", "business_need", "#ced4da", "third_party", "#ee00ee", "agreement", "#484676", "raw_data", "#cdad00", "authority", "#d689c4", "resource_pool", "#8ea44c", "field_of_study", "#c39fe9", "geo_loc", "#c0ff3e", "TBD", "#ccdbfd"],
-"ner_explain": {},
-"_ctags_": [],
-"CAT": [],
-"cat_explain": {}
-}
 
+def main():
+    st.set_page_config(layout='wide')
+    st.write(
+        "Tags Json is the file we get from exporting tags from d555 tag editor. You may need to format it to be in proper json format. For the demo, use this Tag json file https://github.com/kandysh/odin_json_sheets/blob/main/tag_color.json")
+    with st.sidebar:
+        uploaded_file = st.file_uploader("Upload the SENTENCE Json", type="json")
+        uploaded_color = st.file_uploader("Upload the TAG Json", type="json")
+    if uploaded_file and uploaded_color is not None:
+        raw_data = json.load(uploaded_file)
+        tags_data = json.load(uploaded_color)
+        st.title(f'{uploaded_file.name.split(".")[0].upper()}')
+        df_list = [process_df(data) for data in raw_data]
+        st.plotly_chart(scatter_document(df_list, tags_data), use_container_width=True)
+        key = 0
+        for df in df_list:
+            ents = list_ents(df)
+            tags = list(no_of_tags(df).keys())
+            doc = [{
+                "text": ' '.join(df['words']),
+                "ents": ents,
+                "title": None
+            }]
+            st.text(f"Sentence {key}")
+            spacy_streamlit.visualize_ner(
+                doc,
+                labels=tags,
+                show_table=False,
+                title=None,
+                manual=True,
+                displacy_options={
+                    "colors": color_creator(tags_data["NER"])
+                },
+                key=f"{key}"
+            )
+            key += 1
 
 
-st.set_page_config(layout='wide')
-st.title("My Tag Visualizer")
-with st.sidebar:
-    uploaded_file = st.file_uploader("Upload the sentence Json", type="json")
-    # uploaded_color = st.file_uploader("Upload the tag Json", type="json")
-if uploaded_file is not None:
-    raw_data = json.load(uploaded_file)
-    # tags_data = json.load(uploaded_color)
-    df_list = []
-    key = 0
-    for data in raw_data:
-        df_list.append(process_df(data))
-    st.plotly_chart(scatter_document(df_list, tags_data), use_container_width=True)
-    tags = []
-    ent = []
-    for df in df_list:
-        ent = list_ents(df)
-        tags = list(no_of_tags(df).keys())
-        doc = [{
-            "text": ' '.join(df['words']),
-            "ents": ent,
-            "title": None
-        }]
-        spacy_streamlit.visualize_ner(
-            doc,
-            labels=tags,
-            show_table=False,
-            title=None,
-            manual=True,
-            displacy_options={
-                "colors": color_creator(tags_data["NER"])
-            },
-            key=f"{key}"
-        )
-        key += 1
+if __name__ == "__main__":
+    main()