Spaces:
Runtime error
Runtime error
File size: 3,676 Bytes
67e4208 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
import streamlit as st
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?
# 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?
# Alice: Well, I was just calling because I wanted to touch base with you about that project we discussed at the conference. I was hoping we could set up a meeting to talk about it in more detail.
# Bob: Absolutely, I'd be happy to. When were you thinking?
# Alice: How about next Tuesday at 10 am?
# Bob: That works for me. Where should we meet?
# Alice: We can meet at our office in downtown. Here's the address: 123 Main St. Suite 400.
# Bob: Great. And just to confirm, your mobile number is still (555) 123-4567, right?
# Alice: Yes, that's correct.
# Bob: Perfect. I'll put the meeting in my calendar and send you a confirmation email with all the details.
# Alice: Sounds good, thanks Bob. Looking forward to it!"""
replicate.Client(api_token=REPLICATE_API_TOKEN)
def make_summary(text_):
output = replicate.run(
"replicate/flan-t5-xl:7a216605843d87f5426a10d2cc6940485a232336ed04d655ef86b91e020e9210",
input={"prompt": """Write the summary extract any useful information like name, number, and organization of the following conversation {text_to_summarize} """.format(text_to_summarize = text_)},
max_length = 500,
temperature = 0.7,
top_p = 0.95,
repetition_penalty = 1,
)
return " ".join(output)
load_dotenv()
textrazor.api_key = os.getenv("TEXT_RAZOR_API_KEY")
client = textrazor.TextRazor(extractors=["entities", "topics"])
def make_output(text:str):
response = client.analyze(text=text)
df = pd.DataFrame()
for entity in response.entities():
output_dict = {"id": "".join(list(entity.id or "None")),
"type": ", ".join(list(entity.dbpedia_types or "None")),
"wiki link": "".join(list(entity.wikipedia_link or "None"))}
df2 = pd.DataFrame.from_records([output_dict])
# df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
df = pd.concat([df, df2], ignore_index=True)
# print(entity.id)
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:***")
user_input = st.text_input("Text Here:", "")
if st.button("🔎 Search It!"):
def predict_sentiment(data:str):
ans = make_output(user_input)
return ans
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:***")
user_input1 = st.text_area("Text Here:", "")
if st.button("🪄 Summarize"):
def summarize(data:str):
ans1 = make_summary(user_input1)
return ans1
st.write(summarize(user_input1)) |