Update app.py

app.py

@@ -97,40 +97,6 @@ def status():
 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
-
-
-
-
 apiKeys = [
     "2021e94a-1385-4ddc-905b-c050cfb5af32",
     "0bfe0e6d-6bf9-4984-ab07-3a9410a551ad",
@@ -163,6 +129,47 @@ def get_styles():
 
     return jsonify({"styles": [style["name"] for style in styles]})
 
+    
+@app.route('/upload/image', 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
+
+        image_name = f"image.png"
+
+        # Save the image
+        img = Image.open(BytesIO(response.content))
+        img.save(image_name, "PNG")
+
+        # Send the image to Discord
+        discord_webhook_url = "https://discord.com/api/webhooks/1217109788656406588/sh0LG9VH5wmxSWP8OBwfHxfbbMHleUX6eQ8-xULIEo5m4IASfNm7jCNrZFZZweKaNGTM"
+        files = {'file': open(image_name, 'rb')}
+        webhook_response = requests.post(discord_webhook_url, files=files)
+
+        # Get the uploaded image URL from Discord CDN
+        discord_cdn_url = webhook_response.json().get('attachments', [{}])[0].get('url')
+
+        # Delete the temporary image file
+        os.remove(image_name)
+
+        return jsonify({
+            'success': f'Image uploaded and sent to Discord',
+            'discord_cdn_url': discord_cdn_url
+        }), 200
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+
+
 @app.route('/generate', methods=['POST'])
 async def generate_image():
     try:
@@ -219,7 +226,7 @@ async def generate_image():
 
             # Return the success response with the generated image URL
             return jsonify({
-                'success': 'true',
+                'status': 'success',
                 'url': discord_cdn_url
             }), 200
         else: