blacksw0rd commited on
Commit
324fd59
·
verified ·
1 Parent(s): ae8245a

Upload 4 files

Browse files
Files changed (4) hide show
  1. app.py +49 -0
  2. movies.csv +0 -0
  3. requirements.txt +0 -0
  4. similarity_matrix.pkl +3 -0
app.py ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import joblib
3
+ import streamlit as st
4
+ import requests
5
+
6
+ def recommend_movies(movie):
7
+
8
+ idx = df[df['title'] == movie].index[0]
9
+ similar_movies = similarity_matrix[idx]
10
+
11
+ titles = []
12
+ posters = []
13
+
14
+ for film in similar_movies[1:9]:
15
+ url = f"https://www.omdbapi.com/?i={df['imdb_id'][film]}&apikey=8c4102bb"
16
+
17
+ response = requests.get(url)
18
+ if response.status_code == 200:
19
+ data = response.json()
20
+ titles.append(data["Title"])
21
+ posters.append(data["Poster"])
22
+
23
+
24
+
25
+ index = 0
26
+ for _ in range(2):
27
+ columns = st.columns(4)
28
+ for col in columns:
29
+ if index < len(titles):
30
+ if posters[index] != 'N/A':
31
+ col.image(posters[index])
32
+ col.write(titles[index])
33
+ else:
34
+ col.write(titles[index])
35
+ index += 1
36
+
37
+
38
+
39
+
40
+ st.title("Movie Recommendation System")
41
+
42
+ similarity_matrix = joblib.load("similarity_matrix.pkl")
43
+ df = pd.read_csv("movies.csv")
44
+
45
+ movie = st.selectbox("Select a Movie",df["title"])
46
+
47
+ if st.button("Search"):
48
+ recommend_movies(movie)
49
+
movies.csv ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
Binary file (1.53 kB). View file
 
similarity_matrix.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:926ba45fb4a721daefe1966cebc41e8603b1d4a5753aa71f177737dc0d58ad3d
3
+ size 339809