Update app.py

app.py

@@ -19,28 +19,27 @@ class MovieRecommender:
 
     def find_movies_by_genre(self, genre):
         with self.driver.session() as session:
-            query = """
-            MATCH (t:Title)-[:title_to_genere]->(g:Genere {genres: $genre})
-            MATCH (t)-[:title_to_year]->(y:year)
-            MATCH (t)-[:title_to_votes]->(v:Votes)
+            query = """ 
+            MATCH (t:Title)-[:TITLE_TO_GENRE]->(g:Genre {genres: $genre})
+            MATCH (t)-[:TITLE_TO_YEAR]->(y:Year)
+            MATCH (t)-[:TITLE_TO_VOTES]->(v:Votes)
             RETURN 
                 t.title as movie, 
                 y.releaseYear as year, 
-                v.avgVote as vote_rating
-            ORDER BY vote_rating DESC
+                v.avgVote as avgVote
+            ORDER BY avgVote DESC
             LIMIT 50
             """
             result = session.run(query, genre=genre)
             df = pd.DataFrame([dict(record) for record in result])
-            df = df.rename(columns={'avgVote': 'vote_rating'})
             
             # Add rating category column
+            df['vote_rating'] = df['avgVote']
             df['rating_category'] = pd.cut(
                 df['vote_rating'], 
                 bins=[0, 5, 7, float('inf')], 
                 labels=['Low Rating', 'Medium Rating', 'High Rating']
             )
-            
             return df
 
 def create_interface():