Update vision_analyzer.py

vision_analyzer.py

@@ -133,7 +133,7 @@ def log_message(message: str):
 def log_failed_file(filename: str, error: str):
     """Log failed files to persistent file"""
     with open(FAILED_FILES_LOG, "a") as f:
-        f.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - {filename}: {error}\n")
+        f.write(f"{time.strftime("%Y-%m-%d %H:%M:%S")} - {filename}: {error}\n")
 
 def get_disk_usage(path: str) -> Dict[str, float]:
     """Get disk usage statistics in GB"""
@@ -330,8 +330,8 @@ def analyze_single_frame(image_path: str, preprompt: str = "") -> dict:
         
         # Process with proper padding configuration
         inputs = vision_processor(
-            images=image,
-            text=preprompt,
+            images=[image], # Wrap the single image in a list
+            text=[preprompt], # Wrap the single preprompt in a list
             return_tensors="pt",
             padding="max_length",
             max_length=512,
@@ -380,9 +380,9 @@ def process_video_frames(frames_dir: str, video_filename: str, output_file: str)
         # Validate first frame
         first_frame_result = analyze_single_frame(str(frames[0]), get_preprompt(video_filename))
         print("\n=== FIRST FRAME VALIDATION ===")
-        print(f"Image: {first_frame_result['image']}")
-        print(f"Result: {first_frame_result['description']}")
-        print(f"Status: {'Success' if first_frame_result['success'] else 'Failed'}\n")
+        print(f"Image: {first_frame_result["image"]}")
+        print(f"Result: {first_frame_result["description"]}")
+        print(f"Status: {"Success" if first_frame_result["success"] else "Failed"}\n")
 
         if not first_frame_result["success"]:
             print("❌ Aborting due to first frame failure")
@@ -466,7 +466,7 @@ def process_rar_file(rar_path: str) -> bool:
         # Process each video
         for video_path in video_files:
             video_filename = Path(video_path).name
-            frames_dir = os.path.join(FRAMES_OUTPUT_FOLDER, f"{course_name}_{video_filename.replace('.', '_')}_frames")
+            frames_dir = os.path.join(FRAMES_OUTPUT_FOLDER, f"{course_name}_{video_filename.replace(".", "_")}_frames")
             ensure_dir(frames_dir)
             
             # Extract frames
@@ -477,7 +477,7 @@ def process_rar_file(rar_path: str) -> bool:
             processing_status["extracted_frames_count"] += extracted_count
             
             # Analyze frames
-            analysis_output = os.path.join(ANALYSIS_OUTPUT_FOLDER, f"{course_name}_{video_filename.replace('.', '_')}_analysis.json")
+            analysis_output = os.path.join(ANALYSIS_OUTPUT_FOLDER, f"{course_name}_{video_filename.replace(".", "_")}_analysis.json")
             if process_video_frames(frames_dir, video_filename, analysis_output):
                 processing_status["analyzed_frames_count"] += extracted_count
                 processing_status["extracted_videos"] += 1
@@ -727,4 +727,5 @@ __all__ = [
     "extract_frames",
     "DEFAULT_FPS",
     "ensure_dir"
-]
+]
+