Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|