Update main.py

main.py

@@ -67,7 +67,6 @@ def mobilenet_similarity(img1, img2):
 
         sim = cosine_similarity(feat1, feat2)[0][0]  # Valor entre -1 e 1
         sim_score = (sim + 1) * 50  # Escalar para 0-100
-        print(f"MobileNet similarity score is {sim_score}")
         return float(sim_score)
     except Exception as e:
         print("Erro ao calcular similaridade com MobileNet")
@@ -77,6 +76,7 @@ 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}")
         with tempfile.NamedTemporaryFile(suffix='.jpg', delete=False) as temp_frame:
             frame_path = temp_frame.name
 
@@ -96,6 +96,8 @@ def load_image(source, ffmpeg_path='ffmpeg', frame_time=1):
 
         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'):
@@ -142,9 +144,7 @@ async def save(image_data: RequestModel):
 
     if img1 is not None and img2 is not None:
         similarity_orb = orb_sim(img1, img2)
-        print(f"Similaridade ORB entre {image_data.originSource} e {image_data.source}: {similarity_orb}")
         similarity_mobilenet = mobilenet_similarity(img1, img2)
-        print(f"Similaridade Mobilenet entre {image_data.originSource} e {image_data.source}: {similarity_mobilenet}")
 
     data_to_save = image_data.dict()
     if similarity_orb is not None: