add some data visualisations

stream_lit.py

@@ -1,17 +1,16 @@
 import streamlit as st
-# from main import make_output
-# from summarization import make_summary
+from collections import Counter
 import textrazor
 from dotenv import load_dotenv
 import os
 import pandas as pd
 import replicate
-
+import plotly.express as px
 
 load_dotenv()
 REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
 
-# text_ = """Alice: Hi there, is this Bob?
+text_ = """Alice: Hi there, is this Bob?
 # Bob: Yes, speaking. Who's calling?
 # Alice: Hey Bob, it's Alice from Acme Inc. We met at the conference last month.
 # Bob: Oh, hey Alice! Good to hear from you. How can I help you today?
@@ -54,23 +53,43 @@ def make_output(text:str):
         # df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
         df = pd.concat([df, df2], ignore_index=True)
         # print(entity.id)
-        df = df.drop_duplicates()
+        df1 = df.drop_duplicates()
+
     return(df)
 
 st.title("CultureScout NLP Tool 🤖")
 
 # taking user inputs for context search
-st.write("Enter Text You Need Help With:")
+st.write("***Enter Text You Need Help With:***")
 user_input = st.text_input("Text Here:", "")
 
 if st.button("🔎 Search It!"):
     def predict_sentiment(data:str):
         ans = make_output(user_input)
         return ans
-    st.table(predict_sentiment(user_input))
+    df = predict_sentiment(user_input)
+    df = df.drop_duplicates()
+    st.table(df)
+
+    id_df = predict_sentiment(user_input)['id']
+    id_freq = id_df.value_counts()
+    most_common_id = id_df.value_counts().index[0]
+
+    type_df = predict_sentiment(user_input)['type']
+    type_freq = type_df.value_counts()
+    most_common_type = type_df.value_counts().index[0]
+
+
+    st.bar_chart(id_freq)
+    st.bar_chart(type_freq)
+
+    st.write(f"Most appeared id is {most_common_id}.")
+    st.write(f"Most appeared type is {most_common_type}.")
+    st.write("""
+             """)
 
 # taking user inputs for summarization
-st.write("Enter Text You Need to Summarize:")
+st.write("***Enter Text You Need to Summarize:***")
 user_input1 = st.text_area("Text Here:", "")
 
 if st.button("🪄 Summarize"):