Update main.py

main.py

@@ -76,7 +76,8 @@ def load_image(source, ffmpeg_path='ffmpeg', frame_time=1):
     Image.MAX_IMAGE_PIXELS = None
 
     def extract_frame_from_video(video_path_or_url, time_sec):
-        print(f"A extrair video com url {video_path_or_url}")
+        print(f"[INFO] A extrair frame do vídeo: {video_path_or_url} no segundo {time_sec}")
+        
         with tempfile.NamedTemporaryFile(suffix='.jpg', delete=False) as temp_frame:
             frame_path = temp_frame.name
 
@@ -89,47 +90,64 @@ def load_image(source, ffmpeg_path='ffmpeg', frame_time=1):
             "-y",
             frame_path
         ]
-        subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+        print(f"[DEBUG] Comando ffmpeg: {' '.join(command)}")
+
+        result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+        if result.returncode != 0:
+            print(f"[ERRO] ffmpeg falhou com código {result.returncode}")
+            print(f"[ERRO] stderr: {result.stderr.decode('utf-8')}")
+            raise RuntimeError("Erro ao extrair frame com ffmpeg.")
 
         if not os.path.exists(frame_path):
-            raise ValueError("Failed to extract frame from video.")
+            print("[ERRO] Frame não criado. Verifica se o caminho do vídeo está correto e acessível.")
+            raise ValueError("Frame não encontrado após execução do ffmpeg.")
 
         frame = cv2.imread(frame_path, cv2.IMREAD_GRAYSCALE)
         os.remove(frame_path)
 
-        print(f"Video com url {video_path_or_url} extraido")
-        return frame
-
-    if source.startswith('http'):
-        mime_type, _ = mimetypes.guess_type(source)
-        if mime_type and mime_type.startswith('video'):
-            return extract_frame_from_video(source, frame_time)
+        if frame is None:
+            print("[ERRO] Falha ao ler frame extraído com OpenCV.")
+            raise ValueError("Erro ao carregar frame extraído.")
 
-        # Assume imagem
-        response = requests.get(source)
-        img = np.asarray(bytearray(response.content), dtype=np.uint8)
-        img = cv2.imdecode(img, cv2.IMREAD_GRAYSCALE)
-        return img
+        print(f"[SUCESSO] Frame extraído com sucesso de {video_path_or_url}")
+        return frame
 
-    else:
-        try:
-            img_bytes = base64.b64decode(source)
-            mime_type = mimetypes.guess_type("data")[0]  # fallback
-
-            # Tenta abrir como imagem
-            img = Image.open(BytesIO(img_bytes))
-            img = np.array(img)
-            img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+    try:
+        if source.startswith('http'):
+            mime_type, _ = mimetypes.guess_type(source)
+            print(f"[INFO] Mime_type de {source} é {mime_type}")
+            
+            if mime_type and mime_type.startswith('video'):
+                return extract_frame_from_video(source, frame_time)
+
+            print(f"[INFO] A carregar imagem de URL: {source}")
+            response = requests.get(source)
+            img = np.asarray(bytearray(response.content), dtype=np.uint8)
+            img = cv2.imdecode(img, cv2.IMREAD_GRAYSCALE)
             return img
-        except Exception:
-            # Se falhar, assumimos que é vídeo em base64 e extraímos frame
-            with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_video:
-                temp_video.write(base64.b64decode(source))
-                temp_video_path = temp_video.name
-
-            frame = extract_frame_from_video(temp_video_path, frame_time)
-            os.remove(temp_video_path)
-            return frame
+
+        else:
+            print("[INFO] A tentar carregar base64 como imagem.")
+            try:
+                img_bytes = base64.b64decode(source)
+                img = Image.open(BytesIO(img_bytes))
+                img = np.array(img)
+                img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+                return img
+            except Exception:
+                print("[INFO] Base64 não é imagem, a assumir que é vídeo.")
+                with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_video:
+                    temp_video.write(base64.b64decode(source))
+                    temp_video_path = temp_video.name
+
+                frame = extract_frame_from_video(temp_video_path, frame_time)
+                os.remove(temp_video_path)
+                return frame
+    except Exception as e:
+        print(f"[ERRO] Falha ao carregar imagem: {e}")
+        return None
 
 @app.post("/save")
 async def save(image_data: RequestModel):