|
import streamlit as st
|
|
import pandas as pd
|
|
import joblib
|
|
from PIL import Image
|
|
|
|
st.set_page_config(layout="wide")
|
|
|
|
header = st.container()
|
|
dataset = st.container()
|
|
recommendation = st.container()
|
|
|
|
with header:
|
|
header_ = '<p style="font-family:sans-serif; color:darkcyan; font-size: 46px;">MOVIE RECOMMENDATION</p>'
|
|
st.markdown(header_, unsafe_allow_html=True)
|
|
text0 = '<p style="font-family:Courier; color:firebrick; font-size: 24px;">With this application, we would like to make a list of movies that you will love watching!</p>'
|
|
st.markdown(text0, unsafe_allow_html=True)
|
|
|
|
with dataset:
|
|
st.header('WE BRING TO YOU THE BEST MOVIE COLLECTION')
|
|
col1, col2, col3 , col4, col5 = dataset.columns(5)
|
|
logo_netflix = Image.open('source/image_streamlit/netflix.jpg')
|
|
logo_amazon_prime = Image.open('source/image_streamlit/primevideo.png')
|
|
logo_apple_tv = Image.open('source/image_streamlit/apple_tv.png')
|
|
logo_hbo = Image.open('source/image_streamlit/HBO.jpg')
|
|
logo_disney = Image.open('source/image_streamlit/disney.png')
|
|
logo_paramount = Image.open('source/image_streamlit/paramount.png')
|
|
with col1:
|
|
st.image(logo_netflix)
|
|
st.image(logo_hbo)
|
|
with col3:
|
|
st.image(logo_amazon_prime)
|
|
st.image(logo_apple_tv)
|
|
with col5:
|
|
st.image(logo_paramount)
|
|
st.image(logo_disney)
|
|
|
|
with recommendation:
|
|
st.header('LET US MAKE RECOMMENDATION TO YOU')
|
|
text1 = '<p style="font-family:Courier; color:darkcyan; font-size: 20px;">Now you show us one of your favorite movies ...</p>'
|
|
st.markdown(text1, unsafe_allow_html=True)
|
|
text2 = '<p style="font-family:Courier; color:darkcyan; font-size: 20px;">... then let we recommend you movies that you will love!</p>'
|
|
st.markdown(text2, unsafe_allow_html=True)
|
|
|
|
frame = joblib.load('source/model/pickle_frame.joblib')
|
|
st.write('### **SELECT** or **TYPE** here below the movie title you loved watching:')
|
|
movie_chosen = st.selectbox('', options=[title for title in frame.columns.sort_values()])
|
|
st.write('### **Your favourite movie chosen:**')
|
|
st.subheader(movie_chosen)
|
|
st.write('### **Here are movies that we recommend you:**', unsafe_allow_html=True)
|
|
recommendation_list = list(frame[movie_chosen].sort_values(ascending=False)[1:11].index.values)
|
|
title = ''
|
|
for i in recommendation_list:
|
|
title += "- " + i + "\n"
|
|
st.markdown(title)
|
|
|
|
|
|
|