Add Bookmarks

Intro.md

@@ -0,0 +1,3 @@
+# Yuyu-Tei Price Crawler
+
+[Yuyu-Tei Website](https://yuyu-tei.jp/game_ws/sell/sell_price.php)

app.py

@@ -1,4 +1,5 @@
 import datetime
+import json
 
 import gradio as gr
 import pandas as pd
@@ -9,23 +10,60 @@ from bs4.element import Tag
 
 class App:
     def __init__(self) -> None:
-        with gr.Blocks() as self.demo:
-            gr.Markdown("# Yuyu Tei Price Crawler")
+        with open("bookmarks.json", "rt", encoding="UTF-8") as fp:
+            self.bookmark_info = json.load(fp)
+
+        with open("Intro.md", "rt", encoding="UTF-8") as fp:
+            intro = fp.read()
+
+        theme = gr.themes.Soft()
+        with gr.Blocks(title="Yuyu Tei Crawler", theme=theme) as self.demo:
+            gr.Markdown(intro)
+            with gr.Row(equal_height=False):
+                self.__CreateColumns__()
+            self.__RegisterEvents__()
+
+    def __CreateColumns__(self):
+        with gr.Column():
+            ph = "e.g. https://yuyu-tei.jp/game_ws/sell/sell_price.php?ver=uma&menu=newest"
+            self.input_url = gr.Textbox(label="URL", placeholder=ph)
+            self.submit_btn = gr.Button("Submit")
+            self.__CreateBookmarks__()
+
+        with gr.Column():
+            self.output_file = gr.File(label="Result", file_count="single")
+            self.status = gr.Textbox("Ready", label="Status", interactive=False)
+
+    def __CreateBookmarks__(self):
+        with gr.Tab("Bookmarks"):
             with gr.Row():
-                with gr.Column():
-                    ph = "e.g. https://yuyu-tei.jp/game_ws/sell/sell_price.php?ver=uma&menu=newest"
-                    self.input_url = gr.Textbox(label="URL", placeholder=ph)
-                    self.submit_btn = gr.Button("Submit")
-                with gr.Column():
-                    self.output_file = gr.File(label="Excel", file_count="single")
-                    self.status = gr.Markdown("Ready")
-
-            self.submit_btn.click(
-                self.Download,
-                self.input_url,
-                [self.output_file, self.status],
+                self.bookmarks = {name: gr.Button(value=name) for name in self.bookmark_info}
+
+    def __RegisterEvents__(self):
+        args_submit = KwargsToDict(
+            fn=self.Download,
+            inputs=self.input_url,
+            outputs=[self.output_file, self.status],
+        )
+
+        self.submit_btn.click(**args_submit)
+        self.input_url.submit(**args_submit)
+
+        def GetArgsBookmark(name):
+            return KwargsToDict(
+                fn=self.ClickBookmark,
+                inputs=self.bookmarks[name],
+                outputs=self.input_url,
+                show_progress=False,
             )
 
+        for name in self.bookmarks:
+            args_bookmark = GetArgsBookmark(name)
+            self.bookmarks[name].click(**args_bookmark)
+
+    def ClickBookmark(self, name):
+        return self.bookmark_info[name]
+
     def Download(self, url):
         try:
             ts = datetime.datetime.utcnow() + datetime.timedelta(hours=8)
@@ -36,7 +74,7 @@ class App:
             return None, f"Error: {e}"
 
     def Launch(self):
-        self.demo.launch()
+        self.demo.launch(favicon_path="icon.png")
 
 
 def CrawlPage(url, output_path):
@@ -95,5 +133,9 @@ def GetCardNameFromPage(url):
     return card_name.text.strip()
 
 
+def KwargsToDict(**kwargs):
+    return kwargs
+
+
 if __name__ == "__main__":
     App().Launch()

bookmarks.json

@@ -0,0 +1,4 @@
+{
+    "UMA": "https://yuyu-tei.jp/game_ws/sell/sell_price.php?ver=uma&menu=newest",
+    "LRC": "https://yuyu-tei.jp/game_ws/sell/sell_price.php?ver=lrc&menu=newest"
+}