Update app.py

app.py

@@ -2,17 +2,30 @@ import streamlit as st
 import os
 import yt_dlp
 import psutil
-import time
 
-# Function to get system information
-def get_system_info():
-    ram_info = psutil.virtual_memory()
+# Function to get disk usage information
+def get_disk_info():
     disk_info = psutil.disk_usage('/')
-    return ram_info, disk_info
+    return disk_info
 
 # Streamlit UI setup
 st.title("YouTube Video Downloader 8K supported")
 
+# Placeholder for disk usage info
+disk_info_placeholder = st.empty()
+
+# Function to update disk usage information
+def update_disk_info():
+    disk_info = get_disk_info()
+    disk_used_gb = disk_info.used / (1024 ** 3)
+    disk_total_gb = disk_info.total / (1024 ** 3)
+    
+    disk_info_placeholder.markdown("## Server Information")
+    disk_info_placeholder.markdown(f"**Disk Usage:** {disk_info.percent}% used ({disk_used_gb:.2f} GB of {disk_total_gb:.2f} GB)")
+
+# Initial disk information update
+update_disk_info()
+
 # Streamlit input fields
 url = st.text_input("Enter the YouTube URL:")
 quality = st.selectbox("Select the desired quality:", ['4320', '2160', '1440', '1080', '720', '480'])
@@ -27,28 +40,6 @@ ydl_opts = {
     'outtmpl': f'{save_file_name}.%(ext)s',  # Save the video with the specified name
 }
 
-# Placeholder for system info
-system_info_placeholder = st.empty()
-
-# Function to update system information
-def update_system_info():
-    ram_info, disk_info = get_system_info()
-    ram_used_gb = ram_info.used / (1024 ** 3)
-    ram_total_gb = ram_info.total / (1024 ** 3)
-    disk_used_gb = disk_info.used / (1024 ** 3)
-    disk_total_gb = disk_info.total / (1024 ** 3)
-    
-    system_info_placeholder.markdown("## Server Information")
-    system_info_placeholder.markdown(f"**RAM Usage:** {ram_info.percent}% used ({ram_used_gb:.2f} GB of {ram_total_gb:.2f} GB)")
-    system_info_placeholder.markdown(f"**Disk Usage:** {disk_info.percent}% used ({disk_used_gb:.2f} GB of {disk_total_gb:.2f} GB)")
-
-# Initial system information update
-update_system_info()
-
-# Button to update system information
-if st.button("Update System Information"):
-    update_system_info()
-
 if st.button("Download"):
     with st.spinner("Downloading... Please wait."):
         try:
@@ -81,8 +72,5 @@ if st.button("Download"):
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
 
-# Periodically update the system information
-while True:
-    update_system_info()
-    time.sleep(5)
-    st.experimental_rerun()
+# Update disk information whenever the user interacts with the app
+update_disk_info()