salman/igny8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

image url

Browse Source
This commit is contained in:
Desktop
2025-11-12 10:46:01 +05:00
parent a61471c34a
commit 1a51e6bc39
3 changed files with 142 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
backend/igny8_core/ai/tasks.py
View File

@@ -527,67 +527,23 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
}
)
# Download and save image to /data/app/images
# Download and save image to /data/app/igny8/frontend/public/images/ai-images
saved_file_path = None
if image_url:
try:
import os
import requests
from django.conf import settings
from pathlib import Path
import time
# Create images directory if it doesn't exist
# Use frontend/public/images/ai-images/ for web accessibility (like /images/logo/)
# This allows images to be served via app.igny8.com/images/ai-images/
write_test_passed = False
images_dir = None
# Use the correct path: /data/app/igny8/frontend/public/images/ai-images
# This is web-accessible via /images/ai-images/ (Vite serves from public/)
images_dir = '/data/app/igny8/frontend/public/images/ai-images'
# Try frontend/public/images/ai-images/ first (web-accessible)
try:
# Use absolute path: /data/app/igny8/frontend/public/images/ai-images/
# This matches the structure where frontend is at project root level
images_dir = '/data/app/igny8/frontend/public/images/ai-images'
os.makedirs(images_dir, exist_ok=True)
# Test write access
test_file = os.path.join(images_dir, '.write_test')
with open(test_file, 'w') as f:
f.write('test')
os.remove(test_file)
write_test_passed = True
logger.info(f"[process_image_generation_queue] Image {image_id} - Directory writable (web-accessible): {images_dir}")
except Exception as web_dir_error:
logger.warning(f"[process_image_generation_queue] Image {image_id} - Web-accessible directory not writable: {images_dir}, error: {web_dir_error}")
# Fallback to /data/app/igny8/images (mounted volume)
try:
images_dir = '/data/app/igny8/images'
os.makedirs(images_dir, exist_ok=True)
test_file = os.path.join(images_dir, '.write_test')
with open(test_file, 'w') as f:
f.write('test')
os.remove(test_file)
write_test_passed = True
logger.info(f"[process_image_generation_queue] Image {image_id} - Using mounted volume directory: {images_dir}")
except Exception as mounted_error:
logger.warning(f"[process_image_generation_queue] Image {image_id} - Mounted directory not writable: {images_dir}, error: {mounted_error}")
# Final fallback to /data/app/images
try:
images_dir = '/data/app/images'
os.makedirs(images_dir, exist_ok=True)
test_file = os.path.join(images_dir, '.write_test')
with open(test_file, 'w') as f:
f.write('test')
os.remove(test_file)
write_test_passed = True
logger.info(f"[process_image_generation_queue] Image {image_id} - Using fallback directory: {images_dir}")
except Exception as final_error:
logger.error(f"[process_image_generation_queue] Image {image_id} - All directories not writable. Last error: {final_error}")
raise Exception(f"None of the image directories are writable. Last error: {final_error}")
if not write_test_passed:
raise Exception(f"Failed to find writable directory for saving images")
# Create directory if it doesn't exist
os.makedirs(images_dir, exist_ok=True)
logger.info(f"[process_image_generation_queue] Image {image_id} - Using directory: {images_dir}")
# Generate filename: image_{image_id}_{timestamp}.png (or .webp for Runware)
import time
timestamp = int(time.time())
# Use webp extension if provider is Runware, otherwise png
file_ext = 'webp' if provider == 'runware' else 'png'