.

app.py

@@ -33,8 +33,7 @@ def find_similar_movies(movie_id, X, k, metric='cosine', show_distance=False):
     neighbour_ids.pop(0)
     return neighbour_ids
 
-def recommend_movies(movie_name):
-    movie_id = movies[movies['title'] == movie_name]['movieId'].values[0]
+def recommend_movies(movie_id):
     similar_ids = find_similar_movies(movie_id, X, k=10)
     movie_title = movie_titles[movie_id]
     recommendations = []
@@ -59,14 +58,12 @@ X, user_mapper, movie_mapper, user_inv_mapper, movie_inv_mapper = create_matrix(
 movie_titles = dict(zip(movies['movieId'], movies['title']))
 
 # Set up Gradio interface
-movie_name = gr.inputs.Textbox
-
+movie_id = gr.inputs.Number(default=3, label="Movie ID")
 iface = gr.Interface(
     fn=recommend_movies, 
-    inputs=movie_name, 
+    inputs=movie_id, 
     outputs="text",
     title="Movie Recommender System",
     description="Enter a movie ID and get recommendations for similar movies."
 )
-iface.launch()
-
+gr.launch()