From 2fdc5d04190968807536e8561ee4f2d72044896e Mon Sep 17 00:00:00 2001 From: Serge Bibauw <sbibauw@gmail.com> Date: Thu, 20 Mar 2025 09:47:38 +0100 Subject: [PATCH] ktests: Fix: response images can be WEBP/JPG/JPEG --- scripts/surveys/survey_maker.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/scripts/surveys/survey_maker.py b/scripts/surveys/survey_maker.py index b1dbd6c6..70443bdd 100644 --- a/scripts/surveys/survey_maker.py +++ b/scripts/surveys/survey_maker.py @@ -28,6 +28,8 @@ for i, row in df_questions_qcm.iterrows(): o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.jpeg" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.jpg"): o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.jpg" + elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.webp"): + o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.webp" else: print(f"Failed to find a question for item {id_}") @@ -57,6 +59,8 @@ for i, row in df_questions_qcm.iterrows(): o["question_qcm"][op] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/{j}.jpeg" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/{j}.jpg"): o["question_qcm"][op] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/{j}.jpg" + elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/{j}.webp"): + o["question_qcm"][op] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/{j}.webp" # PARSE GAPFILL QUESTIONS @@ -78,6 +82,8 @@ for i, row in df_questions_gapfill.iterrows(): o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.jpeg" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.jpg"): o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.jpg" + elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.webp"): + o["question"] = f"image:{REMOTE_ITEMS_FOLDER}/{id_}/q.webp" else: print(f"Failed to find a question for item {id_}") -- GitLab