diff --git a/whisk_client.py b/whisk_client.py index 51ab54e..861e324 100644 --- a/whisk_client.py +++ b/whisk_client.py @@ -137,14 +137,16 @@ def generate_image_whisk(prompt, cookie_str, **kwargs): access_token = get_session_token(cookies) - ref_image_path = kwargs.get('reference_image_path') - media_generation_id = None - - if ref_image_path: - try: - media_generation_id = upload_reference_image(ref_image_path, cookies) - except Exception as e: - logger.warning(f"Skipping reference image due to upload error: {e}") +def load_template_favorites(): + path = get_config_path('gallery_favorites.json') # Typo in original? No, it uses gallery_favorites for both? Let's check logic. + # Ah, original code had a separate logic or file? + # Let me check the original code via view first to be safe. + # It seems there was no load_template_favorites definition in the previous view. + # I will replace the top of the file definitions if they exist there. + pass + +# Retrying with correct strategy: replace the specific lines if I can find them. +# I'll view the file top to locate these helpers first. aspect_ratio_map = { "1:1": "IMAGE_ASPECT_RATIO_SQUARE", @@ -242,8 +244,14 @@ def generate_image_whisk(prompt, cookie_str, **kwargs): images.append(img['encodedImage']) if not images: - logger.error(f"Unexpected response structure: {json_resp.keys()}") - raise WhiskClientError("No images found in response") + import json + logger.error(f"WHISK DEBUG - Full Response: {json.dumps(json_resp)}") + # check for common non-standard errors + if 'error' in json_resp: + err_msg = json_resp['error'] + raise WhiskClientError(f"Whisk API returned error: {err_msg}") + + raise WhiskClientError(f"No images found in response. Response keys: {list(json_resp.keys())}") return base64.b64decode(images[0])