Upload app.py

app.py

@@ -152,6 +152,8 @@ def load_dataset():
 
     # Create dataframe
     data = pd.read_csv('movies_metadata.csv', low_memory=False)
+    data['year'] = data["release_date"].map(lambda x: x.split('-')[0] if isinstance(x, str) else '0')
+    data['title_year'] = data['title'] + ' (' + data['year'] + ')'
 
     return data
 
@@ -203,8 +205,7 @@ def generate_poster(movie_data):
 
     st.text("")
     st.text("")
-    st.title("""
-        #### Synopsis: """)
+    st.subheader("Synopsis:")
     st.text("Synopsis summary: " + synopsis_sum)
     st.text("")
 
@@ -219,8 +220,7 @@ def generate_poster(movie_data):
 
         st.text("")
         st.text("")
-        st.title("""
-            #### Resulting poster: """)
+        st.subheader("Resulting poster:")
         col1, col2, col3 = st.columns(3)
 
         with col1:
@@ -286,12 +286,14 @@ if __name__ == "__main__":
     is_clicked_rerun = None
 
     if is_clicked:
-        poster = generate_poster(data[data.title==selected_movie])
+        poster = generate_poster(data[data.title_year==selected_movie])
 
+    """
     if poster is not None:
         buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
         is_clicked_rerun = col1.button(label="Rerun with same movie!")
 
     if is_clicked_rerun:
-        poster = generate_poster(data[data.title==selected_movie])
+        poster = generate_poster(data[data.title_year==selected_movie])
+    """