Change model to gpt-3.5-turbo with a different prompt

app.py

@@ -168,12 +168,12 @@ def Page():
             messages.value = [*messages.value, {"role": "user", "content": " "}]
         def response(message):
             extraction_stream = client.chat.completions.create_partial(
-                model="gpt-4-turbo-2024-04-09",
+                model="gpt-3.5-turbo",
                 response_model=KnowledgeGraph,
                 messages=[
                     {
                         "role": "user",
-                        "content": f"Help me understand the following by describing it as small knowledge graph: {text_block.value}",
+                        "content": f"Help me understand the following by describing it as small knowledge graph: {text_block.value}. It is important to add variety of colors in the nodes.",
                     },
                 ],
                 temperature=0,