Update app.py

app.py

@@ -310,18 +310,37 @@ def rag_workflow(query):
     
     doc_references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(retrieved_doc_chunks)])
     code_references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(retrieved_code_chunks)])
+
     
-    print(f"Context for the query:\n{doc_context}\n")
+    print("Document Chunks:\n")
+    print("\n\n".join(["="*80 + "\n" + doc_chunk for doc_chunk, _ in retrieved_doc_chunks]))
+    print("\nDocument References:\n")
+    print(doc_references)
+
+    print("\n" + "="*80 + "\n")  # Separator between doc and code
+
+    print("Code Chunks:\n")
+    print("\n\n".join(["="*80 + "\n" + code_chunk for code_chunk, _ in retrieved_code_chunks]))
+    print("\nCode References:\n")
+    print(code_references)
+
+
+   # print(f"Context for the query:\n{doc_context}\n")
     
-    print(f"References for the query:\n{references}\n")
+   # print(f"References for the query:\n{references}\n")
     
-    prompt = f"""You are an expert python developer. Provide a clear and consice answer based only on the information in the retrieved context.
-                    The retrieved context contains source code and documenation of an api library. 
-                    If no related Information is found from the context to answer the query, reply that you do not know.
+    prompt = f"""You are an expert python developer. You are assisting in generating code for users who wants to make use of "kadi-apy", an API library. 
+                    "Doc-context:" provides you with information how to use this API library by givnig code examples and code documentation.
+                    "Code-context:" provides you information of API methods and classes from the "kadi-apy" library. 
+                    Based on the retrieved contexts answer the query.  
+                    If no related Information is found from the contexts to answer the query, reply that you do not know.
 
-            Context:
+            Doc-context:
             {doc_context}
 
+            Code-context:
+            {code_context}
+            
             Query: 
             {query}
     """