Update app.py

app.py

@@ -76,12 +76,12 @@ def navigate(screenshot, task: str, platform: str, history):
     else:
         messages = history
 
-    # for message in messages:
-    #     if message['role'] == 'user' and isinstance(message.get('content'), list):
-    #         for item in message['content']:
-    #             if item.get('type') == 'image_url' and isinstance(item.get('image_url'), str):
-    #                 # This is a base64 string, convert it to a PIL image
-    #                 item['image_url'] = load_base64_to_pil(item['image_url'])
+    for message in messages:
+        if message['role'] == 'user' and isinstance(message.get('content'), list):
+            for item in message['content']:
+                if item.get('type') == 'image' and isinstance(item.get('image'), str):
+                    # This is a base64 string, convert it to a PIL image
+                    item['image'] = load_base64_to_pil(item['image'])
 
     prompt_header = (
             "You are a GUI agent. You are given a task and your action history, with screenshots."