.

main.py

@@ -131,7 +131,7 @@ def load_initial_images(width, height):
     initial_images = []
     if DEBUG_TEACHER_FORCING:
         # Load the previous 7 frames for image_81
-        for i in range(114-7, 114):  # Load images 74-80
+        for i in range(117-7, 117):  # Load images 74-80
             img = Image.open(f"record_10003/image_{i}.png")#.resize((width, height))
             initial_images.append(np.array(img))
     else:
@@ -338,8 +338,8 @@ async def websocket_endpoint(websocket: WebSocket):
                          'L + 0 4 7 3 : + 0 0 8 7', 'N + 0 1 0 9 : + 0 3 4 4', 
                          'N + 0 0 5 2 : + 0 1 9 4', 'N + 0 3 6 5 : + 0 2 3 2', 
                          'N + 0 3 8 9 : + 0 2 4 5', 'N + 0 0 2 0 : + 0 0 5 9', 
-                         'N + 0 4 7 3 : + 0 1 5 7', ]#'L + 0 1 9 1 : + 0 0 8 7', 
-                         #'L + 0 1 9 1 : + 0 0 8 7', 'N + 0 3 4 3 : + 0 2 6 3', ]
+                         'N + 0 4 7 3 : + 0 1 5 7', 'L + 0 1 9 1 : + 0 0 8 7', 
+                         'L + 0 1 9 1 : + 0 0 8 7', 'N + 0 3 4 3 : + 0 2 6 3', ]
                          #'N + 0 2 0 5 : + 0 1 3 3']
         previous_actions = []
         for action in debug_actions[-8:]:
@@ -374,6 +374,7 @@ async def websocket_endpoint(websocket: WebSocket):
                          'N + 0 2 9 0 : + 0 1 4 1', 'N + 0 4 0 2 : + 0 0 0 9', 
                          'N + 0 3 0 7 : + 0 3 3 2', 'N + 0 2 2 0 : + 0 3 7 1', 
                          'N + 0 0 8 2 : + 0 1 5 1']
+        positions = positions[3:]
 #positions = positions[:4]
     position = positions[0]
     positions = positions[1:]
@@ -417,13 +418,13 @@ async def websocket_endpoint(websocket: WebSocket):
                 
                 # Predict the next frame based on the previous frames and actions
                 if DEBUG_TEACHER_FORCING:
-                    print ('predicting', f"record_10003/image_{114+len(previous_frames)}.png")
+                    print ('predicting', f"record_10003/image_{117+len(previous_frames)}.png")
 
                 next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
                 # Load and append the corresponding ground truth image instead of model output
                 print ('here4', len(previous_frames))
                 if True and DEBUG_TEACHER_FORCING:
-                    img = Image.open(f"record_10003/image_{114+len(previous_frames)}.png")
+                    img = Image.open(f"record_10003/image_{117+len(previous_frames)}.png")
                     previous_frames.append(img)
                 elif True:
                     assert False