Spaces:
Sleeping
Sleeping
| """Example data for the movie predictor.""" | |
| from __future__ import annotations | |
| from typing import Any | |
| def get_example_data(feature_options: dict[str, list[str]]) -> list[list[Any]]: | |
| """Generate example movie data for quick testing.""" | |
| genres = feature_options.get("genres", [])[:3] | |
| companies = feature_options.get("production_companies", [])[:2] | |
| keywords = feature_options.get("Keywords", [])[:3] | |
| cast = feature_options.get("cast", [])[:2] | |
| return [ | |
| [ | |
| # Blockbuster Example (similar to Avengers-type movies) | |
| 200_000_000, # budget | |
| 64.0, # popularity | |
| 143, # runtime | |
| "2019-04-26", # release_date | |
| "en", # original_language | |
| True, # belongs_to_collection | |
| True, # homepage | |
| "The Final Showdown", # title | |
| "Whatever it takes", # tagline | |
| "After devastating events, heroes must assemble once more to undo chaos and restore order to the universe.", # overview | |
| 25, # num_of_cast | |
| 50, # num_of_crew | |
| 8, # gender_cast_1 (female) | |
| 15, # gender_cast_2 (male) | |
| 2, # count_cast_other | |
| genres[:3] if len(genres) >= 3 else genres, # genres | |
| companies[:2] if len(companies) >= 2 else companies, # production_companies | |
| keywords[:5] if len(keywords) >= 5 else keywords, # keywords | |
| cast[:5] if len(cast) >= 5 else cast, # cast | |
| ], | |
| [ | |
| # Mid-Budget Comedy | |
| 40_000_000, # budget | |
| 8.2, # popularity | |
| 113, # runtime | |
| "2015-08-06", # release_date | |
| "en", # original_language | |
| True, # belongs_to_collection | |
| True, # homepage | |
| "Royal Wedding", # title | |
| "Royalty has its responsibilities", # tagline | |
| "A young princess must navigate royal duties while finding true love before her coronation.", # overview | |
| 20, # num_of_cast | |
| 30, # num_of_crew | |
| 12, # gender_cast_1 (female) | |
| 8, # gender_cast_2 (male) | |
| 0, # count_cast_other | |
| genres[:2] if len(genres) >= 2 else genres, # genres | |
| companies[:1] if companies else [], # production_companies | |
| keywords[:3] if len(keywords) >= 3 else keywords, # keywords | |
| cast[:3] if len(cast) >= 3 else cast, # cast | |
| ], | |
| [ | |
| # Low-Budget Thriller | |
| 3_300_000, # budget | |
| 35.0, # popularity | |
| 105, # runtime | |
| "2014-10-10", # release_date | |
| "en", # original_language | |
| False, # belongs_to_collection | |
| True, # homepage | |
| "Perfect Rhythm", # title | |
| "Greatness comes at a price", # tagline | |
| "A talented musician pushes beyond limits under the guidance of a demanding instructor.", # overview | |
| 15, # num_of_cast | |
| 25, # num_of_crew | |
| 3, # gender_cast_1 (female) | |
| 10, # gender_cast_2 (male) | |
| 2, # count_cast_other | |
| genres[:1] if genres else [], # genres | |
| companies[:2] if len(companies) >= 2 else companies, # production_companies | |
| keywords[:4] if len(keywords) >= 4 else keywords, # keywords | |
| cast[:2] if len(cast) >= 2 else cast, # cast | |
| ], | |
| ] | |