Update app.py

app.py

@@ -18,24 +18,24 @@ similarity_matrix = calculate_similarity_matrix(genre_df)
 movie = st.text_input("Search for movie")
 if 'random_movies' not in st.session_state:
     st.session_state['random_movies'] = movies_df['title'].sample(5)
-st.columns(2)
+col1,col2 = st.columns(2)
 random_movies = st.session_state['random_movies']
 movie1,id = None,0
 if not movie:
-    st.columns(1)
-    for i,movie in enumerate(random_movies,0):
-        if st.button(movie):
-            movie1 = movie
-            id = i
-            st.write("Selected Movie:",movie1)
-            movie_id = movies_df[movies_df['title'] == random_movies.iloc[id]].index
-            similar_movies_idx = similarity_matrix[movie_id[0]].argsort()[::-1][1:6]
-            similar_movies = movies_df['title'].iloc[similar_movies_idx]
-            st.columns(2)
-                st.header("Recommended Movies")
-                for i,j in enumerate(similar_movies):
-                    st.write(f"{i+1}.{j}")
-            
-        refresher = st.button("Refresh",key='refresh')
-        elif refresher:    
-            st.session_state['random_movies'] = None
+    with col1:
+        for i,movie in enumerate(random_movies,0):
+            if st.button(movie):
+                movie1 = movie
+                id = i
+                st.write("Selected Movie:",movie1)
+                movie_id = movies_df[movies_df['title'] == random_movies.iloc[id]].index
+                similar_movies_idx = similarity_matrix[movie_id[0]].argsort()[::-1][1:6]
+                similar_movies = movies_df['title'].iloc[similar_movies_idx]
+                with col2:
+                    st.header("Recommended Movies")
+                    for i,j in enumerate(similar_movies):
+                        st.write(f"{i+1}.{j}")
+                
+            refresher = st.button("Refresh",key='refresh')
+            elif refresher:    
+                st.session_state['random_movies'] = None