[text2image] suggestions without session_state

text2image.py

@@ -111,39 +111,41 @@ def app():
         
         🤌 Italian mode on! 🤌
 
+
+        You can choose one of our examples down below...
         """
     )
 
-    if "suggestion" not in st.session_state:
-        st.session_state.suggestion = ""
-
-    def update_query(value=""):
-        st.session_state.suggestion = value
+    suggestions = [
+        "Un gatto",
+        "Due gatti",
+        "Un fiore giallo",
+        "Un gatto sopra una sedia",
+    ]
+    sugg_idx = -1
 
-    col1, col2, col3, col4 = st.beta_columns(4)
+    col1, col2, col3, col4 = st.beta_columns([1, 1, 1, 2])
     with col1:
-        st.button("Un gatto", on_click=update_query, kwargs=dict(value="Un gatto"))
+        if st.button(suggestions[0]):
+            sugg_idx = 0
     with col2:
-        st.button("Due gatti", on_click=update_query, kwargs=dict(value="Due gatti"))
+        if st.button(suggestions[1]):
+            sugg_idx = 1
     with col3:
-        st.button(
-            "Un fiore giallo",
-            on_click=update_query,
-            kwargs=dict(value="Un fiore giallo"),
-        )
+        if st.button(suggestions[2]):
+            sugg_idx = 2
     with col4:
-        st.button(
-            "Un gatto sopra una sedia", on_click=update_query, kwargs=dict(value="Un gatto sopra una sedia")
-        )
+        if st.button(suggestions[3]):
+            sugg_idx = 3
 
     col1, col2 = st.beta_columns([3, 1])
     with col1:
-        query = st.text_input(
-            "Insert an italian query text here...", st.session_state.suggestion
-        )
+        query = st.text_input("... or insert an Italian query text")
     with col2:
         dataset_name = st.selectbox("IR dataset", ["Unsplash", "CC"])
 
+    query = query if query else suggestions[sugg_idx] if sugg_idx > -1 else ""
+
     if query:
         with st.spinner("Computing..."):