Update app.py

app.py

@@ -1,7 +1,8 @@
 from flask import Flask, request, jsonify, send_from_directory
-import shutil
 import os
 import requests
+from PIL import Image
+from io import BytesIO
 
 app = Flask(__name__)
 
@@ -41,19 +42,16 @@ def upload_image():
     if 'image' not in content_type:
         return jsonify({"message": "The provided URL does not lead to an image"}), 400
 
-    # Determine the file format
-    if 'jpeg' in content_type:
-        file_extension = 'jpg'
-    elif 'png' in content_type:
-        file_extension = 'png'
-    else:
-        return jsonify({"message": "Unsupported image format"}), 400
+    # Open the image using PIL
+    try:
+        img = Image.open(BytesIO(response.content))
+    except Exception as e:
+        return jsonify({"message": f"Error processing image: {str(e)}"}), 400
 
     # Save the image
-    filename = f"uploaded_image.{file_extension}"
+    filename = os.path.basename(url)
     image_path = os.path.join(IMAGE_FOLDER, filename)
-    with open(image_path, 'wb') as f:
-        f.write(response.content)
+    img.save(image_path)
 
     # Update available images dictionary
     available_images[filename] = {"url": url, "name": filename}
@@ -62,4 +60,4 @@ def upload_image():
 
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=True)