Update app.py

app.py

@@ -35,7 +35,7 @@ user_input = st.text_area("Describe your architecture:",
 if st.button("Generate Diagram"):
     if user_input:
         # Send the user's input to the API
-        payload = {"question": user_input}
+        payload = {"question":"make sure to import from diagrams import Diagram" + user_input + " architecture"}
         output = query(payload)
         
         # Check and display the Python code from the API response