Upload app.py

app.py

@@ -122,6 +122,18 @@ def authenticate_kaggle():
     api.authenticate()
 
 
+def map_year(x):
+    a = ''
+    if isinstance(x, str):
+        try:
+            a = re.search('\(([0-9_]+)\)', x[-7:]).group(1)
+        except:
+            a = '0'
+    else:
+        a = '0'
+    return a
+
+
 @st.experimental_memo(persist=True, show_spinner=False, suppress_st_warning=True, max_entries=1)
 def load_dataset():
     """
@@ -152,8 +164,8 @@ def load_dataset():
 
     # Create dataframe
     data = pd.read_csv('movies.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'] + ')'
+    #data['year'] = data['release_date'].map(lambda x: x.split('-')[0] if isinstance(x, str) else '0')
+    data['year'] = data['title'].map(map_year)
 
     return data
 
@@ -294,7 +306,7 @@ if __name__ == "__main__":
     st.text("")
 
     ## Select box with all the movies as choices
-    session.selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title_year)
+    session.selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title)
 
     st.text("")
     st.text("")
@@ -305,7 +317,7 @@ if __name__ == "__main__":
 
     ## Generate poster
     if is_clicked:
-        poster = generate_poster(data[data.title_year==session.selected_movie])
+        poster = generate_poster(data[data.title==session.selected_movie])
         #generate_poster.clear()
         #st.runtime.legacy_caching.clear_cache()
 
@@ -321,6 +333,6 @@ if __name__ == "__main__":
         is_clicked_rerun = col1.button(label="Rerun with same movie!")
 
     if is_clicked_rerun:
-        poster = generate_poster(data[data.title_year==selected_movie])
+        poster = generate_poster(data[data.title==selected_movie])
     """