ypdate handel

handler.py

@@ -1,6 +1,13 @@
 import os
-import cv2
 import datetime
+
+# Try to import cv2, but make it optional
+try:
+    import cv2
+    CV2_AVAILABLE = True
+except ImportError:
+    CV2_AVAILABLE = False
+    print("Warning: cv2 (OpenCV) not available. Video processing will be disabled.")
 import torch
 import numpy as np
 import hashlib
@@ -88,6 +95,8 @@ def vote_last_response(state, vote_type, model_selector):
             print(f"Failed to upload vote file: {e}")
 
 def is_valid_video_filename(name):
+    if not CV2_AVAILABLE:
+        return False  # Video processing disabled
     video_extensions = ["avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg"]
     ext = name.split(".")[-1].lower()
     return ext in video_extensions
@@ -98,6 +107,9 @@ def is_valid_image_filename(name):
     return ext in image_extensions
 
 def sample_frames(video_file, num_frames):
+    if not CV2_AVAILABLE:
+        raise ImportError("cv2 (OpenCV) not available. Video processing is disabled.")
+    
     video = cv2.VideoCapture(video_file)
     total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
     interval = total_frames // num_frames
@@ -271,9 +283,12 @@ def generate_response(message_text, image_input, temperature=0.05, top_p=1.0, ma
             image.save(filename)
         
         # Process image for model
-        image_tensor = process_images([image], our_chatbot.image_processor, our_chatbot.model.config)[0]
-        image_tensor = image_tensor.half().to(our_chatbot.model.device)
-        image_tensor = image_tensor.unsqueeze(0)
+        try:
+            image_tensor = process_images([image], our_chatbot.image_processor, our_chatbot.model.config)[0]
+            image_tensor = image_tensor.half().to(our_chatbot.model.device)
+            image_tensor = image_tensor.unsqueeze(0)
+        except Exception as e:
+            return {"error": f"Image processing failed: {str(e)}"}
         
         # Prepare conversation
         inp = DEFAULT_IMAGE_TOKEN + "\n" + message_text