Spaces:
Sleeping
Sleeping
Commit
•
37da5cb
1
Parent(s):
d72a444
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,109 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import streamlit as st
|
| 2 |
+
import pickle
|
| 3 |
+
import requests
|
| 4 |
+
import streamlit.components.v1 as components
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
def recommend(movie):
|
| 9 |
+
index = movies[movies['title'] == movie].index[0]
|
| 10 |
+
distances = sorted(list(enumerate(similarity[index])),reverse=True,key = lambda x: x[1])
|
| 11 |
+
recommended_movie_names = []
|
| 12 |
+
recommended_movie_posters = []
|
| 13 |
+
for i in distances[1:8]:
|
| 14 |
+
movie_id = movies.iloc[i[0]].movie_id
|
| 15 |
+
recommended_movie_posters.append(fetch_poster(movie_id))
|
| 16 |
+
recommended_movie_names.append(movies.iloc[i[0]].title)
|
| 17 |
+
|
| 18 |
+
return recommended_movie_names,recommended_movie_posters
|
| 19 |
+
|
| 20 |
+
|
| 21 |
+
H = ['Action', 'Adventure', 'Fantasy', 'ScienceFiction', 'Crime', 'Drama', 'Thriller', 'Animation', 'Family', 'Western', 'Comedy', 'Romance', 'Horror',
|
| 22 |
+
'Mystery', 'History', 'War', 'Music', 'Documentary', 'Foreign', 'TVMovie']
|
| 23 |
+
|
| 24 |
+
def suggest(gen):
|
| 25 |
+
gen = gen.lower()
|
| 26 |
+
url = "https://www.imdb.com/search/title/?genres={}&sort=user_rating,desc".format(gen)
|
| 27 |
+
return url
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
def fetch_poster(movie_id):
|
| 32 |
+
url = "https://api.themoviedb.org/3/movie/{}?api_key=8265bd1679663a7ea12ac168da84d2e8&language=en-US".format(movie_id)
|
| 33 |
+
data = requests.get(url)
|
| 34 |
+
data = data.json()
|
| 35 |
+
poster_path = data['poster_path']
|
| 36 |
+
full_path = "https://image.tmdb.org/t/p/w500/" + poster_path
|
| 37 |
+
return full_path
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
st.markdown("""<p style="color: #3a02ab;font-size: 70px;font-family: sans-serif;"><b>FILM</b><span style="color: #f75e05;font-size: 70px;font-family: sans-serif;"><b>ISH</b></span></p>""", unsafe_allow_html=True)
|
| 42 |
+
|
| 43 |
+
|
| 44 |
+
movies = pickle.load(open('movie_list.pkl', 'rb'))
|
| 45 |
+
similarity = pickle.load(open('similarity.pkl', 'rb'))
|
| 46 |
+
movie_list = movies['title'].values
|
| 47 |
+
|
| 48 |
+
st.markdown(""" <span style="font-size: 30px; font-family: Arial, Helvetica, sans-serif; color: #040330;"> Enter or select a Movie</span>""", unsafe_allow_html=True)
|
| 49 |
+
|
| 50 |
+
selected_movie = st.selectbox("", movie_list)
|
| 51 |
+
if st.button('Show Recommendation'):
|
| 52 |
+
recommended_movie_names,recommended_movie_posters = recommend(selected_movie)
|
| 53 |
+
col1, col2, col3, col4, col5 , col6, col7 = st.columns(7)
|
| 54 |
+
with col1:
|
| 55 |
+
st.text(recommended_movie_names[0])
|
| 56 |
+
st.image(recommended_movie_posters[0])
|
| 57 |
+
with col2:
|
| 58 |
+
st.text(recommended_movie_names[1])
|
| 59 |
+
st.image(recommended_movie_posters[1])
|
| 60 |
+
|
| 61 |
+
with col3:
|
| 62 |
+
st.text(recommended_movie_names[2])
|
| 63 |
+
st.image(recommended_movie_posters[2])
|
| 64 |
+
with col4:
|
| 65 |
+
st.text(recommended_movie_names[3])
|
| 66 |
+
st.image(recommended_movie_posters[3])
|
| 67 |
+
with col5:
|
| 68 |
+
st.text(recommended_movie_names[4])
|
| 69 |
+
st.image(recommended_movie_posters[4])
|
| 70 |
+
with col6:
|
| 71 |
+
st.text(recommended_movie_names[5])
|
| 72 |
+
st.image(recommended_movie_posters[5])
|
| 73 |
+
with col7:
|
| 74 |
+
st.text(recommended_movie_names[6])
|
| 75 |
+
st.image(recommended_movie_posters[6])
|
| 76 |
+
|
| 77 |
+
components.html("<html> <br /> <br /><br /> </html>")
|
| 78 |
+
components.html(
|
| 79 |
+
"""
|
| 80 |
+
<style type="text/css">
|
| 81 |
+
#block {
|
| 82 |
+
height: 100px;
|
| 83 |
+
width: 100%;
|
| 84 |
+
background-color: #f002a5;
|
| 85 |
+
border-radius: 20px;
|
| 86 |
+
position: relative;
|
| 87 |
+
}
|
| 88 |
+
</style>
|
| 89 |
+
<body>
|
| 90 |
+
<br />
|
| 91 |
+
<div id="block">
|
| 92 |
+
<span style="font-size: 42px; color: white; font-family: Arial, Helvetica, sans-serif;"> You will be redirected to IMDB best movies page </span>
|
| 93 |
+
</div>
|
| 94 |
+
</body>
|
| 95 |
+
"""
|
| 96 |
+
)
|
| 97 |
+
|
| 98 |
+
#components.html("<html> <br /> <br /><br /> </html>")
|
| 99 |
+
st.markdown(""" <span style="font-size: 30px; font-family: Arial, Helvetica, sans-serif; color: #040330;"> Enter or select a Genre</span>""", unsafe_allow_html=True)
|
| 100 |
+
selected_genre = st.selectbox("", H)
|
| 101 |
+
if st.button("show movies"):
|
| 102 |
+
link = "[List of movies]("+ suggest(selected_genre)+")"
|
| 103 |
+
st.markdown(link, unsafe_allow_html=True)
|
| 104 |
+
|
| 105 |
+
|
| 106 |
+
|
| 107 |
+
|
| 108 |
+
|
| 109 |
+
|