Update main.py

main.py

@@ -38,7 +38,7 @@ agent_chain = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCR
 
 
 @app.get("/")
-def gello():
+def hello():
     return "Hello! My name is Linlada."
 
 @app.post('/linlada')
@@ -68,8 +68,21 @@ async def generate_image(request: Request):
     response = generate_image_prodia(prompt, model, sampler, seed, neg)
     return jsonify({"image": response})
 
+class User(BaseModel):
+    prompt: str,
+    model: str,
+    sampler: str,
+    seed: int,
+    neg: str or None
+
 @app.route("/test", methods=["POST"])
-async def test(request: Request):
-    data = await request.json()
+async def test(request: User):
+    data = {
+        'prompt': request.prompt,
+        'model': request.model,
+        'sampler': request.sampler,
+        'seed': request.seed,
+        'neg': request.neg
+    }
     return data