ChandimaPrabath commited on
Commit
f294bf8
·
1 Parent(s): fdcd21d

fix genre limit

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -88,19 +88,24 @@ async def get_genre_items(genre: List[str] = Query(...), media_type: Optional[st
88
 
89
  :param genre: The genres to filter by (e.g., 'Comedy').
90
  :param media_type: Optional. Filter by media type ('movie' or 'series').
91
- :param limit: The maximum number of items to return.
92
  :return: A JSON response containing the filtered items.
93
  """
94
  # Get sorted entries based on genres and media type
95
  entries = genre_list.get_entries_by_multiple_genres(genre, media_type=media_type)
 
 
 
 
96
 
97
- # Slice the list to only return the desired number of items
98
- limited_entries = entries[:limit]
 
99
 
100
  # Organize the results by media type
101
  results = {
102
- 'movies': [entry for entry in limited_entries if entry[4] == 'movie'],
103
- 'series': [entry for entry in limited_entries if entry[4] == 'series']
104
  }
105
 
106
  # Return the results in a JSON response
 
88
 
89
  :param genre: The genres to filter by (e.g., 'Comedy').
90
  :param media_type: Optional. Filter by media type ('movie' or 'series').
91
+ :param limit: The maximum number of items to return for each media type.
92
  :return: A JSON response containing the filtered items.
93
  """
94
  # Get sorted entries based on genres and media type
95
  entries = genre_list.get_entries_by_multiple_genres(genre, media_type=media_type)
96
+
97
+ # Separate entries by media type
98
+ movies = [entry for entry in entries if entry[4] == 'movie']
99
+ series = [entry for entry in entries if entry[4] == 'series']
100
 
101
+ # Limit the number of items for each media type
102
+ limited_movies = movies[:limit]
103
+ limited_series = series[:limit]
104
 
105
  # Organize the results by media type
106
  results = {
107
+ 'movies': limited_movies,
108
+ 'series': limited_series
109
  }
110
 
111
  # Return the results in a JSON response