Update app.py

app.py

@@ -161,7 +161,7 @@ body {
 
 .server-status {
     margin-top: 20px;
-    background-color: #f0f0f0;
+    background-color: white;
     padding: 10px;
     border-radius: 5px;
     box-shadow: 0 1px 3px rgba(0,0,0,0.1);
@@ -235,11 +235,10 @@ with gr.Blocks(css=css) as demo:
                 with gr.Row(elem_classes="server-status"):
                     gr.Markdown("#### Server Status")
                     with gr.Row():
-                        cpu_status = gr.Textbox(label="CPU", scale=1, container=False)
-                        memory_status = gr.Textbox(label="Memory", scale=1, container=False)
-                        disk_status = gr.Textbox(label="Disk", scale=1, container=False)
-                    with gr.Accordion("GPU Details", open=False):
-                        gpu_status = gr.JSON(container=False)
+                        cpu_status = gr.Textbox(label="CPU", scale=1)
+                        memory_status = gr.Textbox(label="Memory", scale=1)
+                        disk_status = gr.Textbox(label="Disk", scale=1)
+                    gpu_status = gr.Textbox(label="GPU Memory", scale=1)
                     refresh_button = gr.Button("Refresh", scale=1, size="sm")
 
         with gr.Column():
@@ -282,13 +281,35 @@ with gr.Blocks(css=css) as demo:
     def enhance_prompt_func(prompt):
         return convert_prompt(prompt, retry_times=1)
 
+    def get_server_status():
+        cpu_percent = psutil.cpu_percent()
+        memory = psutil.virtual_memory()
+        disk = psutil.disk_usage('/')
+        try:
+            gpus = GPUtil.getGPUs()
+            if gpus:
+                gpu = gpus[0]  # 只获取第一个GPU的信息
+                gpu_memory = f"{gpu.memoryUsed}/{gpu.memoryTotal}MB ({gpu.memoryUtil*100:.1f}%)"
+            else:
+                gpu_memory = "No GPU found"
+        except:
+            gpu_memory = "GPU information unavailable"
+        
+        return {
+            'cpu': f"{cpu_percent}%",
+            'memory': f"{memory.percent}%",
+            'disk': f"{disk.percent}%",
+            'gpu_memory': gpu_memory
+        }
+    
+
     def update_server_status():
         status = get_server_status()
         return (
-            f"{status['cpu']}",
-            f"{status['memory']}",
-            f"{status['disk']}",
-            status['gpu']
+            status['cpu'],
+            status['memory'],
+            status['disk'],
+            status['gpu_memory']
         )
 
 
@@ -308,7 +329,7 @@ with gr.Blocks(css=css) as demo:
 
     
     refresh_button.click(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status])
-    demo.load(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], every=60)
+    demo.load(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], every=1)
 
 if __name__ == "__main__":
     demo.queue(max_size=10, default_concurrency_limit=1)