fixing caption suggestions

text2image.py

@@ -130,11 +130,13 @@ def app():
         "Un gatto",
         "Due gatti",
         "Un fiore giallo",
-        "Un gatto sopra una sedia",
+        "Un fiore blu",
+        "Una coppia in montagna",
+        "Una coppia al tramonto"
     ]
     sugg_idx = -1
 
-    col1, col2, col3, col4 = st.beta_columns([1, 1, 1, 2])
+    col1, col2, col3, col4, col5, col6 = st.beta_columns([1, 1, 1, 1, 1.5, 1.5])
     with col1:
         if st.button(suggestions[0]):
             sugg_idx = 0
@@ -147,6 +149,12 @@ def app():
     with col4:
         if st.button(suggestions[3]):
             sugg_idx = 3
+    with col5:
+        if st.button(suggestions[4]):
+            sugg_idx = 4
+    with col6:
+        if st.button(suggestions[5]):
+            sugg_idx = 5
 
     col1, col2 = st.beta_columns([3, 1])
     with col1: