import argparse
import ollama

parser = argparse.ArgumentParser( description = "Use ollama to produce a short image description.")
parser.add_argument('input', default="scene.jpg", type=str, nargs='?', help="Specify input jpeg file name (default: %(default)s).")
args = parser.parse_args()

with open(args.input, "rb") as input:
    jpg_data = input.read()

print("Read %d bytes of image data from %s." % (len(jpg_data), args.input))

prompt = "Please describe this image in twenty words or fewer."
response = ollama.generate(model='gemma3:4b', images=[jpg_data], prompt=prompt)

print("Reply generated in %f seconds:" % (1e-9*response.total_duration))
print(response.response)