Upload main.py

main.py

@@ -204,18 +204,26 @@ async def fetch_trending_videos(reddit, processed_videos, subreddit_name):
 
 
 async def download_video(url, filename):
-    """Download video from Reddit using RedDownloader."""
-    try:
-        # RedDownloader saves the video to a file whose name is based on the given output prefix.
-        RedDownloader.Download(url, quality=720, output=filename.split(".")[0])
-        if os.path.exists(filename):
-            logging.info(f"Video downloaded to {filename}")
-            return filename
-        else:
-            logging.error(f"Video file not found after download: {filename}")
-            return None
-    except Exception as e:
-        logging.error(f"Error downloading video: {e}")
+    """
+    Offload the synchronous RedDownloader.Download call to a thread.
+    """
+    loop = asyncio.get_running_loop()
+
+    def blocking_download():
+        try:
+            # This call is blocking.
+            RedDownloader.Download(url, quality=720, output=filename.split('.')[0])
+            return os.path.exists(filename)
+        except Exception as e:
+            logging.error(f"Error downloading video: {e}")
+            return False
+
+    exists = await loop.run_in_executor(None, blocking_download)
+    if exists:
+        logging.info(f"Video downloaded to {filename}")
+        return filename
+    else:
+        logging.error(f"Video file not found after download: {filename}")
         return None
 
 
@@ -500,39 +508,32 @@ async def process_video_task(video_info: dict):
     try:
         api_key = os.environ.get("GOOGLE_API_KEY")
         if not api_key:
-            processing_results[video_id] = {
-                "status": "error",
-                "error": "Missing GOOGLE_API_KEY",
-            }
+            processing_results[video_id] = {"status": "error", "error": "Missing GOOGLE_API_KEY"}
             return
 
         reddit = await initialize_reddit()
         submission = await reddit.submission(id=video_id)
         comments = await get_n_comments(submission, 5)
         comments_string = "\n\n".join(comments) if comments else "No comments found."
+
         temp_video_filename = f"{video_id}.mp4"
-        downloaded_video = await download_video(
-            video_info.get("url"), temp_video_filename
-        )
+        downloaded_video = await download_video(video_info.get("url"), temp_video_filename)
         if not downloaded_video:
-            processing_results[video_id] = {
-                "status": "error",
-                "error": "Failed to download video",
-            }
+            processing_results[video_id] = {"status": "error", "error": "Failed to download video"}
             await reddit.close()
             return
 
-        # Use the process pool to offload heavy MoviePy work
+        # Offload heavy MoviePy processing using the process pool.
         loop = asyncio.get_running_loop()
         result = await loop.run_in_executor(
             process_executor,
-            generate_video,  # blocking function
+            generate_video,  # blocking function using MoviePy
             downloaded_video,
             video_info.get("title"),
             comments_string,
             api_key,
             f"{video_id}",
-            "@damnsointeresting",
+            "@damnsointeresting"
         )
 
         if result and result[0]:
@@ -542,13 +543,10 @@ async def process_video_task(video_info: dict):
                 "status": "completed",
                 "video_url": video_url,
                 "generated_title": generated_title,
-                "generated_caption": generated_caption,
+                "generated_caption": generated_caption
             }
         else:
-            processing_results[video_id] = {
-                "status": "error",
-                "error": "Video processing failed",
-            }
+            processing_results[video_id] = {"status": "error", "error": "Video processing failed"}
 
         if os.path.exists(temp_video_filename):
             os.remove(temp_video_filename)