src/utils/image_qa.py

import streamlit as st
from PIL import Image as PILImage

def query_and_print_results(image_vdb, query):
    results = 3
    # Query the database
    query_results = image_vdb.query(
        query_texts=[query],
        n_results=results,
        include=['uris', 'distances']
    )

    # 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)
            with columns[idx]:
                st.image(img, width=300)
        except Exception as e:
            st.error(f"Error loading image {uri}: {e}")

# Example usage
# query_and_print_results(your_image_vdb, your_query)