Spaces:
Sleeping
Sleeping
| import base64 | |
| from dotenv import load_dotenv | |
| from . import anthropic_api as anthropic_extractor | |
| from . import openai_api as openai_info_extractor | |
| def encode_image(image_path): | |
| with open(image_path, "rb") as image_file: | |
| return base64.b64encode(image_file.read()).decode("utf-8") | |
| def extract_info(image_paths, provider, schema, known_data=None): | |
| if provider == "anthropic": | |
| extractor = anthropic_extractor | |
| elif provider == "openai": | |
| extractor = openai_info_extractor | |
| else: | |
| raise ValueError("Invalid provider") | |
| return extractor.extract_info(image_paths, schema, known_data=known_data) | |
| def follow_structure(json_info, provider, schema, known_data=None): | |
| if provider == "anthropic": | |
| extractor = anthropic_extractor | |
| elif provider == "openai": | |
| extractor = openai_info_extractor | |
| else: | |
| raise ValueError("Invalid provider") | |
| return extractor.follow_structure(json_info, schema, known_data=known_data) | |