add content arg

app.py

@@ -35,8 +35,8 @@ def generate_response(query, diagram_type):
     response = client.chat.complete(
         model=model_name,
         messages=[
-            SystemMessage(system_prompt),
-            UserMessage(user_prompt),
+            SystemMessage(content=system_prompt),
+            UserMessage(content=user_prompt),
         ],
         max_tokens=1000
     )