Update app.py

app.py

@@ -35,15 +35,13 @@ def start_game():
     # Initialize game state
     questions_asked = 0
     possible_objects = list(objects_and_questions.keys())  # Start with all objects
-    current_questions = {}  # Dict to track questions for each object
+    current_questions = {obj: 0 for obj in possible_objects}  # Initialize the index for each object
 
     while questions_asked < 10 and len(possible_objects) > 1:
         question = None
         for obj in possible_objects:
-            if obj not in current_questions:
-                current_questions[obj] = 0  # Start with the first question for this object
+            # Make sure we have not exhausted all questions for the object
             if current_questions[obj] < len(objects_and_questions[obj]):
-                # Get the next unanswered question for the object
                 question = objects_and_questions[obj][current_questions[obj]]
                 break
         
@@ -53,13 +51,13 @@ def start_game():
         # Ask the question
         answer = st.radio(question, ('Yes', 'No'), key=questions_asked)
         
-        # Update the progress
+        # Filter objects based on the answer
         if answer == 'Yes':
             possible_objects = [obj for obj in possible_objects if objects_and_questions[obj][current_questions[obj]] == question]
         else:
             possible_objects = [obj for obj in possible_objects if objects_and_questions[obj][current_questions[obj]] != question]
         
-        # Move to the next question for the object
+        # Move to the next question for the filtered objects
         for obj in possible_objects:
             current_questions[obj] += 1