Update main.py

main.py

@@ -148,9 +148,15 @@ def image_detail(detail_id: int = Path(None, description="The ID of 1.model id a
 class Test(BaseModel):
     prompt: str
     model: str = 'gpt-3.5-turbo'
-    neg: str = False
+    neg: str = True
 
 @app.post("/test")
 def test(request: Test):
-        return {'data': f'Prompt is {request.prompt} Model is {request.model} Neg is {request.neg}'}
+            if neg == True:
+                chat = "hello world"
+            elif neg == "True":
+                chat = "Hi there"
+            else:
+                chat = "Hi world"
+        return {'data': f'Prompt is {request.prompt} Model is {request.model} Neg is {request.neg} Chat is {chat}'}