Upload folder using huggingface_hub

app/main.py

@@ -1,21 +1,19 @@
 from fastapi import FastAPI
-
-# testers routers
+# Import the new router
+from app.routers.image import jpgcompressor 
+# Import the existing 'testers' routers
 from app.routers.testers import random_number_generator
 from app.routers.testers import help
 from app.routers.testers import calculator
 
-# add your more module here
-
 app = FastAPI()
 
 # Include all routers
+app.include_router(jpgcompressor.router) # <-- Include the new image router
 app.include_router(random_number_generator.router)
 app.include_router(help.router)
 app.include_router(calculator.router)
 
-# add more router here
-
 @app.get("/")
 def greet_json():
     return {"message": "Go to /docs to see all available endpoints!", "link" : "https://sameerbanchhor-multi-use-api.hf.space/docs"}

app/routers/image/jpgcompressor.py

@@ -0,0 +1,60 @@
+from fastapi import APIRouter, UploadFile, File, Form, HTTPException
+from fastapi.responses import StreamingResponse
+from PIL import Image
+from io import BytesIO
+
+# Create the router for image processing endpoints
+router = APIRouter(
+    prefix="/image",
+    tags=["Image Processing"]
+)
+
+@router.post("/compress_jpg")
+async def compress_jpg(
+    file: UploadFile = File(..., description="The JPG image file to compress."),
+    quality: int = Form(75, description="JPEG compression quality (1 to 100). Default is 75.")
+):
+    """
+    Compresses a JPEG image based on the specified quality setting.
+    The compressed image is returned as a file.
+    """
+    
+    if file.content_type != "image/jpeg":
+        raise HTTPException(
+            status_code=400, 
+            detail="Invalid file type. Only JPEG files are supported for this endpoint."
+        )
+
+    if not 1 <= quality <= 100:
+        raise HTTPException(
+            status_code=400, 
+            detail="Quality must be an integer between 1 and 100."
+        )
+
+    # Read the image file content
+    content = await file.read()
+    
+    # Use BytesIO to handle the image in memory
+    input_buffer = BytesIO(content)
+    output_buffer = BytesIO()
+    
+    try:
+        # Open the image using Pillow
+        img = Image.open(input_buffer)
+        
+        # Save the image to the output buffer with specified quality
+        img.save(output_buffer, "JPEG", quality=quality)
+        
+        # Move the buffer cursor to the start so it can be read
+        output_buffer.seek(0)
+        
+        # Return the compressed image as a file stream
+        return StreamingResponse(
+            output_buffer, 
+            media_type="image/jpeg", 
+            headers={"Content-Disposition": f"attachment; filename=compressed_{file.filename}"}
+        )
+
+    except Exception as e:
+        # Handle potential errors during image processing
+        raise HTTPException(status_code=500, detail=f"Image processing error: {e}")

requirements.txt

@@ -1,3 +1,4 @@
 fastapi 
 uvicorn
-requests
+requests
+pillow