Update app.py

app.py

@@ -18,7 +18,7 @@ def get_diagram_code(prompt, diagram_type, api_key):
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": f"Generate a {diagram_type} diagram in Mermaid.js syntax based on the following prompt: {prompt}"}
             ],
-            max_tokens=500
+            max_tokens=2048
         )
 
         return response['choices'][0]['message']['content'].strip()