Spaces:
Sleeping
Sleeping
import recomender | |
import streamlit as st | |
import requests | |
import pandas as pd | |
df_cbf_Q = pd.read_csv('df_cbf_Q.csv', low_memory = False) | |
list_of_all_movies = list(df_cbf_Q['original_title']) | |
################################################################ | |
def get_movie_information(movie_title): | |
params={"apikey": "c176e26f", "t": movie_title, "plot": "full"} | |
response = requests.get("http://www.omdbapi.com/", params=params) | |
if response.status_code == 200: | |
data = response.json() | |
if data["Response"] == "False": | |
print(data["Error"]) | |
else: | |
return data | |
else: | |
print("Error:", response.status_code) | |
####################################################################### | |
def show_image(index): | |
dict_movie =get_movie_information(list_of_movies[index]) | |
st.header(dict_movie["Title"]) | |
st.subheader(dict_movie["Year"]) | |
if dict_movie["Poster"] != "N/A": | |
st.image(dict_movie["Poster"], use_column_width=False) | |
else: | |
st.write("Poster is not available!") | |
st.markdown(f"**IMDB Rating:** {dict_movie['imdbRating']} / 10") | |
st.markdown(f"**Director:** {dict_movie['Director']}") | |
st.markdown(f"**Actors:** {dict_movie['Actors']}") | |
st.markdown(f"**Writer:** {dict_movie['Writer']}") | |
st.markdown(f"**Genre:** {dict_movie['Genre']}") | |
st.markdown(f"**Year:** {dict_movie['Year']}") | |
st.markdown(f"**Country:** {dict_movie['Country']}") | |
st.markdown(f"**Language:** {dict_movie['Language']}") | |
st.write(f"**Plot:** {dict_movie['Plot']}") | |
##################################################################################################### | |
st.set_page_config(page_title="Movie Info", page_icon=":movie_camera:") | |
st.title("Movie Recommender Engine") | |
system_option = st.radio(" How would you like us to choose your next movie?" | |
,("Best Movies of all time","Trend Movies","Special for You:)", " People with similar tast like (CFR)")) | |
movies_watched = st.multiselect("What are your top three fovorite movies? (At least 3 movies)", list_of_all_movies) | |
########################################################################################### | |
suggest_button = st.button("Suggests me new movies to watch! ") | |
if suggest_button: | |
if len(movies_watched) < 3: | |
st.error(" You need to mention 3 movies!") | |
else: | |
if system_option == "Best Movies of all time": | |
list_of_movies = recomender.final_recommender_hot_picks_of_all_time(movies_watched) | |
elif system_option == "Trend Movies": | |
list_of_movies = recomender.final_recommender_hot_picks_now(movies_watched) | |
elif system_option == "Special for You:)": | |
list_of_movies = recomender.final_recommender_for_you(movies_watched) | |
else: | |
list_of_movies = recomender.recommender_svd(movies_watched) | |
for i,_ in enumerate(list_of_movies): | |
show_image(i) | |