Update app.py

app.py

@@ -1,10 +1,23 @@
 import streamlit as st
 import os
 import yt_dlp
+import psutil
+
+# Function to get system information
+def get_system_info():
+    ram_info = psutil.virtual_memory()
+    disk_info = psutil.disk_usage('/')
+    return ram_info, disk_info
 
 # Streamlit UI setup
 st.title("YouTube Video Downloader 8K supported")
 
+# Display server RAM and disk usage
+ram_info, disk_info = get_system_info()
+st.write("## Server Information")
+st.write(f"**RAM Usage:** {ram_info.percent}% used ({ram_info.used // (1024 ** 2)} MB of {ram_info.total // (1024 ** 2)} MB)")
+st.write(f"**Disk Usage:** {disk_info.percent}% used ({disk_info.used // (1024 ** 3)} GB of {disk_info.total // (1024 ** 3)} GB)")
+
 # Streamlit input fields
 url = st.text_input("Enter the YouTube URL:")
 quality = st.selectbox("Select the desired quality:", ['4320', '2160', '1440', '1080', '720', '480'])
@@ -20,32 +33,33 @@ ydl_opts = {
 }
 
 if st.button("Download"):
-    try:
-        # Download the video
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            result = ydl.download([url])
-        
-        # Check if download was successful
-        if result == 0:
-            # Get the actual file extension used
-            info_dict = ydl.extract_info(url, download=False)
-            extension = info_dict.get('ext', 'webm')
-            file_path = f'{save_file_name}.{extension}'
-
-            # Check if the file exists
-            if os.path.exists(file_path):
-                # Provide a download link for the file
-                with open(file_path, "rb") as file:
-                    st.download_button(
-                        label="Download video",
-                        data=file,
-                        file_name=f"{save_file_name}.{extension}",
-                        mime=f"video/{extension}"
-                    )
-                st.success('Your video is ready for download.')
+    with st.spinner("Downloading... Please wait."):
+        try:
+            # Download the video
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                result = ydl.download([url])
+            
+            # Check if download was successful
+            if result == 0:
+                # Get the actual file extension used
+                info_dict = ydl.extract_info(url, download=False)
+                extension = info_dict.get('ext', 'webm')
+                file_path = f'{save_file_name}.{extension}'
+
+                # Check if the file exists
+                if os.path.exists(file_path):
+                    # Provide a download link for the file
+                    with open(file_path, "rb") as file:
+                        st.download_button(
+                            label="Download video",
+                            data=file,
+                            file_name=f"{save_file_name}.{extension}",
+                            mime=f"video/{extension}"
+                        )
+                    st.success('Your video is ready for download.')
+                else:
+                    st.error("File not found. Please make sure the download was successful.")
             else:
-                st.error("File not found. Please make sure the download was successful.")
-        else:
-            st.error("Download failed. Please check the URL and try again.")
-    except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+                st.error("Download failed. Please check the URL and try again.")
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")