Move Show changes toggle

Move the "Show changes" toggle closer to the output (from sidebar to below output)

app.py

@@ -78,13 +78,11 @@ with st.sidebar:
         on_change=clean_inputs)
     n_candidates = overlap2candidates[overlap]
 
-    st.subheader('Output')
-    show_changes = st.toggle('Show changes')
-
 
 def handle_input(input_, id_):
     """Generate and display output"""
 
+    # Put everything output-related in a bordered container
     with st.container(border=True):
         st.caption('Output')
 
@@ -95,10 +93,17 @@ def handle_input(input_, id_):
                 output = post_ocr.process(input_, n_candidates)
             st.session_state.outputs[id_] = output
 
-        # Display output
-        output =  st.session_state.outputs[id_]
-        if output is not None:
-            st.write(post_ocr.diff(input_, output) if show_changes else output)
+        # This container is needed to display the `show changes` toggle
+        # after the output text
+        container = st.container()
+        st.divider()
+        show_changes = st.toggle('Show changes', key=f'show_changes_{id_}')
+
+        with container:
+            # Display output
+            output = st.session_state.outputs[id_]
+            if output is not None:
+                st.write(post_ocr.diff(input_, output) if show_changes else output)
 
 
 # Manual entry tab