Update image display in app.py

app.py

@@ -84,11 +84,13 @@ def app():
         search_results = qdrant_client.search(
             collection_name="images-clip",
             query_vector=text_features,
-            limit=20,
+            limit=50,
         )
         elapsed = time() - start
         st.write(f"elapsed: {elapsed:.2f} sec")
         st.write(f"total: {len(search_results)}")
+        images = []
+        captions = []
         for r in search_results:
             score = r.score
             if payload := r.payload:
@@ -96,9 +98,17 @@ def app():
             else:
                 name = "unknown"
             image_url = get_image_url(name, image_type="xsmall")
-            st.write(f"score: {score:.2f}")
-            st.image(image_url, width=200)
-            st.write("---")
+            images.append(image_url)
+            captions.append(f"{name} ({score:.4f})")
+        image_group_n = 6
+        for i in range(0, len(images), image_group_n):
+            target_images = images[i : i + image_group_n]
+            target_captions = captions[i : i + image_group_n]
+            st.image(
+                target_images,
+                caption=target_captions,
+                width=160,
+            )
 
 
 if __name__ == "__main__":