Update logic of the container

app.py

@@ -72,14 +72,16 @@ option = st.selectbox(
 summary_output = "Deafualt"
 col1, col2 = st.columns(2)
 
-col1.title('Input')   
+col1.title('Input')  
+col2.title('Output')   
+
 col1.container(height=500, border=True).text_area("", option, height=1000)
 if col1.button("Summarize"):
     with st.spinner('Wait for it...'):
         print("Summary Function Start", summary_output)
         summary_output = summarize_text(option)
         print("Summary Function End", summary_output)
+        col2.container(height=500, border=True).markdown(summary_output, unsafe_allow_html=True)
+        
         
-col2.title('Output')   
-col2.container(height=500, border=True).markdown(summary_output, unsafe_allow_html=True)