amirakhlaghiqqq's picture
Update app.py
9ffeec1
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)