Update main.py

main.py

@@ -3,7 +3,7 @@ from langchain.memory import ConversationBufferMemory
 from langchain.utilities import GoogleSearchAPIWrapper
 from langchain.agents import AgentType, initialize_agent, Tool
 from lang import G4F
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from ImageCreator import generate_image_prodia
@@ -56,17 +56,17 @@ async def searches(request: Request):
     response = agent_chain.run(input=prompt)
     return response
 
-# @app.post("/imagen")
-# async def generate_image(request: Request):
-#     data = await request.json()
-#     prompt = data['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})
+@app.post("/imagen")
+async def generate_image(request: Request):
+    data = await request.json()
+    prompt = data['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})
 
 class User(BaseModel):
     prompt: str