ocr.py

from openai import OpenAI
import os


def ocr(base64_image):
    api_key = os.getenv('OPENAI_API')
    client = OpenAI(api_key=api_key)
    response = client.chat.completions.create(
    model="gpt-4-turbo",
    messages=[
        {
        "role": "user",
        "content": [
            {"type": "text", "text": "Transcribe the text on here. Describe diagram if there are any. Be concise. If there are any blue highlighting, only return that text inside triple backslash quote with no preamble"},
            {
            "type": "image_url",
            "image_url": {"url": f"data:image/jpeg;base64,{base64_image}"},
            },
        ],
        }
    ],
    max_tokens=300,
    )
    return response.choices[0].message.content

if __name__ == "__main__":
    print(ocr(url = "https://github.com/EllAchE/llama-out-loud/blob/ocr/image/zoom_book.png?raw=true"))