Change layout

app.py

@@ -120,7 +120,7 @@ if state.image_file is None:
 new_col1, new_col2 = st.beta_columns([5,5])
 
 
-if new_col2.button("Get a random example", help="Get a random example from one of the seeded examples."):
+if new_col1.button("Get a random example", help="Get a random example from one of the seeded examples."):
     sample = dummy_data.sample(1).reset_index()
     state.image_file = sample.loc[0, "image_file"]
     state.caption = sample.loc[0, "caption"].strip("- ")
@@ -135,10 +135,11 @@ transformed_image = get_transformed_image(state.image)
 # Display Image
 new_col1.image(state.image, use_column_width="always")
 # Display Reference Caption
-new_col2.write("**Reference Caption**: " + state.caption)
-new_col2.markdown(
-    f"""**English Translation**: {state.caption if state.lang_id == "en" else translate(state.caption, 'en')}"""
-)
+with new_col1.beta_expander("Reference Caption"):
+    st.write("**Reference Caption**: " + state.caption)
+    st.markdown(
+        f"""**English Translation**: {state.caption if state.lang_id == "en" else translate(state.caption, 'en')}"""
+    )
 
 # Select Language
 options = list(code_to_name.keys())
@@ -157,10 +158,10 @@ if new_col2.button("Generate Caption", help="Generate a caption in the specified
 # print(sequence)
 
 if sequence!=['']:
-    st.write(
+    new_col2.write(
         "**Generated Caption**: "+sequence[0]
     )
 
-    st.write(
+    new_col2.write(
         "**English Translation**: "+ sequence[0] if lang_id=="en" else translate(sequence[0])
     )