fluxo

app.py

@@ -36,32 +36,39 @@ def traduzir_en_pt(text):
 # Cabeçalho
 st.title('Tradutor de datasets (inglês para português)')
 # Carrega dataset
-dataset = st.file_uploader("Carrege o dataset (coluna a ser traduzida deve ser nomeada como 'texto')", type=["csv"])
-st.write('Carregando dataset...')
 
-if dataset is not None:
-    st.write('🎲 Dataset carregado com sucesso!')
-    dataset = pd.read_csv(dataset)
-    st.write(dataset)
-    
+with st.beta_container():
+    dataset = st.file_uploader("Carrege o dataset (coluna a ser traduzida deve ser nomeada como 'texto')", type=["csv"])
+    st.write('Carregando dataset...')
 
-modelo_selecionado = st.selectbox('Escolha um modelo', modelos_opcao)
+with st.beta_expander():
+    if dataset is not None:
+        st.write('🎲 Dataset carregado com sucesso!')
+        dataset = pd.read_csv(dataset)
+        st.write(dataset)
+    
 
-if st.button("Carregar modelo escolhido"):
-    model, tokenizer = carregar_modelo_e_tokenizador_mbart(modelo_selecionado)
-    st.write(f"🎰 Modelo {modelo_selecionado} carregado com sucesso!  🔥")
+with st.beta_container():
+    modelo_selecionado = st.selectbox('Escolha um modelo', modelos_opcao)
+    if st.button("Carregar modelo escolhido"):
+        model, tokenizer = carregar_modelo_e_tokenizador_mbart(modelo_selecionado)
+        st.write(f"🎰 Modelo {modelo_selecionado} carregado com sucesso!  🔥")
 
-qtde_linhas_traduzir = st.slider('Quantidade de linhas a serem traduzidas', 1, len(dataset), 50)
-if st.button(f"Traduzir {qtde_linhas_traduzir} linhas"):
-    for i in range(qtde_linhas_traduzir):
-        st.write(f'🔡  Traduzindo linha {i+1}...')
-        st.write(f'Texto: {dataset.iloc[i]["texto"]}')
-        texto_traduzido= traduzir_en_pt(dataset.iloc[i]["texto"])
-        st.write(f'Tradução: {texto_traduzido}')
+with st.beta_container():
+    qtde_linhas_traduzir = st.slider('Quantidade de linhas a serem traduzidas', 1, len(dataset), 50)
+    if st.button(f"Traduzir {qtde_linhas_traduzir} linhas"):
+        for i in range(qtde_linhas_traduzir):
+            st.write(f'🔡  Traduzindo linha {i+1}...')
+            st.write(f'Texto: {dataset.iloc[i]["texto"]}')
+            texto_traduzido= traduzir_en_pt(dataset.iloc[i]["texto"])
+            st.write(f'Tradução: {texto_traduzido}')
 
-        # adiciona traducao em nova coluna dataset
-        dataset["traduzido"]= texto_traduzido
+            # adiciona traducao em nova coluna dataset
+            dataset["traduzido"]= texto_traduzido
+        st.write("Traduzido 👍")
+with st.beta_expander():
+    st.write(dataset)
 
-st.write(dataset)
-st.download_button(label='Baixar dataset', data=dataset.to_csv(index=False))
-st.write("Fim 👍")
+with st.beta_expander():
+    st.download_button(label='Baixar dataset', data=dataset.to_csv(index=False))
+    st.write("Fim 👍")