listing1.py

'''
Created on Nov 9, 2023

@author: immanueltrummer
'''
import argparse
import openai
import time

client = openai.OpenAI()


def analyze_image(image_url, question):
    """ Use language model to answer question about image.
    
    Args:
        image_url: URL leading to image.
        question: question about image.
    
    Returns:
        Answer generated by the language model.
    """
    for nr_retries in range(1, 4):
        try:
            response = client.chat.completions.create(
                model='gpt-4o',
                messages=[
                    {'role':'user', 'content':[
                        {'type':'text', 'text':question},
                        {'type':'image_url', 'image_url':{
                            'url':image_url, 'detail':'low'
                            }
                        }]
                    }]
                )
            return response.choices[0].message.content
        except:
            time.sleep(nr_retries * 2)
    raise Exception('Cannot query OpenAI model!')


if __name__ == '__main__':
    
    parser = argparse.ArgumentParser()
    parser.add_argument('imageurl', type=str, help='URL to image')
    parser.add_argument('question', type=str, help='Question about the image')
    args = parser.parse_args()
    
    answer = analyze_image(args.imageurl, args.question)
    print(answer)