Update app.py

app.py

@@ -12,7 +12,7 @@ def get_diagram_code(prompt, diagram_type, api_key):
 
         # For GPT-4 and other chat models, use the v1/chat/completions endpoint
         response = openai.ChatCompletion.create(
-            model="gpt-4",  # Make sure you're using the right model
+            model="gpt-4o",  # Make sure you're using the right model
             messages=[
                 {"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}"}