阿里百炼识别本地图片内容

import os
import base64
import requests
from datetime import datetime
from openai import OpenAI
'''
内容识别完成，测试通过,date:250730
'''

class AliBailianImageService:
def __init__(self, configuration):
self.configuration = configuration
def get_api_key(self):
api_key = os.getenv("DASHSCOPE_API_KEY")
# 若没有配置环境变量，请用百炼API Key将下行替换为：api_key = "sk-xxx"
if not api_key:
print("API Key 未设置。请确保环境变量 'DASHSCOPE_API_KEY' 已设置。")
return None
return api_key
def get_image_base64_string_and_save(self, image_url):
response = requests.get(image_url)
if response.status_code != 200:
raise Exception(f"Failed to download image: {response.status_code}")
image_data = response.content
encoded_image = base64.b64encode(image_data).decode('utf-8')
return encoded_image
def send_post_request(self, url, json_content, api_key):
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json",
"Accept": "application/json"
}
response = requests.post(url, data=json_content, headers=headers)
response_body = response.text
self.write_response_to_log(response_body)
if response.status_code >= 200 and response.status_code < 300:
return response_body
else:
return f"请求失败: {response.status_code}"
def write_response_to_log(self, response_body):
log_file_path = "../convert_files/Logs/response.log"
log_dir = os.path.dirname(log_file_path)
if not os.path.exists(log_dir):
os.makedirs(log_dir)
with open(log_file_path, 'a') as f:
f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Response Body: {response_body}\n")

def image_to_base64(self,file_path):
with open(file_path, "rb") as image_file:
# 读取图片二进制数据
image_data = image_file.read()
# 转换为Base64编码字节
base64_bytes = base64.b64encode(image_data)
# 转换为UTF-8字符串
base64_string = base64_bytes.decode("utf-8")
return base64_string
def get_results(self,file_path):
api_key = self.get_api_key()
print("api_key>>",api_key)
if not api_key:
return None
encoded_image = self.image_to_base64(file_path)
client = OpenAI(
# 若没有配置环境变量，请用百炼API Key将下行替换为：api_key="sk-xxx",
api_key=api_key,
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
)
completion = client.chat.completions.create(
model="qwen-vl-plus",
# 此处以qwen-vl-plus为例，可按需更换模型名称。模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
messages=[{"role": "user", "content": [
{"type": "image_url", "image_url": {"url": f"data:image/png;base64,{encoded_image}"}},
{"type": "text", "text": "识别图片内容，只输出图片内容"},
]}]
)
print(completion.model_dump_json())

if __name__ == "__main__":
img_file_path = "../convert_files/words_001.png"
# 示例配置
config = {
"DASHSCOPE_API_KEY": "your_api_key"
}
service = AliBailianImageService(config)
result = service.get_results(img_file_path)
if result:
print("Result:", result)