Refactor with expander

app.py

@@ -41,17 +41,18 @@ def load_page(split):
             meta = data["meta"]
             with st.expander("Render Content"):
                 st.write(content)
-            st.write("### Content:")
-            st.text(content)
-            st.write("### Meta:")
-            st.write(ast.literal_eval(meta))
-            #Tokenizer related count
-            tokenized =  tokenizer(content, return_length=True)['length'][0]
-            token_count_metric = st.metric("Token Count(compared to 2048)",value=tokenized,delta=4096-tokenized)
+            with st.expander("Raw Content"):
+                st.text(content)
+                st.write("### Meta:")
+                st.write(ast.literal_eval(meta))
+            # Tokenizer-related count
+            with st.expander("Metrics"):
+                tokenized =  tokenizer(content, return_length=True)['length'][0]
+                token_count_metric = st.metric("Token Count(compared to 2048)",value=tokenized,delta=4096-tokenized)
             #Word related count
-            split_words = re.findall(r'\w+', content)
-            word_count_metric = st.metric("Word Count",value=len(split_words))
-            
+                split_words = re.findall(r'\w+', content)
+                word_count_metric = st.metric("Word Count",value=len(split_words))
+                
 
 
 demo_name = st.sidebar.selectbox("Choose a demo", splits)