Update main.py

main.py

@@ -68,3 +68,14 @@ def generate_image():
     response = generate_image_prodia(prompt, model, sampler, seed, neg)
     return jsonify({"image": response})
 
+@app.route("/test", methods=["POST"])
+def test():
+    data = request.get_json()
+    prompt = data.get("prompt")
+    model = data.get("model")
+    sampler = data.get("sampler")
+    seed = int(data.get("seed"))
+    neg = data.get("neg")
+
+    res = [prompt, \n, model, \n, sampler, \n, seed, \n, neg]
+