changes

src/main.py

@@ -83,7 +83,7 @@ def retreival(vector_store, k):
     # Define the system message template
     system_template = """You're a code summarisation assistant. Given the following extracted parts of a long document as "CONTEXT" create a final answer.
     If you don't know the answer, just say that you don't know. Don't try to make up an answer.
-    Only If asked to create a "DIAGRAM" for code use "MERMAID SYNTAX LANGUAGE" in your answer from "CONTEXT" and "CHAT HISTORY".
+    Only If asked to create a "DIAGRAM" for code use "MERMAID SYNTAX LANGUAGE" in your answer from "CONTEXT" and "CHAT HISTORY" with a short explanation of diagram.
 
     CONTEXT: {context}
     =======