Update app.py

app.py

@@ -1,67 +1,126 @@
-# app.py
-
-import os
-from flask import Flask, request, send_file, jsonify
-from rembg import remove
-from PIL import Image
-import io
-
-# --- Create the Flask App ---
-app = Flask(__name__)
-
-# --- Configuration ---
-# Get the API Key from an environment variable for security.
-API_KEY = os.environ.get("BG_REMOVER_API_KEY") 
-
-# --- API Endpoints ---
-
-# A simple root endpoint to check if the server is running.
-@app.route('/')
-def index():
-    return "Background Remover API is running!"
-
-# The main endpoint for removing the background.
-@app.route('/remove-bg', methods=['POST'])
-def remove_background_api():
-    # 1. --- API Key Authentication ---
-    api_key_header = request.headers.get('x-api-key')
-    if not api_key_header or api_key_header != API_KEY:
-        return jsonify({"error": "Unauthorized. Invalid or missing API Key."}), 401
-
-    # 2. --- Image Validation ---
-    if 'file' not in request.files:
-        return jsonify({"error": "No file part in the request"}), 400
-
-    file = request.files['file']
-
-    if file.filename == '':
-        return jsonify({"error": "No selected file"}), 400
-
-    # 3. --- Image Processing ---
-    if file:
-        try:
-            input_image_bytes = file.read()
-            output_image_bytes = remove(input_image_bytes)
-            
-            # 4. --- Send the Response ---
-            return send_file(
-                io.BytesIO(output_image_bytes),
-                mimetype='image/png',
-                as_attachment=True,
-                download_name='background_removed.png'
-            )
-        except Exception as e:
-            return jsonify({"error": "Failed to process image", "details": str(e)}), 500
-
-    return jsonify({"error": "An unknown error occurred"}), 500
-
-# --- Run the App ---
-if __name__ == '__main__':
-    # For local testing, ensure the environment variable is set.
-    if not API_KEY:
-        # If you're using the temporary hard-coded key method, you can ignore this error.
-        # Otherwise, this reminds you to set the variable.
-        print("WARNING: BG_REMOVER_API_KEY is not set. The API will not be secured.")
-        print("For local testing, run 'set BG_REMOVER_API_KEY=your-key' (Windows) or 'export BG_REMOVER_API_KEY=your-key' (Mac/Linux) first.")
-    
-    app.run(debug=True, port=5000)
+# app.py
+
+import os
+from flask import Flask, request, send_file, jsonify
+from rembg import remove
+from PIL import Image
+import io
+
+# --- Create the Flask App ---
+app = Flask(__name__)
+
+# --- Configuration ---
+# Get the API Key from an environment variable for security.
+API_KEY = os.environ.get("BG_REMOVER_API_KEY") 
+
+# --- API Endpoints ---
+
+# A simple root endpoint to check if the server is running.
+@app.route('/')
+def index():
+    return "Background Remover API is running!"
+
+# The main endpoint for removing the background.
+@app.route('/remove-bg', methods=['POST'])
+def remove_background_api():
+    # 1. --- API Key Authentication ---
+    api_key_header = request.headers.get('x-api-key')
+    if not api_key_header or api_key_header != API_KEY:
+        return jsonify({"error": "Unauthorized. Invalid or missing API Key."}), 401
+
+    # 2. --- Image Validation ---
+    if 'file' not in request.files:
+        return jsonify({"error": "No file part in the request"}), 400
+
+    file = request.files['file']
+
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+
+    # 3. --- Image Processing ---
+    if file:
+        try:
+            input_image_bytes = file.read()
+            output_image_bytes = remove(input_image_bytes)
+            
+            # 4. --- Send the Response ---
+            return send_file(
+                io.BytesIO(output_image_bytes),
+                mimetype='image/png',
+                as_attachment=True,
+                download_name='background_removed.png'
+            )
+        except Exception as e:
+            return jsonify({"error": "Failed to process image", "details": str(e)}), 500
+
+    return jsonify({"error": "An unknown error occurred"}), 500
+
+# --- IMAGE COMPRESSOR API ENDPOINT ---
+
+@app.route('/compress-image', methods=['POST'])
+def compress_image_api():
+    # 1. --- API Key Authentication (Bilkul pehle jaisa) ---
+    api_key_header = request.headers.get('x-api-key')
+    if not api_key_header or api_key_header != API_KEY:
+        return jsonify({"error": "Unauthorized. Invalid or missing API Key."}), 401
+
+    # 2. --- File aur Quality Level ko Check Karna ---
+    if 'file' not in request.files:
+        return jsonify({"error": "No file part in the request"}), 400
+
+    file = request.files['file']
+    
+    # User se quality level haasil karna (default 85)
+    # request.form se text data liya jaata hai
+    quality = int(request.form.get('quality', 85))
+
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+
+    # 3. --- Image Processing (Pillow Library ka istemal) ---
+    if file:
+        try:
+            # File ko memory mein bytes ki tarah parhna
+            input_image_bytes = file.read()
+            
+            # Pillow ka istemal karke image ko kholna
+            img = Image.open(io.BytesIO(input_image_bytes))
+            
+            # Ek khali "in-memory" file banana jahan hum compressed image save karenge
+            output_buffer = io.BytesIO()
+
+            # Image ko 'RGB' mode mein convert karna (JPG ke liye zaroori)
+            if img.mode in ("RGBA", "P"):
+                img = img.convert("RGB")
+
+            # Image ko buffer mein save karna, lekin is baar quality ke saath
+            # quality parameter 1 se 95 tak hota hai (95 best quality)
+            img.save(output_buffer, format='JPEG', quality=quality)
+            
+            # Compressed image ke bytes haasil karna
+            output_image_bytes = output_buffer.getvalue()
+
+            # 4. --- Send the Response ---
+            # Compressed image ko user ko wapas bhejna
+            return send_file(
+                io.BytesIO(output_image_bytes),
+                mimetype='image/jpeg',
+                as_attachment=True,
+                download_name=f'compressed_quality_{quality}.jpg'
+            )
+        except Exception as e:
+            return jsonify({"error": "Failed to process image", "details": str(e)}), 500
+
+    return jsonify({"error": "An unknown error occurred"}), 500
+
+# --- Run the App ---
+if __name__ == '__main__':
+    # For local testing, ensure the environment variable is set.
+    if not API_KEY:
+        # If you're using the temporary hard-coded key method, you can ignore this error.
+        # Otherwise, this reminds you to set the variable.
+        print("WARNING: BG_REMOVER_API_KEY is not set. The API will not be secured.")
+        print("For local testing, run 'set BG_REMOVER_API_KEY=your-key' (Windows) or 'export BG_REMOVER_API_KEY=your-key' (Mac/Linux) first.")
+    
+    app.run(debug=True, port=5000)
+