Update app.py

app.py

@@ -69,31 +69,32 @@ class AdvancedPdfChatbot:
         self.overall_chain = self.CustomChain(refinement_chain=refinement_chain, qa_chain=qa_chain)
 
     class CustomChain(Chain):
-    def __init__(self, refinement_chain, qa_chain):
-        super().__init__()
-        self.refinement_chain = refinement_chain
-        self.qa_chain = qa_chain
-
-    @property
-    def input_keys(self):
-        return ["query", "chat_history"]
-
-    @property
-    def output_keys(self):
-        return ["answer"]
-
-    def _call(self, inputs):
-        query = inputs['query']
-        chat_history = inputs.get('chat_history', [])
         
-        # Run the refinement chain to refine the query
-        refined_query = self.refinement_chain.run({'query': query, 'chat_history': chat_history})
-        
-        # Run the QA chain using the refined query and the chat history
-        response = self.qa_chain({'question': refined_query, 'chat_history': chat_history})
-        
-        # Return the answer
-        return {"answer": response['answer']}
+        def __init__(self, refinement_chain, qa_chain):
+            super().__init__()
+            self.refinement_chain = refinement_chain
+            self.qa_chain = qa_chain
+    
+        @property
+        def input_keys(self):
+            return ["query", "chat_history"]
+    
+        @property
+        def output_keys(self):
+            return ["answer"]
+    
+        def _call(self, inputs):
+            query = inputs['query']
+            chat_history = inputs.get('chat_history', [])
+            
+            # Run the refinement chain to refine the query
+            refined_query = self.refinement_chain.run({'query': query, 'chat_history': chat_history})
+            
+            # Run the QA chain using the refined query and the chat history
+            response = self.qa_chain({'question': refined_query, 'chat_history': chat_history})
+            
+            # Return the answer
+            return {"answer": response['answer']}