update time

src/api/notes_routes.py

@@ -2,6 +2,7 @@ import uuid
 import re
 import logging
 import os
+import urllib.request
 from datetime import datetime
 from typing import List, Optional, Dict
 from pathlib import Path
@@ -26,6 +27,46 @@ router = APIRouter(tags=["Notes"])
 # مخزن المهام المؤقت في الذاكرة
 tasks: Dict[str, Dict] = {}
 
+
+# ==========================================
+# ⏱️ YouTube Duration Scraper (stdlib only)
+# ==========================================
+
+def get_youtube_duration(url: str) -> int:
+    """
+    Fetches the YouTube video page and extracts the video duration in seconds
+    by scraping the `lengthSeconds` value from the page HTML.
+
+    Uses only Python standard library (urllib.request + re).
+    Returns the duration as an integer, or 0 if extraction fails for any reason.
+    """
+    try:
+        req = urllib.request.Request(
+            url,
+            headers={
+                "User-Agent": (
+                    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+                    "AppleWebKit/537.36 (KHTML, like Gecko) "
+                    "Chrome/124.0.0.0 Safari/537.36"
+                )
+            },
+        )
+        with urllib.request.urlopen(req, timeout=10) as response:
+            html = response.read().decode("utf-8", errors="ignore")
+
+        # YouTube embeds duration in the page as: "lengthSeconds":"<value>"
+        match = re.search(r'"lengthSeconds"\s*:\s*"(\d+)"', html)
+        if match:
+            duration = int(match.group(1))
+            logger.info(f"⏱️ Extracted video duration: {duration}s")
+            return duration
+
+        logger.warning("⚠️ lengthSeconds not found in YouTube page HTML.")
+        return 0
+    except Exception as e:
+        logger.warning(f"⚠️ Could not fetch YouTube duration: {e}")
+        return 0
+
 # --- Models ---
 class GenerateNotesRequest(BaseModel):
     youtube_url: HttpUrl
@@ -114,11 +155,15 @@ async def process_video_task(task_id: str, youtube_url: str, language: str, user
             "action_items": action_items.get("action_items", [])
         }
 
+        # استخراج مدة الفيديو الحقيقية بدلاً من الصفر المبرمج
+        tasks[task_id]["message"] = "Fetching video metadata..."
+        video_duration = get_youtube_duration(youtube_url)
+
         final_markdown = note_gen.format_final_notes(
             note_gen.format_notes_to_markdown(combined_notes),
             video_title,
             youtube_url,
-            0,
+            video_duration,
         )
 
         # الخطوة 3: الحفظ في Firebase