Spaces:
Sleeping
Sleeping
| import plotly.graph_objects as go | |
| from plotly.subplots import make_subplots | |
| import streamlit as st | |
| import requests | |
| import json | |
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| # AI model code | |
| HF_API_KEY = os.getenv("HF_API_KEY") | |
| # API_URL_ED = "https://api-inference.huggingface.co/models/j-hartmann/emotion-english-distilroberta-base" #alternate ED model(slow loading on first run) | |
| API_URL_ED = "https://api-inference.huggingface.co/models/bhadresh-savani/bert-base-go-emotion" | |
| API_URL_HS = "https://api-inference.huggingface.co/models/IMSyPP/hate_speech_en" | |
| API_URL_SD = "https://api-inference.huggingface.co/models/NLP-LTU/bertweet-large-sexism-detector" | |
| headers = {"Authorization": f"Bearer {HF_API_KEY}"} | |
| def query(payload): | |
| response_ED = requests.request("POST", API_URL_ED, headers=headers, json=payload) | |
| response_HS = requests.request("POST", API_URL_HS, headers=headers, json=payload) | |
| response_SD = requests.request("POST", API_URL_SD, headers=headers, json=payload) | |
| return (json.loads(response_ED.content.decode("utf-8")),json.loads(response_HS.content.decode("utf-8")),json.loads(response_SD.content.decode("utf-8"))) | |
| st.set_page_config( | |
| page_title="GoEmotions Dashboard", | |
| layout="wide" | |
| ) | |
| # Set page title | |
| st.title("GoEmotions Dashboard - Analyzing Emotions in Text") | |
| # Define color map for each emotion category | |
| color_map = { | |
| 'admiration': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'], | |
| 'amusement': ['#ff7f0e', '#98df8a', '#2ca02c', '#d62728'], | |
| 'anger': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'annoyance': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'approval': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'], | |
| 'caring': ['#98df8a', '#2ca02c', '#FF69B4', '#d62728'], | |
| 'confusion': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'], | |
| 'curiosity': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'], | |
| 'desire': ['#2ca02c', '#ff7f0e', '#98df8a', '#d62728'], | |
| 'disappointment': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'disapproval': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'disgust': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'embarrassment': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'], | |
| 'excitement': ['#ff7f0e', '#2ca02c', '#98df8a', '#d62728'], | |
| 'fear': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'gratitude': ['#98df8a', '#2ca02c', '#1f77b4', '#d62728'], | |
| 'grief': ['#ffbb78', '#d62728', '#bcbd22', '#ff7f0e'], | |
| 'joy': ['#ff7f0e', '#98df8a', '#2ca02c', '#d62728'], | |
| 'love': ['#FF69B4', '#98df8a', '#2ca02c', '#d62728'], | |
| 'nervousness': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'], | |
| 'optimism': ['#98df8a', '#2ca02c', '#1f77b4', '#d62728'], | |
| 'pride': ['#98df8a', '#ff7f0e', '#1f77b4', '#d62728'], | |
| 'realization': ['#9467bd', '#ff7f0e', '#ffbb78', '#d62728'], | |
| 'relief': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'], | |
| 'remorse': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'sadness': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'], | |
| 'surprise': ['#ff7f0e', '#ffbb78', '#9467bd', '#d62728'], | |
| 'neutral': ['#2ca02c', '#98df8a', '#1f77b4', '#d62728'] | |
| } | |
| # Labels for Hate Speech Classification | |
| label_hs = {"LABEL_0": "Acceptable", "LABEL_1": "Inappropriate", "LABEL_2": "Offensive", "LABEL_3": "Violent"} | |
| # Define default options | |
| default_options = [ | |
| "I'm so excited for my vacation next week!", | |
| "I'm feeling so stressed about work.", | |
| "I just received great news from my doctor!", | |
| "I can't wait to see my best friend tomorrow.", | |
| "I'm feeling so lonely and sad today." | |
| "I'm so angry at my neighbor for being so rude.", | |
| "You are so annoying!", | |
| "You people from small towns are so dumb.", | |
| "If you don't agree with me, you are a moron.", | |
| "I hate you so much!", | |
| "If you don't listen to me, I'll beat you up!", | |
| ] | |
| with st.sidebar: | |
| # Create dropdown with default options | |
| selected_option = st.selectbox("Select a default option or enter your own text:", default_options) | |
| # Display text input with selected option as default value | |
| text_input = st.text_area("Enter text to analyze emotions:", value = selected_option, height=100) | |
| # Add submit button | |
| submit = st.button("Submit") | |
| # If submit button is clicked | |
| if submit: | |
| # Call API and get predicted probabilities for each emotion category and hate speech classification | |
| payload = {"inputs": text_input, "options": {"wait_for_model": True, "use_cache": True}} | |
| response_ED, response_HS, response_SD = query(payload) | |
| predicted_probabilities_ED = response_ED[0] | |
| predicted_probabilities_HS = response_HS[0] | |
| predicted_probabilities_SD = response_SD[0] | |
| # Creating columns to visualize the results | |
| ED, _, HS, __, SD = st.columns([4,1,2,1,2]) | |
| with ED: | |
| # Get the top 4 emotion categories and their scores | |
| top_emotions = predicted_probabilities_ED[:4] | |
| top_scores = [e['score'] for e in top_emotions] | |
| # Create the gauge charts for the top 4 emotion categories | |
| fig = make_subplots(rows=2, cols=2, specs=[[{'type': 'indicator'}, {'type': 'indicator'}], | |
| [{'type': 'indicator'}, {'type': 'indicator'}]], | |
| vertical_spacing=0.4) | |
| for i, emotion in enumerate(top_emotions): | |
| # Get the emotion category, color, and normalized score for the current emotion | |
| category = emotion['label'] | |
| color = color_map[category] | |
| value = top_scores[i] * 100 | |
| # Calculate the row and column position for adding the trace to the subplots | |
| row = i // 2 + 1 | |
| col = i % 2 + 1 | |
| # Add a gauge chart trace for the current emotion category | |
| fig.add_trace(go.Indicator( | |
| domain={'x': [0, 1], 'y': [0, 1]}, | |
| value=value, | |
| mode="gauge+number", | |
| title={'text': category.capitalize()}, | |
| gauge={'axis': {'range': [None, 100]}, | |
| 'bar': {'color': color[3]}, | |
| 'bgcolor': 'white', | |
| 'borderwidth': 2, | |
| 'bordercolor': color[1], | |
| 'steps': [{'range': [0, 33], 'color': color[0]}, | |
| {'range': [33, 66], 'color': color[1]}, | |
| {'range': [66, 100], 'color': color[2]}], | |
| 'threshold': {'line': {'color': "black", 'width': 4}, | |
| 'thickness': 0.5, | |
| 'value': 50}}), row=row, col=col) | |
| # Update the layout of the figure | |
| fig.update_layout(height=400, margin=dict(t=50, b=5, l=0, r=0)) | |
| # Display gauge charts | |
| st.text("") | |
| st.text("") | |
| st.text("") | |
| st.subheader("Emotion Detection") | |
| st.text("") | |
| st.plotly_chart(fig, use_container_width=True) | |
| with _: | |
| st.text("") | |
| with HS: | |
| # Display Hate Speech Classification | |
| hate_detection = label_hs[predicted_probabilities_HS[0]['label']] | |
| st.text("") | |
| st.text("") | |
| st.text("") | |
| st.subheader("Hate Speech Analysis") | |
| st.text("") | |
| st.image(f"assets/{hate_detection}.jpg", width=200) | |
| st.text("") | |
| st.text("") | |
| st.markdown(f"#### The given text is: {hate_detection}") | |
| with __: | |
| st.text("") | |
| with SD: | |
| label_SD = predicted_probabilities_SD[0]['label'].title() | |
| st.text("") | |
| st.text("") | |
| st.text("") | |
| st.subheader("Sexism Detection") | |
| st.text("") | |
| st.image(f"assets/{label_SD}.jpg", width=200) | |
| st.text("") | |
| st.text("") | |
| st.markdown(f"#### The given text is: {label_SD}") | |
| hide_st_style = """ | |
| <style> | |
| #MainMenu {visibility: hidden;} | |
| footer {visibility: hidden;} | |
| </style> | |
| """ | |
| st.markdown(hide_st_style, unsafe_allow_html=True) |