Update app.py

app.py

@@ -53,7 +53,29 @@ with st.sidebar:
     selected_conta = st.selectbox('Seleciona Conta', df['Conta'].unique())
 
 # Filter the DataFrame based on selected values
-filtered_df = df[(df['Instituição'] == selected_instituicao) & (df['Conta'] == selected_conta)]
+#filtered_df = df[(df['Instituição'] == selected_instituicao) & (df['Conta'] == selected_conta)]
+
+# Initial filtering based on selected 'Instituição'
+instituicao_df = df[df['Instituição'] == selected_instituicao]
+
+# Container for adjusted DataFrame rows
+adjusted_rows = []
+
+# Iterate through each unique 'Conta' within the selected 'Instituição'
+for conta in instituicao_df['Conta'].unique():
+    conta_df = instituicao_df[instituicao_df['Conta'] == conta]
+    
+    # Check if 'Linear Regression' is available for this 'Conta'
+    if 'Linear Regression' in conta_df['Modelo'].values:
+        # If yes, filter to only 'Linear Regression' results for this 'Conta'
+        lr_rows = conta_df[conta_df['Modelo'] == 'Linear Regression']
+        adjusted_rows.append(lr_rows)
+    else:
+        # If not, include all models' results for this 'Conta'
+        adjusted_rows.append(conta_df)
+
+# Combine all adjusted rows back into a single DataFrame
+filtered_df = pd.concat(adjusted_rows)
 
 #col1, col2, col3 = st.columns(3) # This divides the page into three equal parts