Update app.py

app.py

@@ -69,34 +69,34 @@ class AdvancedPdfChatbot:
         
         self.overall_chain = self.CustomChain(refinement_chain=refinement_chain, qa_chain=qa_chain)
 
-    class CustomChain(Chain):
+     class CustomChain(Chain):
         def __init__(self, refinement_chain, qa_chain):
             """Initialize refinement and QA chains as instance attributes."""
             super().__init__()
-            self.refinement_chain = refinement_chain
-            self.qa_chain = qa_chain
-    
+            self._refinement_chain = refinement_chain  # Use a different attribute name
+            self._qa_chain = qa_chain
+        
         @property
         def input_keys(self):
             """Define the input keys that this chain expects."""
             return ["query", "chat_history"]
-    
+        
         @property
         def output_keys(self):
             """Define the output keys that this chain returns."""
             return ["answer"]
-    
+        
         def _call(self, inputs):
             query = inputs['query']
             chat_history = inputs.get('chat_history', [])
             
             # Run the refinement chain to refine the query
             refinement_inputs = {'query': query, 'chat_history': chat_history}
-            refined_query = self.refinement_chain.run(refinement_inputs)
+            refined_query = self._refinement_chain.run(refinement_inputs)  # Use the renamed attribute
             
             # Run the QA chain using the refined query and the chat history
             qa_inputs = {"question": refined_query, "chat_history": chat_history}
-            response = self.qa_chain(qa_inputs)
+            response = self._qa_chain(qa_inputs)  # Use the renamed attribute
             
             return {"answer": response['answer']}