Update app.py

app.py

@@ -63,38 +63,12 @@ def query_gpt4o_mini(query, images):
             {
               "type": "text",
               "text": PROMPT.format(query=query)
-            },
-            {
-              "type": "image_url",
-              "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_images[0]}"
-                },
-            },
-            {
-              "type": "image_url",
-              "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_images[1]}"
-                },
-            },
-            {
-              "type": "image_url",
-              "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_images[2]}"
-                },
-            },
-            {
-              "type": "image_url",
-              "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_images[3]}"
-                },
-            },
-            {
+            }] + [{
               "type": "image_url",
               "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_images[4]}"
+                "url": f"data:image/jpeg;base64,{im}"
                 },
-            },
-          ],
+            } for im in base64_images]
         }
       ],
       max_tokens=500,