INNER CODE UNIT · Python
__init__
AnsahMohammad/shots-studio · legacy/web/core.py:102
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:
response, status_code = self.fetch_ai_response(request_data, headers)
if status_code != 200:
return f"Error: {response}", status_code
return response, status_code