INNER CODE UNIT · Python

GeminiModel

AnsahMohammad/shots-studio · legacy/web/core.py:98

class GeminiModel:
    base_url = "https://generativelanguage.googleapis.com/v1beta/models"
    timeout = 30

    def __init__(self, model_name: str = "gemini-2.0-flash"):
        self.model_name = model_name
        self.api_key = os.getenv("GEMINI_API_KEY")
        if not self.api_key:
            raise ValueError("GEMINI_API_KEY environment variable is not set")

    def process_images(self, images):
        if not images:
            return "No images to process.", 400

        print(f"Processing {len(images)} images...")
        request_data, headers = self._prepare_request_data(images)

        try:

View source record →