Update app.py

app.py

@@ -69,34 +69,29 @@ 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  # Use a different attribute name
+            self._refinement_chain = refinement_chain
             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)  # Use the renamed attribute
+            refined_query = self._refinement_chain.run(refinement_inputs)
             
-            # 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)  # Use the renamed attribute
+            response = self._qa_chain(qa_inputs)
             
             return {"answer": response['answer']}
 
@@ -158,4 +153,4 @@ with gr.Blocks() as demo:
     msg.submit(respond, inputs=[msg, chatbot_interface], outputs=[msg, chatbot_interface])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()