diff --git a/scripts/surveys/survey_maker.py b/scripts/surveys/survey_maker.py index 8d1b3d9c29120fca192d8a1946c43d0084760f67..b1dbd6c60673d1ca984b8a6671ad78d7e33c17a6 100644 --- a/scripts/surveys/survey_maker.py +++ b/scripts/surveys/survey_maker.py @@ -26,6 +26,8 @@ for i, row in df_questions_qcm.iterrows(): o["question"] = f"audio:{REMOTE_ITEMS_FOLDER}/{id_}/q.mp3" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.jpeg"): 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" else: print(f"Failed to find a question for item {id_}") @@ -53,6 +55,8 @@ for i, row in df_questions_qcm.iterrows(): o["question_qcm"][op] = f"audio:{REMOTE_ITEMS_FOLDER}/{id_}/{j}.mp3" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/{j}.jpeg"): 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" # PARSE GAPFILL QUESTIONS @@ -72,6 +76,8 @@ for i, row in df_questions_gapfill.iterrows(): o["question"] = f"audio:{REMOTE_ITEMS_FOLDER}/{id_}/q.mp3" elif os.path.isfile(f"{LOCAL_ITEMS_FOLDER}/{id_}/q.jpeg"): 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" else: print(f"Failed to find a question for item {id_}")