Update app.py

app.py

@@ -1,6 +1,7 @@
 from flask import Flask, request, jsonify, send_from_directory
 import shutil
 import os
+import requests
 
 app = Flask(__name__)
 
@@ -14,9 +15,8 @@ def get_image(filename):
 # Create folders to store uploaded images
 UPLOAD_FOLDER = "uploads"
 IMAGE_FOLDER = os.path.join(UPLOAD_FOLDER, "images")
-VIDEO_FOLDER = os.path.join(UPLOAD_FOLDER, "videos")
 
-for folder in [UPLOAD_FOLDER, IMAGE_FOLDER, VIDEO_FOLDER]:
+for folder in [UPLOAD_FOLDER, IMAGE_FOLDER]:
     os.makedirs(folder, exist_ok=True)
 
 # Create a dictionary to store available images
@@ -25,63 +25,40 @@ available_images = {}
 @app.route("/upload", methods=["POST"])
 def upload_image():
     url = request.form.get("url")
-    file = request.files["file"]
 
-    allowed_formats = [".png", ".jpg"]
-    file_format = os.path.splitext(file.filename)[1]
-
-    if file_format not in allowed_formats:
-        return jsonify({"message": "Unsupported file format"}), 400
-
-    # Save the uploaded file
-    file.save(os.path.join(UPLOAD_FOLDER, file.filename))
-
-    # Determine the folder based on the file format
-    folder = "images" if file_format in [".png", ".jpg"] else "videos"
-
-    # Move the file to the appropriate folder
-    shutil.move(
-        os.path.join(UPLOAD_FOLDER, file.filename),
-        os.path.join(UPLOAD_FOLDER, folder, file.filename)
-    )
+    if not url:
+        return jsonify({"message": "URL parameter is missing"}), 400
+
+    # Check if the URL is valid and get the image content
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+    except requests.exceptions.RequestException as e:
+        return jsonify({"message": f"Error fetching image from URL: {str(e)}"}), 400
+
+    # Check if the response content is an image
+    content_type = response.headers.get('content-type')
+    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
+
+    # Save the image
+    filename = f"uploaded_image.{file_extension}"
+    image_path = os.path.join(IMAGE_FOLDER, filename)
+    with open(image_path, 'wb') as f:
+        f.write(response.content)
 
     # Update available images dictionary
-    available_images[file.filename] = {"url": url, "name": file.filename}
-
-    return jsonify({"message": "File uploaded successfully"})
-
-
-@app.route("/get-all-images", methods=["GET"])
-def get_all_images():
-    password = request.args.get("password")
-    from_folder = request.args.get("from_folder")
-
-    if password != "Kastg@123":
-        return jsonify({"message": "Unauthorized"}), 401
-
-    folder_path = os.path.join(UPLOAD_FOLDER, from_folder)
-    if not os.path.exists(folder_path):
-        return jsonify({"message": "Folder not found"}), 404
-
-    images = os.listdir(folder_path)
-    return jsonify({"images": images})
-
-
-@app.route("/delete-image", methods=["DELETE"])
-def delete_image():
-    image_name = request.args.get("image_name")
-    password = request.args.get("password")
-
-    if password != "Kastg@123":
-        return jsonify({"message": "Unauthorized"}), 401
-
-    image_path = os.path.join(UPLOAD_FOLDER, image_name)
-    if not os.path.exists(image_path):
-        return jsonify({"message": "Image not found"}), 404
+    available_images[filename] = {"url": url, "name": filename}
 
-    os.remove(image_path)
-    available_images.pop(image_name, None)
-    return jsonify({"message": "Image deleted successfully"})
+    return jsonify({"message": "File uploaded successfully", "image_url": f"/images/{filename}"})
 
 
 if __name__ == "__main__":