Update myapp.py

myapp.py

@@ -1,31 +1,38 @@
-from flask import Flask, request, jsonify
+from flask import Flask, jsonify, request, send_file
 from flask_cors import CORS
 from PIL import Image
+import io
 
 # Initialize the Flask app
 myapp = Flask(__name__)
-
-# Enable CORS for the app
-CORS(myapp)
+CORS(myapp)  # Enable CORS if needed
 
 @myapp.route('/')
-def index():
-    return "Welcome to the AI Model Upscale!"
+def home():
+    return "Welcome to the Image Upscaler!"  # Basic home response
+
+@myapp.route('/upscale', methods=['POST'])
+def upscale_image():
+    if 'image' not in request.files:
+        return jsonify({"error": "No image provided"}), 400
+
+    input_image = request.files['image'].read()  # Read the uploaded image
+    width = int(request.form.get('width', 2))  # Default upscale factor
+    height = int(request.form.get('height', 2))
 
-@myapp.route('/resize', methods=['POST'])
-def resize_image():
-    data = request.json
-    input_path = data.get('input_path')
-    output_path = data.get('output_path')
-    width = data.get('width')
-    height = data.get('height')
+    # Open the image using PIL
+    img = Image.open(io.BytesIO(input_image))
 
-    # Open an image file and resize it
-    with Image.open(input_path) as img:
-        resized_img = img.resize((width, height), Image.LANCZOS)
-        resized_img.save(output_path)
+    # Resize the image to upscale (you can adjust the method as needed)
+    upscaled_img = img.resize((img.width * width, img.height * height), Image.LANCZOS)
+    
+    # Save to bytes
+    img_byte_arr = io.BytesIO()
+    upscaled_img.save(img_byte_arr, format='PNG')
+    img_byte_arr.seek(0)
 
-    return jsonify({'message': 'Image resized successfully', 'output_path': output_path})
+    return send_file(img_byte_arr, mimetype='image/png')
 
+# Add this block to make sure your app runs when called
 if __name__ == "__main__":
-    myapp.run(host='0.0.0.0', port=5000)
+    myapp.run(host='0.0.0.0', port=7860)  # Run directly if needed for testing