Update app.py

app.py

@@ -5,22 +5,28 @@ import pandas as pd
 import re
 import streamlit as st
 
+
 def translate_text_blob(text):
     blob = TextBlob(text)
     return str(blob.translate(from_lang="pt", to="en"))
 
+
 def translate_text_M2M100(text, model, tokenizer):
     encoded = tokenizer(text, return_tensors="pt")
-    generated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id("en"))
+    generated_tokens = model.generate(
+        **encoded, forced_bos_token_id=tokenizer.get_lang_id("en"))
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
+
 def translate_BART(text, model, tokenizer):
     inputs = tokenizer(text, return_tensors='pt')
     input_ids = inputs.input_ids
     attention_mask = inputs.attention_mask
-    output = model.generate(input_ids, attention_mask=attention_mask, forced_bos_token_id=tokenizer.lang_code_to_id['en_XX'])
+    output = model.generate(input_ids, attention_mask=attention_mask,
+                            forced_bos_token_id=tokenizer.lang_code_to_id['en_XX'])
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
+
 def sentiment_vader(sentence, vader_object):
 
     sentiment_dict = vader_object.polarity_scores(sentence)
@@ -29,36 +35,43 @@ def sentiment_vader(sentence, vader_object):
     positive = sentiment_dict['pos']
     compound = sentiment_dict['compound']
 
-    if sentiment_dict['compound'] >= 0.05 :
+    if sentiment_dict['compound'] >= 0.05:
         overall_sentiment = "Positive"
 
-    elif sentiment_dict['compound'] <= - 0.05 :
+    elif sentiment_dict['compound'] <= - 0.05:
         overall_sentiment = "Negative"
 
-    else :
+    else:
         overall_sentiment = "Neutral"
-  
+
     return overall_sentiment.upper()
 
+
 def sentiment_finbert(text, pipeline):
     result = pipeline(text)[0]
     return result["label"].upper()
 
+
 m2m100 = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
 m2m100_token = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
 m2m100_token.src_lang = "pt"
 
-BART_token = MBart50TokenizerFast.from_pretrained('Narrativa/mbart-large-50-finetuned-opus-pt-en-translation')
-BART = MBartForConditionalGeneration.from_pretrained('Narrativa/mbart-large-50-finetuned-opus-pt-en-translation')
+BART_token = MBart50TokenizerFast.from_pretrained(
+    'Narrativa/mbart-large-50-finetuned-opus-pt-en-translation')
+BART = MBartForConditionalGeneration.from_pretrained(
+    'Narrativa/mbart-large-50-finetuned-opus-pt-en-translation')
 BART_token.src_lang = 'pt_XX'
 
 finbert_token = AutoTokenizer.from_pretrained("ProsusAI/finbert")
-finbert = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", num_labels=3)
-finbert_pipeline = pipeline("sentiment-analysis", model=finbert, tokenizer=finbert_token)
+finbert = AutoModelForSequenceClassification.from_pretrained(
+    "ProsusAI/finbert", num_labels=3)
+finbert_pipeline = pipeline(
+    "sentiment-analysis", model=finbert, tokenizer=finbert_token)
 
 sid_obj = SentimentIntensityAnalyzer()
 
-classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
+classifier = pipeline("zero-shot-classification",
+                      model="joeddav/xlm-roberta-large-xnli")
 
 sheet_id = "1IGFSKnnmQndKVmGOWMCbsvJJMU_2jvnm"
 sheet_name = "Sheet1"
@@ -66,24 +79,27 @@ url = f"https://docs.google.com/spreadsheets/d/{sheet_id}/gviz/tq?tqx=out:csv&sh
 df = pd.read_csv(url)
 
 header = st.container()
+model_1, model_2 = st.columns(2)
 model = st.container()
 dataset = st.container()
 analysis = st.container()
-model_1, model_2 = st.columns(2)
 
 with st.sidebar:
-   st.markdown("# Lorem Ipsum\nLorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent sapien tortor, suscipit quis ornare ut, laoreet vitae nisi. Mauris quis consectetur risus, non blandit mauris. Sed ut odio tempor, ullamcorper leo eu, mollis eros. Curabitur pretium sollicitudin sapien, vel mattis augue convallis quis. Suspendisse eleifend turpis non nunc gravida, aliquet hendrerit orci viverra. Sed aliquet, nunc eu posuere tempor, libero ex dignissim velit, ut ultricies erat felis at urna. Proin metus augue, commodo in faucibus sed, aliquet ac eros. Nullam turpis leo, dictum eu tellus a, aliquam egestas velit. Suspendisse cursus augue a sem dapibus, eu faucibus nisl ultrices. Cras tortor ipsum, luctus vitae tincidunt id, dapibus id justo. Sed mi nunc, tempor eu iaculis in, tristique cursus massa. Integer metus felis, pulvinar ut aliquam ut, consectetur in nulla.")
-   
+    st.markdown("# Lorem Ipsum\nLorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent sapien tortor, suscipit quis ornare ut, laoreet vitae nisi. Mauris quis consectetur risus, non blandit mauris. Sed ut odio tempor, ullamcorper leo eu, mollis eros. Curabitur pretium sollicitudin sapien, vel mattis augue convallis quis. Suspendisse eleifend turpis non nunc gravida, aliquet hendrerit orci viverra. Sed aliquet, nunc eu posuere tempor, libero ex dignissim velit, ut ultricies erat felis at urna. Proin metus augue, commodo in faucibus sed, aliquet ac eros. Nullam turpis leo, dictum eu tellus a, aliquam egestas velit. Suspendisse cursus augue a sem dapibus, eu faucibus nisl ultrices. Cras tortor ipsum, luctus vitae tincidunt id, dapibus id justo. Sed mi nunc, tempor eu iaculis in, tristique cursus massa. Integer metus felis, pulvinar ut aliquam ut, consectetur in nulla.")
+
 with header:
-   st.title("IC 2022 Classificação de Dados Financeiros")
-   st.write("Lorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent sapien tortor, suscipit quis ornare ut, laoreet vitae nisi. Mauris quis consectetur risus, non blandit mauris. Sed ut odio tempor, ullamcorper leo eu, mollis eros.")
+    st.title("IC 2022 Classificação de Dados Financeiros")
+    st.write("Lorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent sapien tortor, suscipit quis ornare ut, laoreet vitae nisi. Mauris quis consectetur risus, non blandit mauris. Sed ut odio tempor, ullamcorper leo eu, mollis eros.")
 
 with model_1:
-   st.header("Modelo para Tradução e Classificação!")
-   translator  = st.selectbox('Qual modelo você deseja usar para tradução?', ('TextBlob', 'M2M100', 'BART'))
-   sentimentor = st.selectbox('Qual modelo você deseja usar para a análise de sentimentos?', ('Vader', 'FinBERT'))
-   text = st.text_area("Coloque seu texto sobre mercado financeiro em português!", "As ações da Ultrafarma subiram em 98% no último bimestre, segundo os dados da revista!")
-   submit = st.button('Gerar!')
+    st.header("Modelo para Tradução e Classificação!")
+    translator = st.selectbox(
+        'Qual modelo você deseja usar para tradução?', ('TextBlob', 'M2M100', 'BART'))
+    sentimentor = st.selectbox(
+        'Qual modelo você deseja usar para a análise de sentimentos?', ('Vader', 'FinBERT'))
+    text = st.text_area("Coloque seu texto sobre mercado financeiro em português!",
+                        "As ações da Ultrafarma subiram em 98% no último bimestre, segundo os dados da revista!")
+    submit = st.button('Gerar!')
 
 with model_2:
     if submit:
@@ -98,13 +114,17 @@ with model_2:
             sentiment = sentiment_vader(text_en, sid_obj)
         elif sentimentor == "FinBERT":
             sentiment = sentiment_finbert(text_en, finbert_pipeline)
-        
+
+        st.text_area(value="Translation: " + text_en)
+        st.text_area(value="Sentiment: "   + sentiment)
+
         st.write(text_en)
         st.write(sentiment)
 
 with model:
     st.header("Modelo para Tradução e Classificação!")
-    text = st.text_area("Coloque seu texto sobre mercado financeiro em português!")
+    text = st.text_area(
+        "Coloque seu texto sobre mercado financeiro em português!")
 
     submit = st.button('Generate')
 
@@ -126,6 +146,6 @@ with dataset:
     st.write("\t*- Texts:* Coluna que mostra os textos financeiros")
     st.write("\t*- Texts:* Coluna que mostra os textos financeiros")
     st.write("\t*- Texts:* Coluna que mostra os textos financeiros")
-   
+
 with analysis:
-    st.header("Visualização dos dados utilizados")
+    st.header("Visualização dos dados utilizados")