Upload app.py

app.py

@@ -213,22 +213,22 @@ def generate_poster(movie_data):
 
     # Print summarized synopsis
     st.text("")
-    synopsis_expander = st.expander("Expand", expanded=True)
+    synopsis_expander = st.expander("Show synopsis", expanded=True)
     with synopsis_expander:
         st.subheader("Summarized synopsis:")
         col1, col2, col3 = st.columns([1, 5, 1])
         with col2:
-            st.text(synopsis_sum)
+            st.write(synopsis_sum)
     st.text("")
 
 
     # Get image based on synopsis
     with st.spinner("Generating poster image..."):
-        poster_image = query_generate(synopsis_sum)
+        response_content = query_generate(synopsis_sum)
 
     # Show image
     try: 
-        image = Image.open(io.BytesIO(poster_image))
+        image = Image.open(io.BytesIO(response_content))
 
         st.text("")
         st.text("")
@@ -238,9 +238,11 @@ def generate_poster(movie_data):
             st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")
 
     except:
-        st.text(poster_image)
+        col1, col2, col3 = st.columns([1, 5, 1])
+        with col2:
+            st.write(response_content)
 
-    return poster_image
+    return response_content
 # ------------------------------------------------------- #