Upload main.py

main.py

@@ -155,13 +155,40 @@ UPLOAD_DIRECTORY = "./uploads"
 
 class YouTubeBase(BaseModel):
     link: str
+    only_audio: bool = False
+
+class YouTubeSearch(BaseModel):
+    query: str
+    limit: int = 7
 
 def secs_to_mins(secs: int) -> str:
     mins, secs = divmod(secs, 60)
     return f"{mins}:{secs}"
 
-@app.post("/akeno/youtube-video", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-async def youtube_video(payload: YouTubeBase):
+@app.get("/akeno/youtube-search", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def youtube_search(payload: YouTubeSearch):
+    try:
+        results = YoutubeDriver(payload.query, payload.limit).to_dict()
+    except Exception as e:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": str(e)}
+        )
+    if not results:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "No results found."}
+        )
+    text = f"**🔎 𝖳𝗈𝗍𝖺𝗅 𝖱𝖾𝗌𝗎𝗅𝗍𝗌 𝖥𝗈𝗎𝗇𝖽:** `{len(results)}`\n\n"
+    for result in results:
+        text += f"**𝖳𝗂𝗍𝗅𝖾:** `{result['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{result['channel']}`\n**𝖵𝗂𝖾𝗐𝗌:** `{result['views']}`\n**𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{result['duration']}`\n**𝖫𝗂𝗇𝗄:** `https://youtube.com{result['url_suffix']}`\n\n"
+    return SuccessResponse(
+        status="True",
+        randydev={"results": text}
+    )
+
+@app.post("/akeno/youtube", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def youtube_api(payload: YouTubeBase):
     status, url = YoutubeDriver.check_url(payload.link)
     if not status:
         return SuccessResponse(
@@ -169,23 +196,45 @@ async def youtube_video(payload: YouTubeBase):
             randydev={"error": url}
         )
     try:
-        with YoutubeDL(YoutubeDriver.video_options()) as ytdl:
-            yt_data = ytdl.extract_info(url, download=True)
-            yt_file = yt_data["id"]
-        time_second = time.time()
-        with open(f"{yt_file}.mp4", "rb") as video_file:
-            encoded_string = base64.b64encode(video_file.read())
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "video_data": encoded_string,
-                "title": yt_data['title'],
-                "views": yt_data['view_count'],
-                "duration": secs_to_mins(int(yt_data['duration'])),
-                "thumbnail": f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg",
-                "time_second": time_second
-            }
-        )
+        if payload.only_audio:
+            with YoutubeDL(YoutubeDriver.song_options()) as ytdl:
+                yt_data = ytdl.extract_info(url, download=True)
+                yt_file = ytdl.prepare_filename(yt_data)
+                ytdl.process_info(yt_data)
+            time_second = time.time()
+            with open(f"{yt_file}.mp3", "rb") as audio_file:
+                encoded_string = base64.b64encode(audio_file.read())
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "video_data": encoded_string,
+                    "channel": yt_data['channel'],
+                    "title": yt_data['title'],
+                    "views": yt_data['view_count'],
+                    "duration": secs_to_mins(int(yt_data['duration'])),
+                    "thumbnail": f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg",
+                    "time_second": time_second
+                }
+            )
+        else:
+            with YoutubeDL(YoutubeDriver.video_options()) as ytdl:
+                yt_data = ytdl.extract_info(url, download=True)
+                yt_file = yt_data["id"]
+            time_second = time.time()
+            with open(f"{yt_file}.mp4", "rb") as video_file:
+                encoded_string = base64.b64encode(video_file.read())
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "video_data": encoded_string,
+                    "channel": yt_data['channel'],
+                    "title": yt_data['title'],
+                    "views": yt_data['view_count'],
+                    "duration": secs_to_mins(int(yt_data['duration'])),
+                    "thumbnail": f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg",
+                    "time_second": time_second
+                }
+            )
     except Exception as e:
         return SuccessResponse(
             status="False",