Update app.py

app.py

@@ -578,25 +578,22 @@ def find_best_subtitle_region(frame, ocr_model, region_height_ratio=0.35, num_st
         # Fallback to center-bottom strip
         fallback_y = height - int(height * 0.2)
         return frame[fallback_y:, :], (fallback_y, height)
-
-def ocr_frame_worker(args):
+        
+def ocr_frame_worker(args, min_confidence=0.7):
     frame_idx, frame_time, frame = args
 
-    init_ocr_model()  # Ensure model is loaded once per process
-
-    if frame is None or frame.size == 0:
-        return {"time": frame_time, "text": ""}
+    init_ocr_model()  # Load model in thread-safe way
 
-    if not isinstance(frame, np.ndarray):
+    if frame is None or frame.size == 0 or not isinstance(frame, np.ndarray):
         return {"time": frame_time, "text": ""}
 
     if frame.dtype != np.uint8:
         frame = frame.astype(np.uint8)
 
     try:
-        subtitle_crop, _ = find_best_subtitle_region(frame, ocr_model)
-        result = ocr_model.ocr(subtitle_crop, cls=True)
-        texts = [line[1][0] for line in result[0]] if result[0] else []
+        result = ocr_model.ocr(frame, cls=True)
+        lines = result[0] if result else []
+        texts = [line[1][0] for line in lines if line[1][1] >= min_confidence]
         combined_text = " ".join(texts).strip()
         return {"time": frame_time, "text": combined_text}
     except Exception as e: