Update app.py

app.py

@@ -1,6 +1,7 @@
 from flask import Flask, request, jsonify, send_from_directory
 from PIL import Image
 from io import BytesIO
+from prodiapy import Prodia
 import requests
 import os
 import psutil
@@ -126,6 +127,108 @@ def upload_image():
         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",
+    "1452e7a5-d6e2-4600-9641-1c2debde397a",
+    "f4b18c3c-ea4d-4b18-be47-f5ad29d70936",
+    "688659c2-b2e9-4524-8a91-1c72735ec068",
+    "aa64f14e-18d8-44df-91cc-6d4e20051ca3"
+]
+
+# Load styles from style.json file
+with open("style.json", "r") as style_file:
+    styleList = json.load(style_file)
+
+def getRandomApiKey():
+    # Implement your logic to get a random API key here
+    return random.choice(apiKeys)
+
+def getRandomSeed():
+    return random.randint(1, 18446744073709552000)
+
+def getAvailableStyles():
+    return ', '.join([style["name"] for style in styleList])
+
+prodia = Prodia(getRandomApiKey())
+
+@app.route('/styles', methods=['GET'])
+def get_styles():
+    with open("style.json", "r") as style_file:
+        styles = json.load(style_file)
+
+    return jsonify({"styles": [style["name"] for style in styles]})
+
+@app.route('/generate', methods=['POST'])
+async def generate_image():
+    try:
+        data = request.json
+
+        prompt = data.get('prompt', '')
+        userStyle = data.get('style')
+        seed = int(data.get('seed', getRandomSeed()))
+        guidance_scale = int(data.get('guidance_scale', 0))
+
+        if not userStyle:
+            return jsonify({"status": "error", "error": "Style is required. Available styles: " + getAvailableStyles()}), 400
+
+        selectedStyle = next((style for style in styleList if style["name"].lower() == userStyle.lower()), None)
+
+        if not selectedStyle:
+            return jsonify({"status": "error", "error": "Invalid style. Available styles: " + getAvailableStyles()}), 400
+
+        if guidance_scale and (guidance_scale < 1 or guidance_scale > 100):
+            return jsonify({"status": "error", "error": "guidance_scale must be an integer between 1 and 100."}), 400
+
+        job = prodia.sdxl.generate(            
+            prompt=selectedStyle["prompt"].replace('{prompt}', data.get('prompt', '')),
+            model="sd_xl_base_1.0.safetensors [be9edd61]",
+            negative_prompt=selectedStyle["negative_prompt"] + ", duplicate",
+            sampler="DPM++ 2M Karras",
+            cfg_scale=selectedStyle.get('cfg_scale', 7),
+            steps=selectedStyle.get('steps', 20),
+            height=1024,
+            width=1024)
+        wait = prodia.wait(job)
+        url = wait.image_url
+
+        # Discord Bot setup
+        discord_webhook_url = "https://discord.com/api/webhooks/1134794312945778779/DPaF4YGE3l7VMeXI56ft4yAS-5a1cu-C0bgrtU9RtrLgTXHCPFfy17AAB9ABb_IcQ9Pa"
+
+        # Send the generated image URL through the webhook
+        image_name = f"invite_1080035826051854356_best_bot_ever.png"
+        img = Image.open(BytesIO(requests.get(url).content))
+        img.save(image_name, "PNG")
+
+        files = {'file': open(image_name, 'rb')}
+        webhook_response = requests.post(discord_webhook_url, files=files)
+
+        # Print the response for debugging
+        print(webhook_response.text)
+
+        # Check if the request was successful
+        if webhook_response.status_code == 200:
+            discord_cdn_url = webhook_response.json().get('attachments', [{}])[0].get('url')
+
+            # Remove the temporary image file
+            os.remove(image_name)
+
+            # Return the success response with the generated image URL
+            return jsonify({
+                'success': 'true',
+                'url': discord_cdn_url
+            }), 200
+        else:
+            # If the request to the webhook fails, return an error response
+            return jsonify({"status": "error", "error": "Failed to send image through webhook"}), 500
+
+    except Exception as e:
+        print('Error:', str(e))
+        return jsonify({"status": "error", "error": "Internal Server Error"}), 500
     
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)