Update app.py

app.py

@@ -5,6 +5,8 @@ import time
 import datetime
 import json
 import subprocess
+import string 
+import random 
 
 app = Flask(__name__)
 
@@ -89,6 +91,37 @@ def status():
 @app.errorhandler(404)
 def page_not_found(e):
     return send_from_directory(static_dir, '404.html'), 404
+
+
+def generate_random_text(length=10):
+    letters = string.ascii_lowercase
+    return ''.join(random.choice(letters) for _ in range(length))
+
+@app.route('/upload', methods=['GET'])
+def upload_image():
+    try:
+        # Get the URL parameter
+        url = request.args.get('url')
+
+        if not url:
+            return jsonify({'error': 'URL parameter is missing'}), 400
+
+        # Download the image
+        response = requests.get(url)
+        if response.status_code != 200:
+            return jsonify({'error': 'Failed to download image'}), 400
+
+        # Generate random text for the image name
+        random_text = generate_random_text()
+        image_name = f"{random_text}.png"
+
+        # Save the image
+        img = Image.open(BytesIO(response.content))
+        img.save(image_name, "PNG")
+
+        return jsonify({'success': f'Image uploaded as {image_name}'}), 200
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
     
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)