Update app.py

app.py

@@ -1,84 +1,57 @@
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import StreamingResponse
 from PIL import Image
 import io
-import os
-from typing import List, Optional
 
-app = FastAPI()
+app = FastAPI(title="Advanced Image Converter")
 
-# 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
-}
+ALLOWED_OUTPUT_FORMATS = ["jpeg", "png", "gif", "bmp", "tiff", "webp"]
 
-
-
-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
+@app.post("/convert_image/")
+async def convert_image(
+    file: UploadFile = File(...),
+    output_format: str = "jpeg",
+    quality: int = 95  # For JPEG/WebP, adjust compression quality (0-100)
 ):
+    """
+    Converts an uploaded image to the specified format.
+
+    - Supports common image formats as input.
+    - Output formats: JPEG, PNG, GIF, BMP, TIFF, WebP.
+    - For JPEG and WebP, you can adjust the `quality` (0-100).
+    """
+    if output_format.lower() not in ALLOWED_OUTPUT_FORMATS:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Invalid output format. Allowed formats: {', '.join(ALLOWED_OUTPUT_FORMATS)}",
+        )
+
+    if output_format.lower() in ["jpeg", "webp"] and not (0 <= quality <= 100):
+        raise HTTPException(
+            status_code=400, detail="Quality must be between 0 and 100 for JPEG and WebP."
+        )
 
-    return await convert_image(image, output_format, resize) 
-
-
-
+    try:
+        image = Image.open(io.BytesIO(await file.read()))
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid image file.") from e
 
-@app.get("/")
-async def root():
-    return {"message": "Advanced Image Converter"}
+    output_io = io.BytesIO()
+    try:
+        save_kwargs = {}
+        if output_format.lower() in ["jpeg", "webp"]:
+            save_kwargs["quality"] = quality
+        image.save(output_io, format=output_format.upper(), **save_kwargs)
+    except KeyError:
+        raise HTTPException(
+            status_code=400, detail=f"Conversion to {output_format} failed."
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Image processing error: {e}")
+
+    output_io.seek(0)
+    media_type = f"image/{output_format.lower()}"
+    if output_format.lower() == "tiff":
+        media_type = "image/tiff"  # Special case for TIFF
+
+    return StreamingResponse(output_io, media_type=media_type)