Updated prompt.

app.py

@@ -24,7 +24,7 @@ daig_prompt = st.text_area("Prompt for diagrams")
 
 daig_template = PromptTemplate(
     input_variables = ['topic', 'name'],
-    template = 'Using the Python diagrams library generate the code for {topic} and name the diagram {name}. Just return the code. No explaination of the code needed'
+    template = 'Using the Python diagrams library generate the code for {topic} and name the diagram {name}. The Diagram class in the generated code should have the following parameters set: output_format as jpg and show=False. Just return the code. No explaination of the code needed'
 )
 
 daig_memory = ConversationBufferMemory(input_key='topic', memory_key='daig_history')