feat: add get players badges and partners function when user login

app.py

@@ -3,7 +3,7 @@ import json
 import gradio as gr
 
 from theme import Seafoam
-from utils.utils import get_content
+from utils.utils import get_content, get_player_partners, get_player_badges
 
 seafoam = Seafoam()
 
@@ -16,6 +16,8 @@ def get_player_info(player_backend_user_id):
 
 # start of gradio interface
 with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
+    player_info = gr.State()
+
     with gr.Row():
         with gr.Column(
             scale=1,
@@ -88,5 +90,16 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
         )
         f = gr.HTML(html, label="Achievement Log", elem_id="achievement_log")
 
+    # handling player info
+    with gr.Row():
+        player_backend_id = gr.Textbox(
+            "http://id.junyiacademy.org/d6317d299321450191dd5da599a48b3d", visible=False
+        )
+        player_info_query_btn = gr.Button("Query")
+
+    player_info_query_btn.click(get_player_info, player_backend_id, player_info).then(
+        get_player_partners, player_info, pet_gallery
+    ).then(get_player_badges, player_info, badge_gallery)
+
 if __name__ == "__main__":
     demo.launch()

css/style.css

@@ -1,4 +1,4 @@
-.gradio-container {background: url('file=media/bg_clear.png')}
+.gradio-container {background: url('file=medias/bg_clear.png')}
 
 #user_avatar {
     background-color: transparent !important;

utils/utils.py

@@ -1,4 +1,22 @@
+from typing import List
+
+import gradio as gr
+
+MEDIA_PATH = "medias/"
+MEDIA_FILE_TYPE = ".png"
+
+
 def get_content(file_name: str) -> str:
     with open(file_name, "r", encoding="utf-8") as file:
         content = file.read()
     return content
+
+
+def get_player_partners(player_info: gr.State) -> List[str]:
+    return [
+        f"{MEDIA_PATH}{partner}{MEDIA_FILE_TYPE}" for partner in player_info["partners"]
+    ]
+
+
+def get_player_badges(player_info: gr.State) -> List[str]:
+    return [f"{MEDIA_PATH}{badge}{MEDIA_FILE_TYPE}" for badge in player_info["badges"]]