Og2 commited on
Commit
31b9e78
·
verified ·
1 Parent(s): 6db8d96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -594,18 +594,19 @@ async def get_video_frames(file_name: str, frame_id: int) -> dict:
594
  if csv_response.status_code == 200:
595
  csv_bytes = io.BytesIO(csv_response.content)
596
  df_annotations = pd.read_csv(csv_bytes)
 
 
 
597
 
598
  for i in range(frame_id, frame_id + STEP):
599
  frame_filename = f"{Path(file_name).stem}/{Path(file_name).stem}_{i}.png"
600
- print(frame_filename)
601
  image_url = f"{GITHUB_BASE_URL}/frames/{frame_filename}"
602
- print(image_url)
603
- print(csv_url)
604
-
605
 
606
  # Vérifier si l'image existe
607
  image_response = requests.head(image_url)
608
  if image_response.status_code != 200:
 
609
  continue
610
 
611
  class_label, player = None, None
@@ -619,12 +620,13 @@ async def get_video_frames(file_name: str, frame_id: int) -> dict:
619
  raise HTTPException(status_code=404, detail="Aucune image trouvée dans la plage demandée")
620
 
621
  return {"status": "success", "frames": frames}
 
622
  except Exception as e:
 
623
  raise HTTPException(status_code=500, detail=f"Erreur lors de la récupération des images : {str(e)}")
624
 
625
 
626
 
627
-
628
  UPLOAD_DIR = Path("/app/csv") # Dossier temporaire pour stocker les csv
629
  @app.post("/annotate/")
630
  async def annotate(
 
594
  if csv_response.status_code == 200:
595
  csv_bytes = io.BytesIO(csv_response.content)
596
  df_annotations = pd.read_csv(csv_bytes)
597
+ print("CSV Loaded successfully")
598
+ else:
599
+ print(f"CSV not found at {csv_url}")
600
 
601
  for i in range(frame_id, frame_id + STEP):
602
  frame_filename = f"{Path(file_name).stem}/{Path(file_name).stem}_{i}.png"
603
+ print(f"Checking for frame: {frame_filename}")
604
  image_url = f"{GITHUB_BASE_URL}/frames/{frame_filename}"
 
 
 
605
 
606
  # Vérifier si l'image existe
607
  image_response = requests.head(image_url)
608
  if image_response.status_code != 200:
609
+ print(f"Image not found: {image_url}")
610
  continue
611
 
612
  class_label, player = None, None
 
620
  raise HTTPException(status_code=404, detail="Aucune image trouvée dans la plage demandée")
621
 
622
  return {"status": "success", "frames": frames}
623
+
624
  except Exception as e:
625
+ print(f"Error: {str(e)}") # Log the error for debugging
626
  raise HTTPException(status_code=500, detail=f"Erreur lors de la récupération des images : {str(e)}")
627
 
628
 
629
 
 
630
  UPLOAD_DIR = Path("/app/csv") # Dossier temporaire pour stocker les csv
631
  @app.post("/annotate/")
632
  async def annotate(