added suggestions to text input

text2image.py

@@ -73,7 +73,24 @@ def app():
         """
     )
 
-    query = st.text_input("Insert an italian query text here...")
+    if 'suggestion' not in st.session_state:
+        st.session_state.suggestion = ""
+
+    def update_query(value=""):
+        st.session_state.suggestion = value
+
+    col1, col2, col3, col4 = st.beta_columns(4)
+    with col1:
+        st.button('Un gatto', on_click=update_query, kwargs=dict(value='Un gatto'))
+    with col2:
+        st.button('Due gatti', on_click=update_query, kwargs=dict(value='Due gatti'))
+    with col3:
+        st.button('Un fiore giallo', on_click=update_query, kwargs=dict(value='Un fiore giallo'))
+    with col4:
+        st.button('Un fiore blu', on_click=update_query, kwargs=dict(value='Un fiore blu'))
+
+    query = st.text_input('Insert an italian query text here...', st.session_state.suggestion)
+
     if query:
         with st.spinner("Computing in progress..."):
             model = get_model()