testing with container

app.py

@@ -241,29 +241,30 @@ if st.button('Generate Predictions'):
             recommended_changes=(best_target_tones)   
             change=best_target-pred
             #recommend(tones,recommended_changes,change,target)
-            fig2 = go.Figure()
-            fig2.add_trace(go.Bar(
-                y=tone_labels,
-                x=recommended_changes,
-                name='Recommend changes',
-                orientation='h',
-                text=np.round(recommended_changes,3),
-                width=.5,
-                marker=dict(
-                color='#e60f00',
-                line=dict(color='rgba(58, 71, 80, 1.0)', width=1)))
-            )
-            fig2.update_traces(textfont_size=18, textposition="outside", cliponaxis=False)
-            if target == 'Revenue_Per_Email':
-                out = f"${round(change,2)}"
-                st.write("The output will be between the range " + str(round(lower,2)) + ' and ' + str(round(upper,2)))
-                st.write("The Predicted "+str(target) +" is "+ str(round(pred,2)))
-            else:
-                out = f"{round(change,2)*100}%"
-                st.write("The output will be between the range " + str(round(lower,2) * 100) + ' and ' + str(round(upper,2) *100))
-                st.write("The Predicted "+str(target) +" is "+ str(round(pred,2)*100))
-            fig2.update_layout(title_text=f'The following Changes will yield a {out} increase in {target}')
-            st.plotly_chart(fig2, use_container_width=True)
-            #FunctionsModelSA_V1.corrections(best_target_tones,test_predictions))
+            with st.container(): 
+                fig2 = go.Figure()
+                fig2.add_trace(go.Bar(
+                    y=tone_labels,
+                    x=recommended_changes,
+                    name='Recommend changes',
+                    orientation='h',
+                    text=np.round(recommended_changes,3),
+                    width=.5,
+                    marker=dict(
+                    color='#e60f00',
+                    line=dict(color='rgba(58, 71, 80, 1.0)', width=1)))
+                )
+                fig2.update_traces(textfont_size=18, textposition="outside", cliponaxis=False)
+                if target == 'Revenue_Per_Email':
+                    out = f"${round(change,2)}"
+                    st.write("The output will be between the range " + str(round(lower,2)) + ' and ' + str(round(upper,2)))
+                    st.write("The Predicted "+str(target) +" is "+ str(round(pred,2)))
+                else:
+                    out = f"{round(change,2)*100}%"
+                    st.write("The output will be between the range " + str(round(lower,2) * 100) + ' and ' + str(round(upper,2) *100))
+                    st.write("The Predicted "+str(target) +" is "+ str(round(pred,2)*100))
+                fig2.update_layout(title_text=f'The following Changes will yield a {out} increase in {target}')
+                st.plotly_chart(fig2, use_container_width=True)
+                #FunctionsModelSA_V1.corrections(best_target_tones,test_predictions))
         
         placeholder.empty()