Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import os
 import yt_dlp
 import psutil
+from youtube_search import VideosSearch
 
 # Function to get disk usage information
 def get_disk_info():
@@ -27,7 +28,7 @@ def update_disk_info():
 update_disk_info()
 
 # Streamlit input fields
-url = st.text_input("Enter the YouTube URL:")
+query = st.text_input("Enter search query or YouTube URL:")
 quality = st.selectbox("Select the desired quality:", ['4320', '2160', '1440', '1080', '720', '480'])
 save_file_name = st.text_input("Enter the save file name (without extension):")
 
@@ -40,6 +41,27 @@ ydl_opts = {
     'outtmpl': f'{save_file_name}.%(ext)s',  # Save the video with the specified name
 }
 
+# Search for videos if a query is entered
+if query:
+    if not query.startswith("http"):
+        # If the query is not a URL, search for videos
+        st.markdown("### Search Results:")
+        videos_search = VideosSearch(query, limit=5)
+        results = videos_search.result()['result']
+        video_urls = [f"https://www.youtube.com/watch?v={video['id']}" for video in results]
+
+        for i, video in enumerate(results):
+            st.write(f"{i + 1}. **{video['title']}** ({video['duration']})")
+            st.write(f"   Channel: {video['channel']['name']}")
+            st.write(f"   URL: {video_urls[i]}")
+            st.write("\n")
+        
+        selected_video_index = st.number_input("Select video number to download:", min_value=1, max_value=len(results), step=1) - 1
+        url = video_urls[selected_video_index]
+    else:
+        # If the query is a URL, use it directly
+        url = query
+
 if st.button("Download"):
     with st.spinner("Downloading... Please wait."):
         try: