Upload app.py

app.py

@@ -190,7 +190,7 @@ def query_generate(text):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
-
+@st.experimental_singleton()
 def generate_poster(movie_data):
     """
     Function for recommending movies
@@ -221,17 +221,17 @@ def generate_poster(movie_data):
         st.text("")
         st.text("")
         st.subheader("Resulting poster:")
-        col1, col2, col3 = st.columns(3)
+        col1, col2, col3 = st.columns([1, 10, 1])
 
         with col1:
-            st.write(' ')
+            st.write("")
 
         with col2:
             st.text("")
             st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")
 
         with col3:
-            st.write(' ')
+            st.write("")
 
     except:
         st.text(poster_image)
@@ -282,11 +282,13 @@ if __name__ == "__main__":
 
     buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
 
-    is_clicked = col1.button(label="Generate poster!")
-    is_clicked_rerun = None
+    while 1:
+        is_clicked = col1.button(label="Generate poster!")
+        is_clicked_rerun = None
 
-    if is_clicked:
-        poster = generate_poster(data[data.title_year==selected_movie])
+        if is_clicked:
+            poster = generate_poster(data[data.title_year==selected_movie])
+            generate_poster.clear()
 
     _= """
     if poster is not None: