Initial D

README.md

@@ -1,12 +1,9 @@
 ---
-title: Ai Pokemon Card
-emoji: 📚
-colorFrom: green
+title: This Pokémon Does Not Exist
+emoji: 🧬
+colorFrom: blue
 colorTo: yellow
 sdk: gradio
 app_file: app.py
 pinned: false
-license: mit
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,61 @@
+import time
+from flask import Flask, jsonify, render_template, request
+
+from modules.details import load_lists, rand_details
+from modules.inference import generate_image
+
+app = Flask(__name__)
+app.config['TEMPLATES_AUTO_RELOAD'] = True
+
+
+@app.route('/')
+def index():
+    return render_template('index.html', **rand_details())
+
+
+tasks = {}
+
+
+@app.route('/task/create')
+def create_task():
+    prompt = request.args.get('prompt') or "покемон"
+
+    task_id = f"{str(time.time())}_{prompt}"
+
+    tasks[task_id] = {
+        "task_id": task_id,
+        "prompt": prompt,
+        "status": "pending",
+        "poll_count": 0,
+    }
+
+    return jsonify(tasks[task_id])
+
+
+@app.route('/task/queue')
+def queue_task():
+    task_id = request.args.get('task_id')
+
+    tasks[task_id]["value"] = generate_image(tasks[task_id]["prompt"])
+
+    tasks[task_id]["status"] = "complete"
+
+    return jsonify(tasks[task_id])
+
+
+@app.route('/task/poll')
+def poll_task():
+    task_id = request.args.get('task_id')
+
+    tasks[task_id]["poll_count"] += 1
+
+    return jsonify(tasks[task_id])
+
+
+@app.route('/details')
+def generate_details():
+    return jsonify(rand_details())
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)

lists/names.json

@@ -0,0 +1,907 @@
+[
+  "Bulbasaur",
+  "Ivysaur",
+  "Venusaur",
+  "Charmander",
+  "Charmeleon",
+  "Charizard",
+  "Squirtle",
+  "Wartortle",
+  "Blastoise",
+  "Caterpie",
+  "Metapod",
+  "Butterfree",
+  "Weedle",
+  "Kakuna",
+  "Beedrill",
+  "Pidgey",
+  "Pidgeotto",
+  "Pidgeot",
+  "Rattata",
+  "Raticate",
+  "Spearow",
+  "Fearow",
+  "Ekans",
+  "Arbok",
+  "Pikachu",
+  "Raichu",
+  "Sandshrew",
+  "Sandslash",
+  "Nidoran♀",
+  "Nidorina",
+  "Nidoqueen",
+  "Nidoran♂",
+  "Nidorino",
+  "Nidoking",
+  "Clefairy",
+  "Clefable",
+  "Vulpix",
+  "Ninetales",
+  "Jigglypuff",
+  "Wigglytuff",
+  "Zubat",
+  "Golbat",
+  "Oddish",
+  "Gloom",
+  "Vileplume",
+  "Paras",
+  "Parasect",
+  "Venonat",
+  "Venomoth",
+  "Diglett",
+  "Dugtrio",
+  "Meowth",
+  "Persian",
+  "Psyduck",
+  "Golduck",
+  "Mankey",
+  "Primeape",
+  "Growlithe",
+  "Arcanine",
+  "Poliwag",
+  "Poliwhirl",
+  "Poliwrath",
+  "Abra",
+  "Kadabra",
+  "Alakazam",
+  "Machop",
+  "Machoke",
+  "Machamp",
+  "Bellsprout",
+  "Weepinbell",
+  "Victreebel",
+  "Tentacool",
+  "Tentacruel",
+  "Geodude",
+  "Graveler",
+  "Golem",
+  "Ponyta",
+  "Rapidash",
+  "Slowpoke",
+  "Slowbro",
+  "Magnemite",
+  "Magneton",
+  "Farfetch'd",
+  "Doduo",
+  "Dodrio",
+  "Seel",
+  "Dewgong",
+  "Grimer",
+  "Muk",
+  "Shellder",
+  "Cloyster",
+  "Gastly",
+  "Haunter",
+  "Gengar",
+  "Onix",
+  "Drowzee",
+  "Hypno",
+  "Krabby",
+  "Kingler",
+  "Voltorb",
+  "Electrode",
+  "Exeggcute",
+  "Exeggutor",
+  "Cubone",
+  "Marowak",
+  "Hitmonlee",
+  "Hitmonchan",
+  "Lickitung",
+  "Koffing",
+  "Weezing",
+  "Rhyhorn",
+  "Rhydon",
+  "Chansey",
+  "Tangela",
+  "Kangaskhan",
+  "Horsea",
+  "Seadra",
+  "Goldeen",
+  "Seaking",
+  "Staryu",
+  "Starmie",
+  "Mr. Mime",
+  "Scyther",
+  "Jynx",
+  "Electabuzz",
+  "Magmar",
+  "Pinsir",
+  "Tauros",
+  "Magikarp",
+  "Gyarados",
+  "Lapras",
+  "Ditto",
+  "Eevee",
+  "Vaporeon",
+  "Jolteon",
+  "Flareon",
+  "Porygon",
+  "Omanyte",
+  "Omastar",
+  "Kabuto",
+  "Kabutops",
+  "Aerodactyl",
+  "Snorlax",
+  "Articuno",
+  "Zapdos",
+  "Moltres",
+  "Dratini",
+  "Dragonair",
+  "Dragonite",
+  "Mewtwo",
+  "Mew",
+  "Chikorita",
+  "Bayleef",
+  "Meganium",
+  "Cyndaquil",
+  "Quilava",
+  "Typhlosion",
+  "Totodile",
+  "Croconaw",
+  "Feraligatr",
+  "Sentret",
+  "Furret",
+  "Hoothoot",
+  "Noctowl",
+  "Ledyba",
+  "Ledian",
+  "Spinarak",
+  "Ariados",
+  "Crobat",
+  "Chinchou",
+  "Lanturn",
+  "Pichu",
+  "Cleffa",
+  "Igglybuff",
+  "Togepi",
+  "Togetic",
+  "Natu",
+  "Xatu",
+  "Mareep",
+  "Flaaffy",
+  "Ampharos",
+  "Bellossom",
+  "Marill",
+  "Azumarill",
+  "Sudowoodo",
+  "Politoed",
+  "Hoppip",
+  "Skiploom",
+  "Jumpluff",
+  "Aipom",
+  "Sunkern",
+  "Sunflora",
+  "Yanma",
+  "Wooper",
+  "Quagsire",
+  "Espeon",
+  "Umbreon",
+  "Murkrow",
+  "Slowking",
+  "Misdreavus",
+  "Unown",
+  "Wobbuffet",
+  "Girafarig",
+  "Pineco",
+  "Forretress",
+  "Dunsparce",
+  "Gligar",
+  "Steelix",
+  "Snubbull",
+  "Granbull",
+  "Qwilfish",
+  "Scizor",
+  "Shuckle",
+  "Heracross",
+  "Sneasel",
+  "Teddiursa",
+  "Ursaring",
+  "Slugma",
+  "Magcargo",
+  "Swinub",
+  "Piloswine",
+  "Corsola",
+  "Remoraid",
+  "Octillery",
+  "Delibird",
+  "Mantine",
+  "Skarmory",
+  "Houndour",
+  "Houndoom",
+  "Kingdra",
+  "Phanpy",
+  "Donphan",
+  "Porygon2",
+  "Stantler",
+  "Smeargle",
+  "Tyrogue",
+  "Hitmontop",
+  "Smoochum",
+  "Elekid",
+  "Magby",
+  "Miltank",
+  "Blissey",
+  "Raikou",
+  "Entei",
+  "Suicune",
+  "Larvitar",
+  "Pupitar",
+  "Tyranitar",
+  "Lugia",
+  "Ho-Oh",
+  "Celebi",
+  "Treecko",
+  "Grovyle",
+  "Sceptile",
+  "Torchic",
+  "Combusken",
+  "Blaziken",
+  "Mudkip",
+  "Marshtomp",
+  "Swampert",
+  "Poochyena",
+  "Mightyena",
+  "Zigzagoon",
+  "Linoone",
+  "Wurmple",
+  "Silcoon",
+  "Beautifly",
+  "Cascoon",
+  "Dustox",
+  "Lotad",
+  "Lombre",
+  "Ludicolo",
+  "Seedot",
+  "Nuzleaf",
+  "Shiftry",
+  "Taillow",
+  "Swellow",
+  "Wingull",
+  "Pelipper",
+  "Ralts",
+  "Kirlia",
+  "Gardevoir",
+  "Surskit",
+  "Masquerain",
+  "Shroomish",
+  "Breloom",
+  "Slakoth",
+  "Vigoroth",
+  "Slaking",
+  "Nincada",
+  "Ninjask",
+  "Shedinja",
+  "Whismur",
+  "Loudred",
+  "Exploud",
+  "Makuhita",
+  "Hariyama",
+  "Azurill",
+  "Nosepass",
+  "Skitty",
+  "Delcatty",
+  "Sableye",
+  "Mawile",
+  "Aron",
+  "Lairon",
+  "Aggron",
+  "Meditite",
+  "Medicham",
+  "Electrike",
+  "Manectric",
+  "Plusle",
+  "Minun",
+  "Volbeat",
+  "Illumise",
+  "Roselia",
+  "Gulpin",
+  "Swalot",
+  "Carvanha",
+  "Sharpedo",
+  "Wailmer",
+  "Wailord",
+  "Numel",
+  "Camerupt",
+  "Torkoal",
+  "Spoink",
+  "Grumpig",
+  "Spinda",
+  "Trapinch",
+  "Vibrava",
+  "Flygon",
+  "Cacnea",
+  "Cacturne",
+  "Swablu",
+  "Altaria",
+  "Zangoose",
+  "Seviper",
+  "Lunatone",
+  "Solrock",
+  "Barboach",
+  "Whiscash",
+  "Corphish",
+  "Crawdaunt",
+  "Baltoy",
+  "Claydol",
+  "Lileep",
+  "Cradily",
+  "Anorith",
+  "Armaldo",
+  "Feebas",
+  "Milotic",
+  "Castform",
+  "Kecleon",
+  "Shuppet",
+  "Banette",
+  "Duskull",
+  "Dusclops",
+  "Tropius",
+  "Chimecho",
+  "Absol",
+  "Wynaut",
+  "Snorunt",
+  "Glalie",
+  "Spheal",
+  "Sealeo",
+  "Walrein",
+  "Clamperl",
+  "Huntail",
+  "Gorebyss",
+  "Relicanth",
+  "Luvdisc",
+  "Bagon",
+  "Shelgon",
+  "Salamence",
+  "Beldum",
+  "Metang",
+  "Metagross",
+  "Regirock",
+  "Regice",
+  "Registeel",
+  "Latias",
+  "Latios",
+  "Kyogre",
+  "Groudon",
+  "Rayquaza",
+  "Jirachi",
+  "Deoxys",
+  "Turtwig",
+  "Grotle",
+  "Torterra",
+  "Chimchar",
+  "Monferno",
+  "Infernape",
+  "Piplup",
+  "Prinplup",
+  "Empoleon",
+  "Starly",
+  "Staravia",
+  "Staraptor",
+  "Bidoof",
+  "Bibarel",
+  "Kricketot",
+  "Kricketune",
+  "Shinx",
+  "Luxio",
+  "Luxray",
+  "Budew",
+  "Roserade",
+  "Cranidos",
+  "Rampardos",
+  "Shieldon",
+  "Bastiodon",
+  "Burmy",
+  "Wormadam",
+  "Mothim",
+  "Combee",
+  "Vespiquen",
+  "Pachirisu",
+  "Buizel",
+  "Floatzel",
+  "Cherubi",
+  "Cherrim",
+  "Shellos",
+  "Gastrodon",
+  "Ambipom",
+  "Drifloon",
+  "Drifblim",
+  "Buneary",
+  "Lopunny",
+  "Mismagius",
+  "Honchkrow",
+  "Glameow",
+  "Purugly",
+  "Chingling",
+  "Stunky",
+  "Skuntank",
+  "Bronzor",
+  "Bronzong",
+  "Bonsly",
+  "Mime Jr.",
+  "Happiny",
+  "Chatot",
+  "Spiritomb",
+  "Gible",
+  "Gabite",
+  "Garchomp",
+  "Munchlax",
+  "Riolu",
+  "Lucario",
+  "Hippopotas",
+  "Hippowdon",
+  "Skorupi",
+  "Drapion",
+  "Croagunk",
+  "Toxicroak",
+  "Carnivine",
+  "Finneon",
+  "Lumineon",
+  "Mantyke",
+  "Snover",
+  "Abomasnow",
+  "Weavile",
+  "Magnezone",
+  "Lickilicky",
+  "Rhyperior",
+  "Tangrowth",
+  "Electivire",
+  "Magmortar",
+  "Togekiss",
+  "Yanmega",
+  "Leafeon",
+  "Glaceon",
+  "Gliscor",
+  "Mamoswine",
+  "Porygon-Z",
+  "Gallade",
+  "Probopass",
+  "Dusknoir",
+  "Froslass",
+  "Rotom",
+  "Uxie",
+  "Mesprit",
+  "Azelf",
+  "Dialga",
+  "Palkia",
+  "Heatran",
+  "Regigigas",
+  "Giratina",
+  "Cresselia",
+  "Phione",
+  "Manaphy",
+  "Darkrai",
+  "Shaymin",
+  "Arceus",
+  "Victini",
+  "Snivy",
+  "Servine",
+  "Serperior",
+  "Tepig",
+  "Pignite",
+  "Emboar",
+  "Oshawott",
+  "Dewott",
+  "Samurott",
+  "Patrat",
+  "Watchog",
+  "Lillipup",
+  "Herdier",
+  "Stoutland",
+  "Purrloin",
+  "Liepard",
+  "Pansage",
+  "Simisage",
+  "Pansear",
+  "Simisear",
+  "Panpour",
+  "Simipour",
+  "Munna",
+  "Musharna",
+  "Pidove",
+  "Tranquill",
+  "Unfezant",
+  "Blitzle",
+  "Zebstrika",
+  "Roggenrola",
+  "Boldore",
+  "Gigalith",
+  "Woobat",
+  "Swoobat",
+  "Drilbur",
+  "Excadrill",
+  "Audino",
+  "Timburr",
+  "Gurdurr",
+  "Conkeldurr",
+  "Tympole",
+  "Palpitoad",
+  "Seismitoad",
+  "Throh",
+  "Sawk",
+  "Sewaddle",
+  "Swadloon",
+  "Leavanny",
+  "Venipede",
+  "Whirlipede",
+  "Scolipede",
+  "Cottonee",
+  "Whimsicott",
+  "Petilil",
+  "Lilligant",
+  "Basculin",
+  "Sandile",
+  "Krokorok",
+  "Krookodile",
+  "Darumaka",
+  "Darmanitan",
+  "Maractus",
+  "Dwebble",
+  "Crustle",
+  "Scraggy",
+  "Scrafty",
+  "Sigilyph",
+  "Yamask",
+  "Cofagrigus",
+  "Tirtouga",
+  "Carracosta",
+  "Archen",
+  "Archeops",
+  "Trubbish",
+  "Garbodor",
+  "Zorua",
+  "Zoroark",
+  "Minccino",
+  "Cinccino",
+  "Gothita",
+  "Gothorita",
+  "Gothitelle",
+  "Solosis",
+  "Duosion",
+  "Reuniclus",
+  "Ducklett",
+  "Swanna",
+  "Vanillite",
+  "Vanillish",
+  "Vanilluxe",
+  "Deerling",
+  "Sawsbuck",
+  "Emolga",
+  "Karrablast",
+  "Escavalier",
+  "Foongus",
+  "Amoonguss",
+  "Frillish",
+  "Jellicent",
+  "Alomomola",
+  "Joltik",
+  "Galvantula",
+  "Ferroseed",
+  "Ferrothorn",
+  "Klink",
+  "Klang",
+  "Klinklang",
+  "Tynamo",
+  "Eelektrik",
+  "Eelektross",
+  "Elgyem",
+  "Beheeyem",
+  "Litwick",
+  "Lampent",
+  "Chandelure",
+  "Axew",
+  "Fraxure",
+  "Haxorus",
+  "Cubchoo",
+  "Beartic",
+  "Cryogonal",
+  "Shelmet",
+  "Accelgor",
+  "Stunfisk",
+  "Mienfoo",
+  "Mienshao",
+  "Druddigon",
+  "Golett",
+  "Golurk",
+  "Pawniard",
+  "Bisharp",
+  "Bouffalant",
+  "Rufflet",
+  "Braviary",
+  "Vullaby",
+  "Mandibuzz",
+  "Heatmor",
+  "Durant",
+  "Deino",
+  "Zweilous",
+  "Hydreigon",
+  "Larvesta",
+  "Volcarona",
+  "Cobalion",
+  "Terrakion",
+  "Virizion",
+  "Tornadus",
+  "Thundurus",
+  "Reshiram",
+  "Zekrom",
+  "Landorus",
+  "Kyurem",
+  "Keldeo",
+  "Meloetta",
+  "Genesect",
+  "Chespin",
+  "Quilladin",
+  "Chesnaught",
+  "Fennekin",
+  "Braixen",
+  "Delphox",
+  "Froakie",
+  "Frogadier",
+  "Greninja",
+  "Bunnelby",
+  "Diggersby",
+  "Fletchling",
+  "Fletchinder",
+  "Talonflame",
+  "Scatterbug",
+  "Spewpa",
+  "Vivillon",
+  "Litleo",
+  "Pyroar",
+  "Flabébé",
+  "Floette",
+  "Florges",
+  "Skiddo",
+  "Gogoat",
+  "Pancham",
+  "Pangoro",
+  "Furfrou",
+  "Espurr",
+  "Meowstic",
+  "Honedge",
+  "Doublade",
+  "Aegislash",
+  "Spritzee",
+  "Aromatisse",
+  "Swirlix",
+  "Slurpuff",
+  "Inkay",
+  "Malamar",
+  "Binacle",
+  "Barbaracle",
+  "Skrelp",
+  "Dragalge",
+  "Clauncher",
+  "Clawitzer",
+  "Helioptile",
+  "Heliolisk",
+  "Tyrunt",
+  "Tyrantrum",
+  "Amaura",
+  "Aurorus",
+  "Sylveon",
+  "Hawlucha",
+  "Dedenne",
+  "Carbink",
+  "Goomy",
+  "Sliggoo",
+  "Goodra",
+  "Klefki",
+  "Phantump",
+  "Trevenant",
+  "Pumpkaboo",
+  "Gourgeist",
+  "Bergmite",
+  "Avalugg",
+  "Noibat",
+  "Noivern",
+  "Xerneas",
+  "Yveltal",
+  "Zygarde",
+  "Diancie",
+  "Hoopa",
+  "Volcanion",
+  "Rowlet",
+  "Dartrix",
+  "Decidueye",
+  "Litten",
+  "Torracat",
+  "Incineroar",
+  "Popplio",
+  "Brionne",
+  "Primarina",
+  "Pikipek",
+  "Trumbeak",
+  "Toucannon",
+  "Yungoos",
+  "Gumshoos",
+  "Grubbin",
+  "Charjabug",
+  "Vikavolt",
+  "Crabrawler",
+  "Crabominable",
+  "Oricorio",
+  "Cutiefly",
+  "Ribombee",
+  "Rockruff",
+  "Lycanroc",
+  "Wishiwashi",
+  "Mareanie",
+  "Toxapex",
+  "Mudbray",
+  "Mudsdale",
+  "Dewpider",
+  "Araquanid",
+  "Fomantis",
+  "Lurantis",
+  "Morelull",
+  "Shiinotic",
+  "Salandit",
+  "Salazzle",
+  "Stufful",
+  "Bewear",
+  "Bounsweet",
+  "Steenee",
+  "Tsareena",
+  "Comfey",
+  "Oranguru",
+  "Passimian",
+  "Wimpod",
+  "Golisopod",
+  "Sandygast",
+  "Palossand",
+  "Pyukumuku",
+  "Type: Null",
+  "Silvally",
+  "Minior",
+  "Komala",
+  "Turtonator",
+  "Togedemaru",
+  "Mimikyu",
+  "Bruxish",
+  "Drampa",
+  "Dhelmise",
+  "Jangmo-o",
+  "Hakamo-o",
+  "Kommo-o",
+  "Tapu Koko",
+  "Tapu Lele",
+  "Tapu Bulu",
+  "Tapu Fini",
+  "Cosmog",
+  "Cosmoem",
+  "Solgaleo",
+  "Lunala",
+  "Nihilego",
+  "Buzzwole",
+  "Pheromosa",
+  "Xurkitree",
+  "Celesteela",
+  "Kartana",
+  "Guzzlord",
+  "Necrozma",
+  "Magearna",
+  "Marshadow",
+  "Poipole",
+  "Naganadel",
+  "Stakataka",
+  "Blacephalon",
+  "Zeraora",
+  "Meltan",
+  "Melmetal",
+  "Grookey",
+  "Thwackey",
+  "Rillaboom",
+  "Scorbunny",
+  "Raboot",
+  "Cinderace",
+  "Sobble",
+  "Drizzile",
+  "Inteleon",
+  "Skwovet",
+  "Greedent",
+  "Rookidee",
+  "Corvisquire",
+  "Corviknight",
+  "Blipbug",
+  "Dottler",
+  "Orbeetle",
+  "Nickit",
+  "Thievul",
+  "Gossifleur",
+  "Eldegoss",
+  "Wooloo",
+  "Dubwool",
+  "Chewtle",
+  "Drednaw",
+  "Yamper",
+  "Boltund",
+  "Rolycoly",
+  "Carkol",
+  "Coalossal",
+  "Applin",
+  "Flapple",
+  "Appletun",
+  "Silicobra",
+  "Sandaconda",
+  "Cramorant",
+  "Arrokuda",
+  "Barraskewda",
+  "Toxel",
+  "Toxtricity",
+  "Sizzlipede",
+  "Centiskorch",
+  "Clobbopus",
+  "Grapploct",
+  "Sinistea",
+  "Polteageist",
+  "Hatenna",
+  "Hattrem",
+  "Hatterene",
+  "Impidimp",
+  "Morgrem",
+  "Grimmsnarl",
+  "Obstagoon",
+  "Perrserker",
+  "Cursola",
+  "Sirfetch'd",
+  "Mr. Rime",
+  "Runerigus",
+  "Milcery",
+  "Alcremie",
+  "Falinks",
+  "Pincurchin",
+  "Snom",
+  "Frosmoth",
+  "Stonjourner",
+  "Eiscue",
+  "Indeedee",
+  "Morpeko",
+  "Cufant",
+  "Copperajah",
+  "Dracozolt",
+  "Arctozolt",
+  "Dracovish",
+  "Arctovish",
+  "Duraludon",
+  "Dreepy",
+  "Drakloak",
+  "Dragapult",
+  "Zacian",
+  "Zamazenta",
+  "Eternatus",
+  "Kubfu",
+  "Urshifu",
+  "Zarude",
+  "Regieleki",
+  "Regidrago",
+  "Glastrier",
+  "Spectrier",
+  "Calyrex",
+  "Wyrdeer",
+  "Kleavor",
+  "Ursaluna",
+  "Basculegion",
+  "Sneasler",
+  "Overqwil",
+  "Enamorus"
+]

modules/details.py

@@ -0,0 +1,124 @@
+import random
+import json
+
+
+def load_lists(list_names, base_dir="lists"):
+    lists = {}
+
+    for name in list_names:
+        with open(f"{base_dir}/{name}.json", "r") as f:
+            lists[name] = json.load(f)
+
+    return lists
+
+
+def rand_name(names):
+    return random.choices(names)[0]
+
+
+def rand_hp():
+    # Weights from https://bulbapedia.bulbagarden.net/wiki/HP_(TCG)
+
+    hp_range = list(range(30, 340 + 1, 10))
+
+    weights = [156, 542, 1264, 1727, 1477, 1232, 1008, 640, 436, 515, 469, 279, 188,
+               131, 132, 132, 56, 66, 97, 74, 23, 24, 25, 7, 15, 6, 0, 12, 18, 35, 18, 3]
+
+    return random.choices(hp_range, weights)[0]
+
+
+def rand_type(types=['Grass', 'Fire', 'Water', 'Lightning', 'Fighting',
+                     'Psychic', 'Colorless', 'Darkness', 'Metal', 'Dragon', 'Fairy'], can_be_none=False):
+    if can_be_none:
+        return random.choices([random.choices(types)[0], None])[0]
+    else:
+        return random.choices(types)[0]
+
+
+def rand_species(species):
+    random_species = random.choices(species)[0]
+
+    return f'{random_species.capitalize()}'
+
+
+def rand_length():
+    # Weights from https://bulbapedia.bulbagarden.net/wiki/List_of_Pok%C3%A9mon_by_height
+
+    feet_ranges = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '16',
+                   '17', '18', '19', '20', '21', '22', '23', '24', '26', '28', '30', '32', '34', '35', '47', '65', '328']
+
+    weights = [30, 220, 230, 176, 130, 109, 63, 27, 17, 17, 5, 5, 6,
+               4, 3, 2, 2, 2, 1, 2, 3, 1, 2, 1, 1, 1, 2, 1, 1, 2, 1, 1, 1]
+
+    return {
+        "feet": random.choices(feet_ranges, weights)[0],
+        "inches": random.randrange(0, 11 + 1, 1)
+    }
+
+
+def rand_weight():
+    # Weights from https://bulbapedia.bulbagarden.net/wiki/List_of_Pok%C3%A9mon_by_weight
+
+    weight_ranges = [{"start": 1, "end": 22}, {"start": 22, "end": 44}, {"start": 44, "end": 55}, {"start": 55, "end": 110}, {"start": 110, "end": 132}, {"start": 132, "end": 218}, {"start": 218, "end": 220}, {"start": 221, "end": 226}, {
+        "start": 226, "end": 331}, {"start": 331, "end": 441}, {"start": 441, "end": 451}, {"start": 452, "end": 661}, {"start": 661, "end": 677}, {"start": 677, "end": 793}, {"start": 794, "end": 903}, {"start": 903, "end": 2204}]
+
+    # 'weights' as in statistical weightings
+    weights = [271, 145, 53, 204, 57, 122, 1, 11, 57, 28, 7, 34, 4, 17, 5, 31]
+
+    start, end = random.choices(weight_ranges, weights)[0].values()
+
+    random_weight = random.randrange(start, end + 1, 1)
+
+    return f'{random_weight} lbs.'
+
+
+def rand_attack(attacks, name):
+    random_attack = random.choices(attacks)[0]
+    random_attack['text'] = random_attack['text'].replace('<name>', name)
+
+    return random_attack
+
+
+def rand_attacks(attacks, name, n=2):
+    attack1 = rand_attack(attacks, name)
+
+    if n > 1:
+        attack2 = rand_attack(attacks, name)
+        while attack1['text'] == attack2['text']:
+            attack2 = rand_attack(attacks, name)
+        return [attack1, attack2]
+    else:
+        return [attack1]
+
+    return attacks
+
+
+def rand_retreat():
+    return random.randrange(0, 4, 1)
+
+
+def rand_description(descriptions):
+    return random.choices(descriptions)[0]
+
+
+def rand_rarity():
+    return random.choices(['●', '◆', '★'], [10, 5, 1])[0]
+
+
+def rand_details(lists=load_lists(['attacks', 'descriptions', 'names', 'species'])):
+    name = rand_name(lists["names"])
+
+    return {
+        "name": name,
+        "hp": rand_hp(),
+        "energy_type": rand_type(),
+        "species": rand_species(lists["species"]),
+        "length": rand_length(),
+        "weight": rand_weight(),
+        "attacks": rand_attacks(lists["attacks"], name),
+        "weakness": rand_type(can_be_none=True),
+        "resistance": rand_type(can_be_none=True),
+        "retreat": rand_retreat(),
+        "description": rand_description(lists["descriptions"]),
+        "rarity": rand_rarity(),
+    }

modules/inference.py

@@ -0,0 +1,61 @@
+print("Preparing for inference...")  # noqa
+
+from rudalle.pipelines import generate_images
+from rudalle import get_rudalle_model, get_tokenizer, get_vae
+from huggingface_hub import hf_hub_url, cached_download
+import torch
+from io import BytesIO
+import base64
+
+print(f"GPUs available: {torch.cuda.device_count()}")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+fp16 = torch.cuda.is_available()
+
+file_dir = "./models"
+file_name = "pytorch_model.bin"
+config_file_url = hf_hub_url(
+    repo_id="minimaxir/ai-generated-pokemon-rudalle", filename=file_name)
+cached_download(config_file_url, cache_dir=file_dir, force_filename=file_name)
+
+model = get_rudalle_model('Malevich', pretrained=False,
+                          fp16=fp16, device=device)
+model.load_state_dict(torch.load(
+    f"{file_dir}/{file_name}", map_location=f"{'cuda:0' if torch.cuda.is_available() else 'cpu'}"))
+
+vae = get_vae().to(device)
+tokenizer = get_tokenizer()
+
+print("Ready for inference")
+
+
+def english_to_russian(english_string):
+    word_map = {
+        "grass": "трава",
+        "fire": "Пожар",
+        "water": "вода",
+        "lightning": "молния",
+        "fighting": "борьба",
+        "psychic": "психический",
+        "colorless": "бесцветный",
+        "darkness": "темнота",
+        "metal": "металл",
+        "dragon": "Дракон",
+        "fairy": "сказочный"
+    }
+
+    return word_map[english_string.lower()]
+
+
+def generate_image(prompt):
+    if prompt.lower() in ['grass', 'fire', 'water', 'lightning', 'fighting', 'psychic', 'colorless', 'darkness', 'metal', 'dragon', 'fairy']:
+        prompt = english_to_russian(prompt)
+
+    result, _ = generate_images(
+        prompt, tokenizer, model, vae, top_k=2048, images_num=1, top_p=0.995)
+
+    buffer = BytesIO()
+    result[0].save(buffer, format="PNG")
+    base64_bytes = base64.b64encode(buffer.getvalue())
+    base64_string = base64_bytes.decode("UTF-8")
+
+    return "data:image/png;base64," + base64_string

requirements.txt

@@ -0,0 +1,2 @@
+Flask
+rudalle==1.0.0

static/index.js

@@ -0,0 +1,239 @@
+/* HTML generation */
+
+const TYPES = {
+  Grass: '🍃',
+  Fire: '🔥',
+  Water: '💧',
+  Lightning: '⚡',
+  Fighting: '✊',
+  Psychic: '👁️',
+  Colorless: '⭐',
+  Darkness: '🌑',
+  Metal: '⚙️',
+  Dragon: '🐲',
+  Fairy: '🧚',
+};
+
+const energyHTML = (type, types = TYPES) => {
+  return `<span title="${type} energy" class="energy ${type.toLowerCase()}">${types[type]}</span>`;
+};
+
+const attackDescriptionHTML = (text) => {
+  if (!text) {
+    return '';
+  }
+
+  let fontSize;
+
+  if (text.length > 185) {
+    fontSize = 0.7;
+  } else if (text.length > 140) {
+    fontSize = 0.8;
+  } else if (text.length > 90) {
+    fontSize = 0.9;
+  }
+
+  return `<span class="attack-details"${fontSize ? ` style="font-size: ${fontSize.toString()}em"` : ''}>${text}</span>`;
+};
+
+const attackRowsHTML = (attacks) => {
+  return attacks
+    .map((attack) => {
+      const { cost, damage, name, text } = attack;
+
+      return `
+<li class="attacks-row grid-three">
+  <div class="attack-cost">
+    ${cost.map((energy) => energyHTML(energy)).join('')}
+  </div>
+  <span class="attack-text">
+    <span class="attack-name">${name}</span>
+    ${attackDescriptionHTML(text)}
+  </span>
+  <span class="attack-damage">${damage ? damage : ''}</span>
+</li>
+<hr role="presentation" />`;
+    })
+    .join('');
+};
+
+const cardHTML = (details) => {
+  const { hp, energy_type, species, length, weight, attacks, weakness, resistance, retreat, description, rarity } =
+    details;
+
+  const poke_name = details.name; // `name` would be reserved JS word
+
+  return `
+<div class="pokecard ${energy_type.toLowerCase()}" data-displayed="true">
+  <p class="evolves">Basic Pokémon</p>
+  <header>
+    <h1 class="name">${poke_name}</h1>
+    <div>
+      <span class="hp">${hp} HP</span>
+      ${energyHTML(energy_type)}
+    </div>
+  </header>
+  <img class="picture frame" alt="AI generated Pokémon called ${poke_name}" />
+  <div class="species frame">
+    ${species} Pokémon. Length: ${length.feet}'${length.inches}", Weight: ${weight}
+  </div>
+  <ul class="attacks">
+    ${attackRowsHTML(attacks)}
+  </ul>
+  <div class="multipliers">
+    <div class="weakness">
+      <span>weakness</span>
+      ${weakness ? energyHTML(weakness) : ''}
+    </div>
+    <div class="resistance">
+      <span>resistance</span>
+      ${resistance ? energyHTML(resistance) : ''}
+      <span class="resistance-total"
+        >${resistance ? '-30' : ''}</span
+      >
+    </div>
+    <div class="retreat-cost">
+      <span>retreat cost</span>
+      <div>${energyHTML('Colorless').repeat(retreat)}</div>
+    </div>
+  </div>
+  <p class="description frame">${description}</p>
+  <div class="footer grid-three">
+    <span
+      ><a
+        href="https://huggingface.co/minimaxir/ai-generated-pokemon-rudalle"
+        >Illus. Max Woolf</a
+      ></span
+    >
+    <span><a href="">2022 Hugging Face</a></span>
+    <span>${rarity}</span>
+  </div>
+</div>`;
+};
+
+/* Utility */
+
+const getBasePath = () => {
+  return document.location.origin + document.location.pathname;
+};
+
+const generateDetails = async () => {
+  const details = await fetch(`${getBasePath()}/details`);
+  return await details.json();
+};
+
+const createTask = async (prompt) => {
+  const taskResponse = await fetch(`${getBasePath()}task/create?prompt=${prompt}`);
+  const task = await taskResponse.json();
+
+  return task;
+};
+
+const queueTask = (task_id) => {
+  fetch(`${getBasePath()}task/queue?task_id=${task_id}`);
+};
+
+const pollTask = async (task) => {
+  const taskResponse = await fetch(`${getBasePath()}task/poll?task_id=${task.task_id}`);
+
+  return await taskResponse.json();
+};
+
+const longPollTask = async (task, interval = 5_000, max) => {
+  if (task.status === 'complete' || (max && task.poll_count > max)) {
+    return task;
+  }
+
+  const taskResponse = await fetch(`${getBasePath()}task/poll?task_id=${task.task_id}`);
+
+  task = await taskResponse.json();
+
+  if (task.status === 'complete' || task.poll_count > max) {
+    return task;
+  }
+
+  await new Promise((resolve) => setTimeout(resolve, interval));
+
+  return await longPollTask(task, interval, max);
+};
+
+/* DOM */
+
+const generateButton = document.querySelector('button.generate');
+
+const rotateCard = () => {
+  const RANGE = 0.1;
+  const INTERVAL = 13; // ~75 per second
+  let previousTime = 0;
+
+  // Throttle closure
+  return (card, containerMouseEvent) => {
+    const currentTime = performance.now();
+
+    if (currentTime - previousTime > INTERVAL) {
+      previousTime = currentTime;
+
+      const rect = card.getBoundingClientRect();
+
+      const rotateX = (containerMouseEvent.clientY - rect.y - rect.height / 2) * RANGE;
+      const rotateY = -(containerMouseEvent.clientX - rect.x - rect.width / 2) * RANGE;
+
+      card.style.setProperty('--rotate-x', rotateX + 'deg');
+      card.style.setProperty('--rotate-y', rotateY + 'deg');
+    }
+  };
+};
+
+const cardRotationInitiator = (renderSection) => {
+  let currentCard;
+
+  return () => {
+    let handleMouseMove;
+
+    if (currentCard) {
+      handleMouseMove = rotateCard().bind(null, currentCard);
+      renderSection.removeEventListener('mousemove', handleMouseMove, true);
+    }
+
+    const newCard = document.querySelector('.pokecard');
+
+    currentCard = newCard;
+
+    handleMouseMove = rotateCard().bind(null, newCard);
+    renderSection.addEventListener('mousemove', handleMouseMove, true);
+  };
+};
+
+generateButton.addEventListener('click', async () => {
+  const details = await generateDetails();
+  console.log({ details });
+
+  const renderSection = document.querySelector('section.render');
+  const durationSeconds = document.querySelector('.duration > .seconds');
+  const initialiseCardRotation = cardRotationInitiator(renderSection);
+  let duration = 0.0;
+
+  try {
+    const task = await createTask(details.energy_type);
+
+    queueTask(task.task_id);
+
+    const incrementSeconds = setInterval(() => {
+      duration += 0.1;
+      durationSeconds.textContent = duration.toFixed(1).toString();
+    }, 100);
+
+    const completedTask = await longPollTask(task);
+
+    clearInterval(incrementSeconds);
+
+    renderSection.innerHTML = cardHTML(details);
+
+    const picture = document.querySelector('img.picture');
+    picture.src = completedTask.value;
+
+    initialiseCardRotation();
+  } catch (err) {
+    console.error(err);
+  }
+});

static/style.css

@@ -0,0 +1,370 @@
+:root {
+  --card-h: 47;
+  --card-s: 95%;
+  --card-l: 58%;
+  --card-color: hsl(var(--card-h) var(--card-s) var(--card-l));
+}
+
+html,
+body {
+  overflow: scroll;
+}
+
+body {
+  display: grid;
+  place-items: center;
+  height: 100vh;
+  font-family: 'Gill Sans', 'Gill Sans Mt', 'sans-serif';
+  background-color: whitesmoke;
+}
+
+.duration {
+  text-align: right;
+}
+
+button {
+  padding: 0.5rem 1rem;
+  margin: 1rem;
+}
+
+section.render {
+  perspective: 100rem;
+  margin: 0.5rem;
+  height: 40rem;
+}
+
+.grass {
+  --h: 90;
+  --s: 60%;
+  --l: 40%;
+}
+.grass.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(55deg) drop-shadow(0 0 0.1rem green);
+}
+
+.fire {
+  --h: 0;
+  --s: 75%;
+  --l: 45%;
+}
+.fire.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(335deg) drop-shadow(0 0 0.1rem red);
+}
+
+.water {
+  --h: 210;
+  --s: 100%;
+  --l: 58%;
+}
+.water.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(180deg) drop-shadow(0 0 0.1rem cyan);
+}
+
+.lightning {
+  --h: 50;
+  --s: 100%;
+  --l: 58%;
+}
+.lightning.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(5deg) drop-shadow(0 0 0.1rem gold);
+}
+
+.fighting {
+  --h: 25;
+  --s: 72%;
+  --l: 36%;
+}
+.fighting.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(320deg) drop-shadow(0 0 0.1rem brown);
+}
+
+.psychic {
+  --h: 299;
+  --s: 43%;
+  --l: 44%;
+}
+.psychic.energy {
+  filter: grayscale(1) sepia(1) saturate(10) hue-rotate(240deg) drop-shadow(0 0 0.1rem purple);
+}
+
+.colorless {
+  --h: 21;
+  --s: 27%;
+  --l: 85%;
+}
+.colorless.energy {
+  font-size: 0.9em;
+  border-radius: 50%;
+  background-color: hsl(0 0% 100% / 15%);
+  filter: contrast(100) grayscale(1) drop-shadow(0 0 0.1rem white);
+}
+
+.darkness {
+  --h: 100;
+  --s: 3%;
+  --l: 17%;
+}
+.darkness.energy {
+  filter: drop-shadow(0 0 0.1rem black);
+}
+.darkness :not(.species) {
+  color: whitesmoke;
+}
+
+.metal {
+  --h: 240;
+  --s: 20%;
+  --l: 77%;
+}
+.metal.energy {
+  filter: drop-shadow(0 0 0.1rem silver);
+}
+
+.dragon {
+  --h: 30;
+  --s: 6%;
+  --l: 44%;
+}
+.dragon.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(15deg) drop-shadow(0 0 0.1rem gold);
+}
+
+.fairy {
+  --h: 334;
+  --s: 74%;
+  --l: 55%;
+}
+.fairy.energy {
+  filter: contrast(0.75) grayscale(1) sepia(1) saturate(10) hue-rotate(300deg) drop-shadow(0 0 0.1rem pink);
+}
+
+.pokecard,
+.pokecard * {
+  box-sizing: border-box;
+}
+
+.pokecard {
+  --color: hsl(var(--h) var(--s) var(--l));
+  --lighter: hsl(var(--h) var(--s) calc(var(--l) + 10%));
+  --lightest: hsl(var(--h) var(--s) calc(var(--l) + 30%));
+  --rotate-x: 0deg;
+  --rotate-y: 0deg;
+  --rotate-z: 0deg;
+  display: flex;
+  flex-direction: column;
+  width: 25rem;
+  aspect-ratio: 63 / 88;
+  padding: 0.5rem 1rem;
+  border: 1rem solid;
+  border-radius: 0.75rem;
+  border-color: var(--card-color);
+  background-image: linear-gradient(
+    45deg,
+    var(--lighter) 0%,
+    var(--lightest) 15%,
+    var(--lightest) 30%,
+    var(--color) 50%,
+    var(--lightest) 90%,
+    var(--lighter) 100%
+  );
+  transform-origin: center;
+  transform-style: preserve-3d;
+  transform: rotateX(var(--rotate-x)) rotateY(var(--rotate-y)) translate3d(10px, 10px, 10px);
+  transition: transform 0.5s ease-out;
+}
+
+.pokecard[data-displayed='true'] {
+  display: flex;
+}
+.pokecard[data-displayed='false'] {
+  display: none;
+}
+
+.pokecard :where(header, p) {
+  font-family: 'Gill Sans', 'Gill Sans Mt', 'sans-serif';
+}
+.pokecard :is(.evolves) {
+  font-family: 'Century Gothic', 'sans-serif';
+}
+.pokecard :where(header, p) {
+  font-family: 'Gill Sans', 'Gill Sans Mt';
+}
+
+.evolves {
+  margin: 0;
+  font-weight: bold;
+  font-size: 0.5rem;
+}
+
+header {
+  display: flex;
+  flex-direction: row;
+  justify-content: space-between;
+}
+
+header > * {
+  display: inline-block;
+}
+
+.name {
+  justify-self: left;
+  font-size: 1.25rem;
+  margin: 0;
+}
+
+.hp {
+  font-size: 1.25rem;
+  color: hsl(0 100% 50%);
+}
+
+header .energy {
+  display: inline-block;
+  transform: translateY(-0.15rem);
+}
+
+.frame:is(.picture, .species, .description) {
+  --lighter: hsl(var(--card-h) var(--card-s) calc(var(--card-l) + 15%));
+  --lightest: hsl(var(--card-h) var(--card-s) calc(var(--card-l) + 30%));
+  --darker: hsl(var(--card-h) var(--card-s) calc(var(--card-l) - 15%));
+  border-color: var(--darker) var(--card-color) var(--lighter);
+}
+
+.picture {
+  width: 100%;
+  aspect-ratio: 4 / 3;
+  object-fit: contain;
+  background-color: white;
+  border: 0.375rem solid;
+  box-shadow: 0.25rem 0.25rem 0.5rem black;
+  user-select: none;
+}
+
+.species {
+  width: 90%;
+  padding: 0.1rem;
+  margin: 0.25rem auto;
+  text-align: center;
+  font-size: 0.75rem;
+  font-style: italic;
+  font-weight: bold;
+  background-image: linear-gradient(90deg, var(--card-color), var(--lightest) 45% 55%, var(--card-color));
+  border-width: 0 0.2rem;
+  border-style: solid;
+  border-image: linear-gradient(var(--lightest), var(--darker)) 1 100%;
+}
+
+.grid-three {
+  display: grid;
+  grid-template-columns: repeat(3, 1fr);
+}
+
+.grid-three > *:first-child {
+  text-align: left;
+}
+
+.grid-three > span:last-child {
+  text-align: right;
+}
+
+.attacks {
+  list-style-type: none;
+  padding: 0;
+  margin: 0.1rem auto 0;
+}
+
+.attacks-row {
+  max-height: 5rem;
+  grid-template-columns: 3rem 10fr 1fr;
+  align-items: center;
+  gap: 1rem;
+  font-size: 0.95em;
+}
+
+.attack-text > span:only-child {
+  display: block;
+  margin-left: -2rem;
+  text-align: center;
+}
+
+.attack-cost {
+  display: flex;
+  justify-content: space-evenly;
+  flex-flow: row wrap;
+  text-align: justify;
+}
+
+.energy {
+  width: fit-content;
+  aspect-ratio: 1 / 1;
+  cursor: default;
+  user-select: none;
+}
+
+.energy:only-child {
+  justify-self: flex-start;
+  margin: auto;
+}
+
+.attack-name {
+  font-weight: bold;
+}
+
+.attack-damage {
+  min-width: 2.25rem;
+  font-size: 1.375rem;
+  text-align: center;
+}
+
+hr {
+  background-color: black;
+  border: 0.5px solid black;
+}
+
+.multipliers {
+  display: flex;
+  flex-direction: row;
+  justify-content: space-between;
+  height: 2rem;
+  margin-top: -0.5rem;
+  text-align: center;
+  font-size: 0.75rem;
+  font-weight: bold;
+}
+
+.multipliers > div {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  margin: 0;
+}
+
+.resistance {
+  position: relative;
+}
+
+.resistance-total {
+  position: absolute;
+  top: 1rem;
+  left: 2.5rem;
+}
+
+.description {
+  padding: 0.1rem 0.5rem;
+  margin: 0.25rem 0 0.5rem;
+  border: 0.15rem solid;
+  font-size: 0.65rem;
+  font-style: italic;
+  font-weight: bold;
+}
+
+.footer {
+  text-align: center;
+  font-weight: bold;
+  font-size: 0.6rem;
+  margin: auto 0;
+}
+
+a {
+  text-decoration: none;
+  color: black !important;
+}

templates/index.html

@@ -0,0 +1,22 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>This Pokémon Does Not Exist</title>
+    <link rel="shortcut icon" href="{{ url_for('static', filename='favicon.ico') }}" />
+    <link rel="stylesheet" id="stylesheet-tag" />
+    <script type="module" id="script-tag"></script>
+    <script>
+      const basePath = document.location.origin + document.location.pathname;
+      document.getElementById('script-tag').src = `${basePath}static/index.js`;
+      document.getElementById('stylesheet-tag').href = `${basePath}static/style.css`;
+    </script>
+  </head>
+  <body>
+    <h1>This Pokémon Does Not Exist</h1>
+    <div class="duration"><span class="seconds">0.0</span>/500s</div>
+    <button class="generate">Generate Pokémon Card with AI</button>
+    <section class="render"></section>
+  </body>
+</html>