Update main.py

main.py

@@ -6,6 +6,7 @@ from lang import G4F
 from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
+from ImageCreator import generate_image_prodia
 
 app = FastAPI()
 
@@ -55,3 +56,15 @@ async def searches(request: Request):
     response = agent_chain.run(input=prompt)
     return response
 
+@app.route("/", methods=["POST"])
+def generate_image():
+    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")
+
+    response = generate_image_prodia(prompt, model, sampler, seed, neg)
+    return jsonify({"image": response})
+