|
import streamlit as st |
|
import pickle |
|
import pandas as pd |
|
import requests |
|
original_title = '<p style="color:White; font-size: 80px;">CinePicks</p>' |
|
st.markdown(original_title, unsafe_allow_html=True) |
|
st.markdown( |
|
f""" |
|
<style> |
|
.stApp {{ |
|
background: url("https://wallup.net/wp-content/uploads/2018/03/20/414708-Iron_Man-Iron_Man_2-Iron_Man_3-iron_man__mark_XLIII-The_Avengers.jpg"); |
|
}} |
|
</style> |
|
""", |
|
unsafe_allow_html=True |
|
) |
|
def fetch_poster(movie_id): |
|
url='https://api.themoviedb.org/3/movie/{}?api_key=4e9dfb203bf90ff6a4fc33522c802f3b'.format(movie_id) |
|
data=requests.get(url) |
|
data=data.json() |
|
poster_path=data['poster_path'] |
|
full_path="https://image.tmdb.org/t/p/w500/" + poster_path |
|
return full_path |
|
|
|
|
|
similarity=pickle.load(open('similarity.pkl','rb')) |
|
def recommend(movie): |
|
movie_index = movies[movies['title'] == movie].index[0] |
|
distances = similarity[movie_index] |
|
movies_list = sorted(list(enumerate(distances)), reverse=True, key=lambda x: x[1])[1:6] |
|
recommended_movies=[] |
|
recommended_movie_posters = [] |
|
|
|
for i in movies_list: |
|
movie_id=movies.iloc[i[0]].movie_id |
|
|
|
recommended_movies.append(movies.iloc[i[0]].title) |
|
recommended_movie_posters.append(fetch_poster(movie_id)) |
|
return recommended_movies,recommended_movie_posters |
|
|
|
|
|
|
|
movies_dict=pickle.load(open('movies_dict.pkl','rb')) |
|
movies=pd.DataFrame(movies_dict) |
|
selected_movie_name=st.selectbox('Recently Watched',movies['title'].values) |
|
if st.button('Show Recommendations'): |
|
recommended_movie_names, recommended_movie_posters = recommend(selected_movie_name) |
|
col1, col2, col3, col4, col5 = st.columns(5) |
|
with col1: |
|
st.text(recommended_movie_names[0]) |
|
st.image(recommended_movie_posters[0]) |
|
with col2: |
|
st.text(recommended_movie_names[1]) |
|
st.image(recommended_movie_posters[1]) |
|
|
|
with col3: |
|
st.text(recommended_movie_names[2]) |
|
st.image(recommended_movie_posters[2]) |
|
with col4: |
|
st.text(recommended_movie_names[3]) |
|
st.image(recommended_movie_posters[3]) |
|
with col5: |
|
st.text(recommended_movie_names[4]) |
|
st.image(recommended_movie_posters[4]) |
