Upload 2 files

app.py

@@ -20,15 +20,6 @@ def new_streak(intensity):
     new_streak = utils.get_streak()
     return f"{new_streak} streak days", '## ' + '✅  ' * new_streak, reward
 
-# def update_pokemon_gallery():
-#     pokemon_exps = []
-
-#     pokemons = utils.get_pokemon()
-#     for dex in pokemons.keys():
-#         info = gr.Markdown(f"**EXP**: {utils.get_poke_energy(dex)}")
-#         pokemon_exps.append(info)
-#     return pokemon_exps
-
 def restart():
     with open('dummy.txt', 'w') as f:
         f.write(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
@@ -43,28 +34,25 @@ with gr.Blocks() as demo:
 
     gr.Markdown('<h1 style="text-align: center;">PokeVerse</h1>')
     reward_md = gr.Markdown()
-    winning_btn = gr.Button("Are you winning today? 💪")
-    restart_btn = gr.Button("Restart 🍀")
+    winning_btn = gr.Button("💪 Are you winning today? 💪")
+    restart_btn = gr.Button("🍀 Restart 🍀")
     intensity = gr.Slider(0, 3, step=1, label="Intensity")
-    streak_md = gr.Markdown(f"{streaks} streak days")
+    streak_md = gr.Markdown(f"### 🔥 {streaks} streak days")
     streakboard = gr.Markdown('## ' + '✅  ' * streaks)
-
+    money_md = gr.Markdown(f"### 💰 Money saved: {'{:,}'.format(utils.get_money())} VND")
+    money_md = gr.Markdown(f"### 🧱 Minifigure: {utils.get_total_minifigure()}")
     gr.Markdown('<h2 style="text-align: center;">My Poke Friends</h2>')
 
-    pokemon_exps = []
-
     with gr.Row():
         for dex in pokemons.keys():
-            with gr.Group():
-                with gr.Column(min_width=200) as poke_col:
-                    gr.Image(pokemons[dex]['image'], show_label=False)
-                    with gr.Accordion(pokemons[dex]['name'], open=False):
-                         gr.Markdown(pokemons[dex]['description'])
-                    info = gr.Markdown(f"**EXP**: {utils.get_poke_energy(dex)}")
-                    pokemon_exps.append(info)
-                
+            if pokemons[dex]['visible'] == True:
+                with gr.Group():
+                    with gr.Column(min_width=200) as poke_col:
+                        gr.Image(pokemons[dex]['image'], show_label=False)
+                        with gr.Accordion(pokemons[dex]['name'], open=False):
+                            gr.Markdown(pokemons[dex]['description'])
+                        info = gr.Markdown(f"**EXP**: {utils.get_poke_energy(dex)}")
 
     winning_btn.click(new_streak, inputs=[intensity], outputs=[streak_md, streakboard, reward_md])
     restart_btn.click(restart, None, None)
-    # streak_md.change(update_pokemon_gallery, outputs=pokemon_exps)
 demo.launch()

utils.py

@@ -84,7 +84,7 @@ def get_poke_energy(dex):
     metadata = read_metadata()
     return metadata["pokemon"][dex]["energy"]
 
-def pokemon(dex):
+def get_pokemon_info(dex):
     pokemon = pypokedex.get(dex=dex)
     description = pokemon.get_descriptions()
     try:
@@ -99,6 +99,62 @@ def pokemon(dex):
         sprite = None
     return name, description, image, sprite
 
+def get_evolution_chain(dex):
+    # Fetch the Pokémon species data for Ivysaur
+    pokemon = pypokedex.get(dex=dex)
+    name = pokemon.name
+    species_url = f"https://pokeapi.co/api/v2/pokemon-species/{name}"
+    species_response = requests.get(species_url)
+    species_data = json.loads(species_response.text)
+
+    # Extract the evolution chain URL
+    evolution_chain_url = species_data['evolution_chain']['url']
+
+    # Fetch the evolution chain data
+    evolution_chain_response = requests.get(evolution_chain_url)
+    evolution_chain_data = json.loads(evolution_chain_response.text)
+
+    # Parse the evolution chain data and get the full list of evolutions
+    evolution_list = []
+    chain = evolution_chain_data['chain']
+    while chain:
+        species_name = chain['species']['name']
+        evolution_list.append(pypokedex.get(name=species_name).dex)
+        if chain.get('evolves_to') != []:
+            chain = chain.get('evolves_to')[0]
+        else:
+            break
+
+    return evolution_list
+
+def can_evolve(dex, energy):
+    metadata = read_metadata()
+    if metadata["pokemon"][dex]["max_energy"] <= energy:
+        return True
+    else:
+        return False
+
+def evolve_to(dex):
+    evolution_chain = get_evolution_chain(dex)
+    print(evolution_chain)
+    # If fully evolved
+    if dex == evolution_chain[-1]:
+        return None
+    else:
+        evolve_dex = evolution_chain[evolution_chain.index(dex) + 1]
+        name, description, image, sprite = get_pokemon_info(evolve_dex)
+        return {
+            evolve_dex: {
+                "name": name,
+                "image": image,
+                "sprite": sprite,
+                "description": description,
+                "energy": 0,
+                "max_energy": 100 + 20*(evolution_chain.index(dex) + 1),
+                "visible": True
+            }
+        }
+
 def random_pokemon():
     min_dex = 1
     max_dex = 1017
@@ -106,8 +162,9 @@ def random_pokemon():
     all_dex = get_all_pokemon_dex()
     while True:
         dex = random.randint(min_dex, max_dex)
-        if dex not in all_dex:
-            name, description, image, sprite = pokemon(dex)
+        # if not in the team and the first level
+        if (dex not in all_dex) and (dex == get_evolution_chain(dex)[0]):
+            name, description, image, sprite = get_pokemon_info(dex)
             break
 
     random_pokemon = {
@@ -116,7 +173,9 @@ def random_pokemon():
             "image": image,
             "sprite": sprite,
             "description": description,
-            "energy": 0
+            "energy": 0,
+            "max_energy": 100,
+            "visible": True
         }
     }
 
@@ -124,8 +183,7 @@ def random_pokemon():
 
 def add_new_pokemon(new_pokemon):
     metadata = read_metadata()
-    dex = list(new_pokemon.keys())[0]
-    metadata["pokemon"][dex] = new_pokemon[dex]
+    metadata["pokemon"].update(new_pokemon)
     return metadata
 
 def is_lucky(streak):
@@ -185,6 +243,44 @@ def new_streak(intensity):
     streak = metadata["streak"]
     reward = ""
 
+    date1 = datetime.datetime.today()
+    date2 = datetime.datetime.strptime(metadata["last_streak_day"], "%d/%m/%Y")
+    delta = date1 - date2
+    distance = delta.days
+
+    if (distance >= 2):
+        metadata = {
+            "streak": 0,
+            "total_minifigure": 0,
+            "donation": [
+                {
+                    "wildact": {
+                        "donation_link": "https://www.wildact-vn.org/en/support-us/donate/",
+                        "image": "https://a-us.storyblok.com/f/1014782/567x567/7338612d04/wa_logo-2023-vn.png"
+                    }
+                }
+            ],
+            "last_streak_day": datetime.datetime.today().strftime("%d/%m/%Y"),
+            "money": 0,
+            "pokemon": {
+                "4": {
+                    "name": "Charmander",
+                    "image": "https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/other/official-artwork/4.png",
+                    "sprite": "https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/4.png",
+                    "description": "Obviously prefers hot places. When it rains, steam is said to spout from the tip of its tail.",
+                    "energy": 179,
+                    "max_energy": 100,
+                    "visible": false
+                }
+            },
+            "wishlist": [
+                "https://u6wdnj9wggobj.vcdn.cloud/media/catalog/product/cache/a237138a07ed0dd2cc8a6fa440635ea6/7/6/76414_copy_0.jpg",
+                "https://u6wdnj9wggobj.vcdn.cloud/media/catalog/product/cache/a237138a07ed0dd2cc8a6fa440635ea6/6/0/60302_7_.jpg"
+            ]
+        }
+        save_metadata(metadata)
+        return "# 🔥🔥🔥 YOU SKIP 1 DAYYYYYY. EVERYTHING IS BURNTTTTTT. ALL YOUR FRIENDS ARE VANISHINGGGGG 🔥🔥🔥"
+
     if is_lucky(streak):
         gift = lucky_draw()
         if gift[0] == 'money':
@@ -199,19 +295,32 @@ def new_streak(intensity):
             gift_str = "a minifigure"
 
         if gift[0] == 'energy':
-            gift_str = f"{gift[2]} energy point to {get_pokemon_by_dex(gift[1])['name']}"
+            gift_str = f"{gift[2]} energy point to **{get_pokemon_by_dex(gift[1])['name']}**"
 
-        reward += f"**Lucky*!!!* You are rewarded {gift_str}.\n\n"
+        reward += f"**Lucky!!!** You are awarded {gift_str} 🎁.\n\n \n\n"
 
         # Update new reward
         metadata = read_metadata()
     
     all_dex = get_all_pokemon_dex()
-    random_dex = random.choice(all_dex)
-    energy = poke_energy(streak, intensity)
-    metadata["pokemon"][random_dex]["energy"] += energy
-    reward += f"Your **{metadata['pokemon'][random_dex]['name']}** has just got **{energy} EXP**.\n\n"
+    while True:
+        random_dex = random.choice(all_dex)
+        if metadata["pokemon"][random_dex]["visible"] == True:
+            energy = poke_energy(streak, intensity)
+            metadata["pokemon"][random_dex]["energy"] += energy
+
+            if can_evolve(random_dex, metadata["pokemon"][random_dex]["energy"]):
+                metadata["pokemon"][random_dex]["visible"] = False
+                evolve_pokemon = evolve_to(int(random_dex))
+                evolve_dex = list(evolve_pokemon.keys())[0]
+                metadata["pokemon"].update(evolve_pokemon)
+                reward += f"Your **{metadata['pokemon'][random_dex]['name']}** has just evolved into **{metadata['pokemon'][evolve_dex]['name']}**. Hoorayyyyy!!! 🥳.\n\n \n\n"
+
+            reward += f"Your **{metadata['pokemon'][random_dex]['name']}** has just got **{energy} EXP** ☀️.\n\n \n\n"
 
+            break
+    
+    metadata["last_streak_day"] = datetime.datetime.today().strftime("%d/%m/%Y")
     save_metadata(metadata)
 
     is_first_day_of_week = datetime.date.today().weekday() == 0