Dksony8 patch 1

app.py

@@ -0,0 +1,32 @@
+pip install flask
+pip install flask-cors
+
+from flask import Flask, request, jsonify, send_file
+import os
+from realesrgan import RealESRGAN
+
+app = Flask(__name__)
+
+# Initialize Real-ESRGAN model (ensure model weights are downloaded)
+model = RealESRGAN('RealESRGAN_x4plus')
+
+@app.route('/enhance', methods=['POST'])
+def enhance_image():
+    try:
+        # Save the uploaded image
+        file = request.files['image']
+        input_path = 'input.jpg'
+        file.save(input_path)
+
+        # Enhance the image using Real-ESRGAN
+        output_path = 'output.jpg'
+        model.enhance(input_path, output_path)
+
+        # Return the enhanced image
+        return send_file(output_path, mimetype='image/jpeg')
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)

requirements.txt

@@ -7,3 +7,6 @@ Pillow
 torch>=1.7
 torchvision
 tqdm
+flask
+gunicorn
+flask-cors