Function upload_image

main.py:1310–1366

Upload image with WebP conversion and upload to S3

(request: Request, file: UploadFile = File(...))

@app.post("/api/upload-image")
async def upload_image(request: Request, file: UploadFile = File(...)):
    """Upload image with WebP conversion and upload to S3"""
    try:
        # Check if user is authenticated
        if USE_POSTGRES:
            from questions.auth import get_current_user

            current_user = get_current_user(request, next(get_db()))
            if not current_user:
                raise HTTPException(status_code=401, detail="Authentication required")

        # Validate file type
        if not file.content_type or not file.content_type.startswith("image/"):
            raise HTTPException(status_code=400, detail="File must be an image")

        # Read image data
        image_data = await file.read()

        # Convert to WebP with quality 85
        image = Image.open(io.BytesIO(image_data))

        # Convert to RGB if necessary (for WebP compatibility)
        if image.mode in ("RGBA", "LA", "P"):
            image = image.convert("RGB")

        # Create WebP image
        webp_buffer = io.BytesIO()
        image.save(webp_buffer, format="WEBP", quality=85, optimize=True)
        webp_data = webp_buffer.getvalue()

        # Generate filename
        import uuid

        filename = f"uploaded_{uuid.uuid4().hex}.webp"

        # Upload to S3 (assuming AWS credentials are configured)
        s3_client = boto3.client("s3")
        bucket_name = "textgeneratorstatic.netwrck.com"

        try:
            s3_client.put_object(
                Bucket=bucket_name, Key=filename, Body=webp_data, ContentType="image/webp", ACL="public-read"
            )

            # Return the URL
            image_url = f"https://{bucket_name}/{filename}"
            return JSONResponse({"success": True, "url": image_url, "filename": filename})

        except Exception as e:
            logger.error(f"Error uploading to S3: {e}")
            raise HTTPException(status_code=500, detail="Failed to upload image")

    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"Error processing image upload: {e}")
        raise HTTPException(status_code=500, detail="Failed to process image")