Update app.py

app.py

@@ -3,7 +3,7 @@
 import os
 from flask import Flask, request, send_file, jsonify
 from rembg import remove
-from PIL import Image
+from PIL import Image, ImageEnhance, ImageFilter
 import io
 
 # --- Create the Flask App ---
@@ -113,6 +113,66 @@ def compress_image_api():
 
     return jsonify({"error": "An unknown error occurred"}), 500
 
+# --- AI IMAGE ENHANCER API ENDPOINT ---
+
+@app.route('/enhance-image', methods=['POST'])
+def enhance_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 Check Karna ---
+    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 (Pillow Library ka Jadoo) ---
+    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))
+            
+            # Enhancement Step 1: Contrast ko behtar karna
+            # 1.0 original contrast hai. 1.2 ka matlab hai 20% zyada contrast.
+            enhancer_contrast = ImageEnhance.Contrast(img)
+            img = enhancer_contrast.enhance(1.2)
+            
+            # Enhancement Step 2: Sharpness ko behtar karna
+            # 1.0 original sharpness hai. 1.4 ka matlab hai 40% zyada sharp.
+            enhancer_sharpness = ImageEnhance.Sharpness(img)
+            img = enhancer_sharpness.enhance(1.4)
+            
+            # Ek khali "in-memory" file banana
+            output_buffer = io.BytesIO()
+            
+            # Enhanced image ko buffer mein save karna
+            # PNG format mein save karein taake transparency (agar ho) barqarar rahe
+            img.save(output_buffer, format='PNG')
+            
+            # Enhanced image ke bytes haasil karna
+            output_image_bytes = output_buffer.getvalue()
+
+            # 4. --- Send the Response ---
+            # Enhanced image ko user ko wapas bhejna
+            return send_file(
+                io.BytesIO(output_image_bytes),
+                mimetype='image/png',
+                as_attachment=True,
+                download_name='enhanced_image.png'
+            )
+        except Exception as e:
+            return jsonify({"error": "Failed to enhance 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.