created video parser

video_parser.py

@@ -0,0 +1,64 @@
+import cv2
+import requests
+import tempfile
+import os
+from urllib.parse import urlparse, parse_qs
+import yt_dlp
+
+class VideoParser:
+    def __init__(self):
+        self.temp_dir = tempfile.mkdtemp()
+    
+    def download_youtube_video(self, url: str) -> str:
+        """Download YouTube video and return local path"""
+        ydl_opts = {
+            'format': 'best[height<=720]',
+            'outtmpl': os.path.join(self.temp_dir, '%(title)s.%(ext)s'),
+            'quiet': True
+        }
+        
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            return ydl.prepare_filename(info)
+    
+    def analyze_video_frames(self, video_path: str, sample_rate: int = 30):
+        """Analyze video frames for object detection/counting"""
+        cap = cv2.VideoCapture(video_path)
+        frame_count = 0
+        results = []
+        
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+                
+            if frame_count % sample_rate == 0:
+                # Basic frame analysis - you'd integrate with object detection here
+                results.append({
+                    'frame': frame_count,
+                    'timestamp': frame_count / cap.get(cv2.CAP_PROP_FPS),
+                    'frame_data': frame
+                })
+            
+            frame_count += 1
+        
+        cap.release()
+        return results
+    
+    def extract_audio(self, video_path: str) -> str:
+        """Extract audio from video for speech analysis"""
+        audio_path = video_path.rsplit('.', 1)[0] + '.wav'
+        
+        # Use ffmpeg to extract audio
+        import subprocess
+        subprocess.run([
+            'ffmpeg', '-i', video_path, '-vn', '-acodec', 'pcm_s16le', 
+            '-ar', '16000', '-ac', '1', audio_path, '-y'
+        ], capture_output=True)
+        
+        return audio_path
+    
+    def cleanup(self):
+        """Clean up temporary files"""
+        import shutil
+        shutil.rmtree(self.temp_dir, ignore_errors=True)