Update ui.py

ui.py

@@ -13,10 +13,14 @@ def get_system_metrics():
 
 # Function to get internet speed
 def get_internet_speed():
-    st = speedtest.Speedtest()
-    download_speed = st.download() / 1_000_000  # Convert to Mbps
-    upload_speed = st.upload() / 1_000_000  # Convert to Mbps
-    return download_speed, upload_speed
+    try:
+        st = speedtest.Speedtest()
+        st.get_best_server()
+        download_speed = st.download() / 1_000_000  # Convert to Mbps
+        upload_speed = st.upload() / 1_000_000  # Convert to Mbps
+        return download_speed, upload_speed
+    except:
+        return None, None
 
 # Function to create a gauge chart
 def create_gauge_chart(value, title, min_value=0, max_value=100):
@@ -34,9 +38,11 @@ def create_gauge_chart(value, title, min_value=0, max_value=100):
 # Streamlit layout
 st.title("System Monitoring Dashboard")
 
-# Create columns for each gauge
-col1, col2 = st.columns(2)
-col3, col4 = st.columns(2)
+# Create placeholders for each gauge
+cpu_placeholder = st.empty()
+memory_placeholder = st.empty()
+download_placeholder = st.empty()
+upload_placeholder = st.empty()
 
 # Update the metrics in real-time
 while True:
@@ -44,14 +50,20 @@ while True:
     download_speed, upload_speed = get_internet_speed()
 
     # Create gauge charts
-    with col1:
+    with cpu_placeholder:
         st.plotly_chart(create_gauge_chart(cpu_usage, "CPU Usage"), use_container_width=True)
-    with col2:
+    with memory_placeholder:
         st.plotly_chart(create_gauge_chart(memory_usage, "Memory Usage"), use_container_width=True)
-    with col3:
-        st.plotly_chart(create_gauge_chart(download_speed, "Download Speed (Mbps)", 0, 1000), use_container_width=True)
-    with col4:
-        st.plotly_chart(create_gauge_chart(upload_speed, "Upload Speed (Mbps)", 0, 1000), use_container_width=True)
+    with download_placeholder:
+        if download_speed is not None:
+            st.plotly_chart(create_gauge_chart(download_speed, "Download Speed (Mbps)", 0, 1000), use_container_width=True)
+        else:
+            st.write("Error retrieving download speed")
+    with upload_placeholder:
+        if upload_speed is not None:
+            st.plotly_chart(create_gauge_chart(upload_speed, "Upload Speed (Mbps)", 0, 1000), use_container_width=True)
+        else:
+            st.write("Error retrieving upload speed")
 
-    # Refresh the metrics every 5 seconds
-    time.sleep(5)
+    # Refresh the metrics every 10 seconds
+    time.sleep(10)