Update src/utils/image_qa.py

src/utils/image_qa.py

@@ -10,14 +10,16 @@ def query_and_print_results(image_vdb, query):
         include=['uris', 'distances']
     )
 
-    # Print the results
+    # Create columns to display images side by side
+    columns = st.columns(results)
+    
     for idx, uri in enumerate(query_results['uris'][0]):
         try:
             img = PILImage.open(uri)
-            st.image(img, width=300)
+            with columns[idx]:
+                st.image(img, width=300)
         except Exception as e:
             st.error(f"Error loading image {uri}: {e}")
 
-# Example usage (this part is just for illustration and testing, remove it in the actual Streamlit app):
-# Assuming image_vdb is already created and query is given
-# query_and_print_results(image_vdb, "example query")
+# Example usage
+# query_and_print_results(your_image_vdb, your_query)