TaiYouWeb commited on
Commit
b051760
1 Parent(s): 03840d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -14,6 +14,7 @@ from playwright.sync_api import sync_playwright
14
  from languages import get_language_names
15
  from subtitle import text_output, subtitle_output
16
 
 
17
  import psutil
18
  import subprocess
19
  from gpustat import GPUStatCollection
@@ -68,7 +69,7 @@ def update_gpu_status():
68
  memory_used = gpu.memory_used
69
  memory_total = gpu.memory_total
70
  memory_utilization = (memory_used / memory_total) * 100
71
- gpu_status=(f"**GPU** {gpu_id}: {gpu_name}\nUtilization: {gpu_utilization}%\n**Memory Used**: {memory_used}MB\n**Memory Total**: {memory_total}MB\n**Memory Utilization**: {memory_utilization:.2f}%\n")
72
  return gpu_status
73
 
74
  except Exception as e:
@@ -83,18 +84,21 @@ def torch_update_gpu_status():
83
  free_memory=gpu_memory[0] /(1024 *1024)
84
  used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024)
85
 
86
- gpu_status = f"**GPU**: {gpu_info}\n**Free Memory**: {free_memory}MB\n**Total Memory**: {total_memory:.2f} MB\n**Used Memory**: {used_memory:.2f} MB\n"
87
  else:
88
  gpu_status = "No GPU available"
89
  return gpu_status
90
 
91
  def update_cpu_status():
92
- import datetime
93
  current_time = datetime.datetime.utcnow()
94
  time_str = current_time.strftime("%Y-%m-%d %H:%M:%S")
95
 
96
  cpu_percent = psutil.cpu_percent()
97
- cpu_status = f"**{time_str} (UTC+0)**\n**CPU Usage**: {cpu_percent}%\n"
 
 
 
 
98
  return cpu_status
99
 
100
  def update_status():
@@ -446,6 +450,7 @@ with demo:
446
  refresh_button = gr.Button("Refresh System Status")
447
  refresh_button.click(refresh_status, None, sys_status_output)
448
  sys_status_output.value = refresh_status()
 
449
 
450
  if __name__ == "__main__":
451
  demo.queue().launch(ssr_mode=False)
 
14
  from languages import get_language_names
15
  from subtitle import text_output, subtitle_output
16
 
17
+ import datetime
18
  import psutil
19
  import subprocess
20
  from gpustat import GPUStatCollection
 
69
  memory_used = gpu.memory_used
70
  memory_total = gpu.memory_total
71
  memory_utilization = (memory_used / memory_total) * 100
72
+ gpu_status=(f"**GPU Name** {gpu_id}: {gpu_name}\nUtilization: {gpu_utilization}%\n**Memory Used**: {memory_used}MB\n**Memory Total**: {memory_total}MB\n**Memory Utilization**: {memory_utilization:.2f}%\n")
73
  return gpu_status
74
 
75
  except Exception as e:
 
84
  free_memory=gpu_memory[0] /(1024 *1024)
85
  used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024)
86
 
87
+ gpu_status = f"**GPU Name**: {gpu_info}\n**Free Memory**: {free_memory}MB\n**Total Memory**: {total_memory:.2f} MB\n**Used Memory**: {used_memory:.2f} MB\n"
88
  else:
89
  gpu_status = "No GPU available"
90
  return gpu_status
91
 
92
  def update_cpu_status():
 
93
  current_time = datetime.datetime.utcnow()
94
  time_str = current_time.strftime("%Y-%m-%d %H:%M:%S")
95
 
96
  cpu_percent = psutil.cpu_percent()
97
+
98
+ cpu_info = psutil.cpu_info()
99
+ cpu_name = cpu_info[0].brand_raw if cpu_info else "Unknown CPU"
100
+
101
+ cpu_status = f"**{time_str} (UTC+0)**\n\n**CPU Name**: {cpu_name}\n**CPU Usage**: {cpu_percent}%\n\n"
102
  return cpu_status
103
 
104
  def update_status():
 
450
  refresh_button = gr.Button("Refresh System Status")
451
  refresh_button.click(refresh_status, None, sys_status_output)
452
  sys_status_output.value = refresh_status()
453
+ demo.load(update_status, None, sys_status_output, every=3)
454
 
455
  if __name__ == "__main__":
456
  demo.queue().launch(ssr_mode=False)