Upload 3 files

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,84 @@
+from fastapi import FastAPI, UploadFile, File
+from PIL import Image
+import io
+import os
+from typing import List, Optional
+
+app = FastAPI()
+
+# Supported output formats and their respective PIL modes (if applicable)
+supported_formats = {
+    "png": "PNG",
+    "jpg": "JPEG",
+    "jpeg": "JPEG",
+    "webp": "WEBP",
+    "gif": "GIF",  # Note: GIF conversion may lose animation if the input isn't GIF
+    "bmp": "BMP",
+    "tiff": "TIFF",
+    "pdf": None,  # PDF conversion handled differently
+}
+
+
+
+async def convert_image(image: UploadFile, output_format: str, resize: Optional[List[int]] = None):
+
+    try:
+        img_bytes = await image.read()  # Read image bytes
+        img = Image.open(io.BytesIO(img_bytes))
+
+        if resize: #  Resize if requested
+            img = img.resize(tuple(resize))
+
+
+        output_buffer = io.BytesIO()
+
+        if output_format.lower() == "pdf":
+            img.save(output_buffer, "PDF", resolution=300.0) #Higher resolution PDF
+            content_type = "application/pdf"
+            filename = f"{image.filename.split('.')[0]}.pdf"
+
+        elif output_format.lower() in supported_formats :
+
+            mode = supported_formats[output_format.lower()]
+
+            # Convert image mode if necessary (e.g., for converting to grayscale)
+            # Example usage: to convert a PNG to a grayscale JPEG
+            # if mode:
+            #     if img.mode != 'L': #L is for grayscale
+            #         img = img.convert('L')  
+
+            img.save(output_buffer, mode) 
+            content_type = f"image/{output_format.lower()}"  # Correct content type
+            filename = f"{image.filename.split('.')[0]}.{output_format.lower()}"
+        else:
+            return {"error": "Unsupported output format"}
+
+
+        return {
+                "filename": filename,
+                "file": output_buffer.getvalue(),  # Return bytes
+                "content_type": content_type,
+            }
+
+
+
+    except Exception as e: # Handle errors like invalid image format
+        print(f"Error converting image: {e}")
+        return {"error": f"Error converting image: {e}"}
+
+
+@app.post("/convert")
+async def convert_image_endpoint(
+    image: UploadFile = File(...), 
+    output_format: str = "png", 
+    resize: Optional[List[int]] = None, #Example: ?resize=200&resize=300
+):
+
+    return await convert_image(image, output_format, resize) 
+
+
+
+
+@app.get("/")
+async def root():
+    return {"message": "Advanced Image Converter"}

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+pillow