make image functionality optional

app.py

@@ -17,7 +17,7 @@ EXAM_MAX_QUESTIONS = int(
     os.getenv("EXAM_MAX_QUESTIONS", 5)
 )  # Limit quiz to max questions
 EXAM_PASSING_SCORE = float(os.getenv("EXAM_PASSING_SCORE", 0.8))
-EXAM_DATASET_ID = "burtenshaw/dummy-code-quiz"
+EXAM_DATASET_ID = "agents-course/smolagents-quiz-data"
 
 # prep the dataset for the quiz
 ds = load_dataset(EXAM_DATASET_ID, split="train", download_mode="force_redownload")
@@ -26,6 +26,9 @@ quiz_data = list(ds)  # Convert dataset to list instead of using to_list()
 if EXAM_MAX_QUESTIONS:
     quiz_data = quiz_data[:EXAM_MAX_QUESTIONS]
 
+# Check if dataset has image feature
+HAS_IMAGE_FEATURE = "image" in ds.features
+
 
 def format_python_code(code: str) -> str:
     """Format Python code using black."""
@@ -268,6 +271,13 @@ def handle_quiz(question_idx, user_answers, submitted_code, is_start):
         # Show the next question
         q = quiz_data[question_idx]
         challenge_text = f"## Question {question_idx + 1} \n### {q['challenge']}"
+
+        # Only show image if the feature exists and the question has an image
+        show_image = HAS_IMAGE_FEATURE and q.get("image") is not None
+        image_update = gr.update(
+            value=q.get("image") if show_image else None, visible=show_image
+        )
+
         return (
             challenge_text,  # question_text
             gr.update(value=q["placeholder"], visible=True),  # code_input
@@ -278,9 +288,7 @@ def handle_quiz(question_idx, user_answers, submitted_code, is_start):
             gr.update(visible=True),  # next_btn visible
             gr.update(visible=False),  # submit_btn hidden
             gr.update(visible=False),  # final_markdown hidden
-            gr.update(
-                value=q["image"], visible=True if q["image"] else False
-            ),  # question_image with current question image
+            image_update,  # question_image with current question image
         )
 
 
@@ -302,10 +310,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             question_text = gr.Markdown("")
             question_image = gr.Image(
-                label="Question Image", visible=True, type="pil"
+                label="Question Image",
+                visible=True if HAS_IMAGE_FEATURE else False,
+                type="pil",
             )  # Add image component
         with gr.Column():
-            code_input = gr.Code(language="python", label="Your Solution", visible=False)
+            code_input = gr.Code(
+                language="python", label="Your Solution", visible=False
+            )
 
     with gr.Row(variant="compact"):
         status_text = gr.Markdown("")