Update app.py

app.py

@@ -43,14 +43,15 @@ def update_data():
     data = get_available_free()
     df = pd.DataFrame(data)
     
-    def custom_sort_status(status):
-        if status == "⌀":
-            return 1
-        elif status == "✓":
-            return 0
-        else:
-            return 2
-    df = df.sort_values(by=['API', 'Text Completion', 'Chat Completion'], key=lambda x: (x.map(custom_sort_status), x))
+    status_mapping = {"✓": 0, "⌀": 1, "---": 2}
+
+    df['Text Completion'] = df['Text Completion'].map(status_mapping)
+    df['Chat Completion'] = df['Chat Completion'].map(status_mapping)
+    
+    df = df.sort_values(by=['API', 'Text Completion', 'Chat Completion'])
+    
+    df['Text Completion'] = df['Text Completion'].map({v: k for k, v in status_mapping.items()})
+    df['Chat Completion'] = df['Chat Completion'].map({v: k for k, v in status_mapping.items()})
     
     return df