Update

app.py

@@ -7,35 +7,8 @@ import pandas as pd
 
 from demo_list import DemoList
 
-TITLE = "# List of hysts' Spaces"
-
 demo_list = DemoList()
 
-COLUMN_NAMES = [
-    'title',
-    'arxiv',
-    'likes',
-    'tags',
-    'last_modified',
-    'created',
-    'sdk',
-    'sdk_version',
-    'status',
-    'hardware',
-]
-COLUMN_DATATYPES = [
-    'markdown',
-    'markdown',
-    'markdown',
-    'number',
-    'str',
-    'str',
-    'str',
-    'str',
-    'str',
-    'str',
-    'str',
-]
 DEFAULT_COLUMNS = [
     'title',
     'arxiv',
@@ -52,9 +25,8 @@ def update_df() -> pd.DataFrame:
 
 
 with gr.Blocks(css='style.css') as demo:
-    gr.Markdown(TITLE)
     df = gr.Dataframe(value=demo_list.df,
-                      datatype=COLUMN_DATATYPES,
+                      datatype=demo_list.column_datatype,
                       type='pandas')
     refresh_button = gr.Button('Refresh')
     refresh_button.click(fn=update_df, outputs=df)

demo_list.py

@@ -1,5 +1,6 @@
 import copy
 import datetime
+import operator
 import os
 import pathlib
 
@@ -14,6 +15,21 @@ HF_TOKEN = os.getenv('HF_TOKEN')
 
 
 class DemoList:
+    COLUMN_INFO = [
+        ['status', 'markdown'],
+        ['title', 'markdown'],
+        ['arxiv', 'markdown'],
+        ['github', 'markdown'],
+        ['likes', 'number'],
+        ['tags', 'str'],
+        ['last_modified', 'str'],
+        ['created', 'str'],
+        ['sdk', 'markdown'],
+        ['sdk_version', 'str'],
+        ['hardware', 'markdown'],
+        ['suggested_hardware', 'str'],
+    ]
+
     def __init__(self):
         self.api = HfApi(token=HF_TOKEN)
         self.data = self.load_data()
@@ -32,7 +48,7 @@ class DemoList:
             card = space_info.cardData
             info = data[url]
             info['title'] = card['title']
-            info['sdk'] = card['sdk']
+            info['sdk'] = self.colorize_sdk(card['sdk'])
             info['sdk_version'] = card.get('sdk_version', '')
             info['likes'] = space_info.likes
             last_modified = datetime.datetime.strptime(
@@ -41,8 +57,11 @@ class DemoList:
             created = datetime.datetime.strptime(info['created'],
                                                  '%Y-%m-%d-%H-%M-%S')
             info['created'] = created.strftime('%Y/%m/%d %H:%M:%S')
-            info['status'] = space_info.runtime['stage']
-            info['hardware'] = space_info.runtime['hardware']['current']
+            info['status'] = self.colorize_status(space_info.runtime['stage'])
+            info['suggested_hardware'] = card.get('suggested_hardware', '')
+            info['hardware'] = self.colorize_hardware(
+                space_info.runtime['hardware']['current'],
+                info['suggested_hardware'])
         return data
 
     @staticmethod
@@ -67,6 +86,46 @@ class DemoList:
         tags = sorted(self.data[url].get('tags', []))
         return ', '.join(tags)
 
+    @property
+    def column_names(self):
+        return list(map(operator.itemgetter(0), self.COLUMN_INFO))
+
+    @property
+    def column_datatype(self):
+        return list(map(operator.itemgetter(1), self.COLUMN_INFO))
+
+    def colorize_status(self, status: str) -> str:
+        if status == 'RUNNING':
+            color = 'green'
+        elif status in ['STOPPED', 'PAUSED']:
+            color = 'yellow'
+        elif status in ['RUNTIME_ERROR', 'BUILD_ERROR']:
+            color = 'red'
+        else:
+            color = ''
+        return f'<div class="{color}">{status}</div>'
+
+    def colorize_sdk(self, sdk: str) -> str:
+        if sdk == 'gradio':
+            color = 'orange'
+        elif sdk == 'docker':
+            color = 'cyan'
+        else:
+            color = ''
+        return f'<div class="{color}">{sdk}</div>'
+
+    def colorize_hardware(self, hardware: str | None,
+                          suggested_hardware: str) -> str:
+        if hardware is None:
+            return ''
+        if 't4' in hardware or 'a10g' in hardware or 'a100' in hardware:
+            color = 'red'
+        elif suggested_hardware and 'cpu' not in suggested_hardware:
+            color = 'blue'
+        else:
+            return hardware
+        return f'<div class="{color}">{hardware}</div>'
+
     def to_df(self) -> pd.DataFrame:
         data = copy.deepcopy(self.data)
         for url in list(data):
@@ -75,17 +134,5 @@ class DemoList:
             info['arxiv'] = self.get_arxiv_link(url)
             info['github'] = self.get_github_link(url)
             info['tags'] = self.get_tag_list(url)
-        df = pd.DataFrame(data).T.loc[:, [
-            'title',
-            'arxiv',
-            'github',
-            'likes',
-            'tags',
-            'last_modified',
-            'created',
-            'sdk',
-            'sdk_version',
-            'status',
-            'hardware',
-        ]]
+        df = pd.DataFrame(data).T.loc[:, self.column_names]
         return df

style.css

@@ -17,3 +17,27 @@ body a:hover {
 body a:hover {
   text-decoration: underline;
 }
+
+div.green {
+  color: green;
+}
+
+div.yellow {
+  color: yellow;
+}
+
+div.red {
+  color: red;
+}
+
+div.blue {
+  color: blue;
+}
+
+div.orange {
+  color: orange;
+}
+
+div.cyan {
+  color: cyan;
+}