Upload main.py

main.py

@@ -30,6 +30,7 @@ import io
 import string
 import openai
 import uuid
+import time
 from io import BytesIO
 from datetime import datetime as dt
 from dotenv import load_dotenv
@@ -154,9 +155,13 @@ UPLOAD_DIRECTORY = "./uploads"
 class YouTubeBase(BaseModel):
     link: str
 
-@app.post("/akeno/youtube-video")
-async def youtube_video(link: str):
-    status, url = YoutubeDriver.check_url(link)
+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):
+    status, url = YoutubeDriver.check_url(payload.link)
     if not status:
         return SuccessResponse(
             status="False",
@@ -165,14 +170,21 @@ async def youtube_video(link: str):
     try:
         with YoutubeDL(YoutubeDriver.video_options()) as ytdl:
             yt_data = ytdl.extract_info(url, download=True)
-            yt_file = ytdl.prepare_filename(yt_data)
-        if not os.path.exists(yt_file):
-            return SuccessResponse(
-                status="False",
-                randydev={"error": "Video download failed"}
+            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
+                }
             )
-        file_stream = open(yt_file, "rb")
-        return StreamingResponse(file_stream, media_type="video/mp4")
     except Exception as e:
         return SuccessResponse(
             status="False",