from dotenv import find_dotenv, load_dotenv
from transformers import pipeline
# from langchain import PromptTemplate, LLMChain, OpenAI
from langchain_openai import ChatOpenAI
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from langchain_openai import ChatOpenAI
import requests
import os

load_dotenv(find_dotenv())
# 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 = os.getenv("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")

# openai_api_key = os.getenv("OPENAI_API_KEY") 

# if openai_api_key is None:  
#     raise ValueError("Khóa API OPENAI_API_KEY không tìm thấy. Vui lòng kiểm tra file .env.") 

# img2text
def img2text(url):
    image_to_text = pipeline("image-to-text", model = "Salesforce/blip-image-captioning-large")

    text = image_to_text(url)[0]["generated_text"]

    print(text)
    return text

img2text("Blackpink.png")