Add time

app/face_check/service.py

@@ -1,18 +1,24 @@
 import httpx
 from app.config import settings
 from fastapi import HTTPException
+import time
 
 class AIService:
     async def call_geolocation(file_bytes, meta_bytes):
         try:
+            start_time = time.perf_counter()
             async with httpx.AsyncClient(base_url=settings.SERVICE_GEOLOCATION_URL, timeout=1000) as client:
                 files_payload = [("files", (name, content, ctype)) for name, content, ctype in file_bytes]
                 files_payload.append(("files", (meta_bytes[0], meta_bytes[1], meta_bytes[2])))
 
                 resp = await client.post("/g3/predict", files=files_payload)
                 resp.raise_for_status()
+
+                elapsed = time.perf_counter() - start_time
+                print(f"⏱️ [call_geolocation] Service call took {elapsed:.2f} seconds")
                 return resp.json()
         except httpx.HTTPError as e:
+            print(f"❌ [call_geolocation] error: {str(e)}")
             raise HTTPException(
                 status_code=500,
                 detail=f"[call_geolocation] error: {str(e)}"
@@ -20,14 +26,19 @@ class AIService:
 
     async def call_timestamp(file_bytes, meta_bytes):
         try:
+            start_time = time.perf_counter()
             async with httpx.AsyncClient(base_url=settings.SERVICE_TIMESTAMP_URL, timeout=1000) as client:
                 files_payload = [("files", (name, content, ctype)) for name, content, ctype in file_bytes]
                 files_payload.append(("metadata_file", (meta_bytes[0], meta_bytes[1], meta_bytes[2])))
 
                 resp = await client.post("/analyze/", files=files_payload)
                 resp.raise_for_status()
+
+                elapsed = time.perf_counter() - start_time
+                print(f"⏱️ [call_timestamp] Service call took {elapsed:.2f} seconds")
                 return resp.json()
         except httpx.HTTPError as e:
+            print(f"❌ [call_timestamp] error: {str(e)}")
             raise HTTPException(
                 status_code=500,
                 detail=f"[call_timestamp] error: {str(e)}"
@@ -35,6 +46,7 @@ class AIService:
 
     async def call_AIGVDetection(file_bytes):
         try:
+            start_time = time.perf_counter()
             async with httpx.AsyncClient(base_url=settings.SERVICE_AIGVDET_URL, timeout=1000) as client:
                 if not file_bytes:
                     raise ValueError("No files provided")
@@ -45,8 +57,12 @@ class AIService:
 
                 resp = await client.post("/predict", files=files_payload)
                 resp.raise_for_status()
+
+                elapsed = time.perf_counter() - start_time
+                print(f"⏱️ [call_AIGVDetection] Service call took {elapsed:.2f} seconds")
                 return resp.json()
         except httpx.HTTPError as e:
+            print(f"❌ [call_AIGVDetection] error: {str(e)}")
             raise HTTPException(
                 status_code=500,
                 detail=f"[call_AIGVDetection] error: {str(e)}"
@@ -54,9 +70,13 @@ class AIService:
         
     async def call_Report(data):
         try:
+            start_time = time.perf_counter()
             async with httpx.AsyncClient(base_url=settings.SERVICE_REPORT_URL, timeout=1000) as client:
                 resp = await client.post("/v1/report", json=data)
                 resp.raise_for_status()
+
+                elapsed = time.perf_counter() - start_time
+                print(f"⏱️ [call_Report] Service call took {elapsed:.2f} seconds")
                 return resp.json()
         except httpx.HTTPError as e:
             raise HTTPException(

app/face_check/view.py

@@ -10,11 +10,13 @@ from app.config import settings
 import os
 from fastapi.responses import FileResponse, JSONResponse
 import shutil
+import time
 
 class FaceCheckView:
     async def face_check(self, files: List[UploadFile], metadata_file: UploadFile):
         try:
             print("[face_check] start")
+            start_time = time.perf_counter()
             file_bytes = [(f.filename, await f.read(), f.content_type) for f in files]
             meta_bytes = (metadata_file.filename, await metadata_file.read(), metadata_file.content_type)
             save_dir = create_timestamped_folder(settings.SAVE_DATA_PATH)
@@ -42,6 +44,7 @@ class FaceCheckView:
 
             media_base64_list = await convert_keyframes_to_base64(ts_result)
             sources = [item.get("source") for item in ts_result.get("results", []) if "source" in item]
+            score = aigv_result.get("authentic_confidence_score")
 
             save_json({
                 "media_id": "1",
@@ -77,16 +80,21 @@ class FaceCheckView:
                 "media_url": ts_result.get("results", [])[0].get("source", ""),
                 "timestamp": ts_result["results"][0]["timestamp"],
                 "source_url": ts_result.get("results", [])[0].get("source", ""),
-
-                "media_analysis": {
-                    "ai_generated": (aigv_result.get("authentic_confidence_score") or 0) < 0.5,
-                    "confidence": aigv_result.get("authentic_confidence_score") or 0
-                },
+                "media_analysis": (
+                    {
+                        "ai_generated": score < 0.5,
+                        "confidence": score
+                    }
+                    if score is not None
+                    else None
+                ),
                 "media": geo_result.get("media") or media_base64_list
             })
 
             save_string_as_md(report_data.get("readme_content", ""), save_dir, "report.md")
-
+            
+            elapsed = time.perf_counter() - start_time
+            print(f"⏱️ [face_check] Service call took {elapsed:.2f} seconds")
             return {"save to": f"{save_dir}"}
         except Exception as e:
             return JSONResponse(