Upload folder using huggingface_hub

.gitignore

@@ -1,2 +1,5 @@
 # prompts file
-prompt.txt
+prompt.txt
+
+#cache
+__pycache__

app/main.py

@@ -1,19 +1,41 @@
 from fastapi import FastAPI
-# Import the new router
-from app.routers.image import jpgcompressor 
-# Import the existing 'testers' routers
+from fastapi.responses import HTMLResponse
+
+# ==========================
+# 📌 Image Module
+# ==========================
+from app.routers.image import jpgcompressor
+
+
+# ==========================
+# 🧪 Test Utilities
+# ==========================
 from app.routers.testers import random_number_generator
 from app.routers.testers import help
 from app.routers.testers import calculator
 
+
+# ==========================
+# 🚀 App Initialization
+# ==========================
 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)
 
-@app.get("/")
+# ==========================
+# 🔗 Router Registration
+# ==========================
+app.include_router(jpgcompressor.router)              # Image compression handler
+app.include_router(random_number_generator.router)    # Random number tester
+app.include_router(help.router)                       # Helper/test info
+app.include_router(calculator.router)                 # Mini calculator service
+
+
+# ==========================
+# 🌐 Root Endpoint
+# ==========================
+@app.get("/", response_class=HTMLResponse)
 def greet_json():
-    return {"message": "Go to /docs to see all available endpoints!", "link" : "https://sameerbanchhor-multi-use-api.hf.space/docs"}
+    return """
+    <h2>Go to the Swagger docs 👇</h2>
+    <a href="/docs">Click here for API Docs</a>
+    """

app/routers/image/jpgcompressor.py

@@ -3,58 +3,74 @@ from fastapi.responses import StreamingResponse
 from PIL import Image
 from io import BytesIO
 
-# Create the router for image processing endpoints
+
+# ==========================
+# 🖼️ Image Processing Router
+# ==========================
 router = APIRouter(
     prefix="/image",
     tags=["Image Processing"]
 )
 
+
+# ==========================
+# 📦 JPG Compression Endpoint
+# ==========================
 @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.
+    🔧 Compresses a JPEG image at a specified quality and returns the optimized file.
     """
-    
+
+    # --------------------------
+    # 🛑 Validation Checks
+    # --------------------------
     if file.content_type != "image/jpeg":
         raise HTTPException(
-            status_code=400, 
+            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, 
+            status_code=400,
             detail="Quality must be an integer between 1 and 100."
         )
 
-    # Read the image file content
+    # --------------------------
+    # 📥 Read Image Bytes
+    # --------------------------
     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
+        # --------------------------
+        # 🧪 Pillow Compression
+        # --------------------------
         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 as Stream
+        # --------------------------
         return StreamingResponse(
-            output_buffer, 
-            media_type="image/jpeg", 
-            headers={"Content-Disposition": f"attachment; filename=compressed_{file.filename}"}
+            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}")
+        # --------------------------
+        # 💥 Error Handling
+        # --------------------------
+        raise HTTPException(
+            status_code=500,
+            detail=f"Image processing error: {e}"
+        )