Update chain.py

chain.py

@@ -89,6 +89,11 @@ def get_answer(
     logging.info(f"User question: {question}")
 
     print("File writing complete.")
+
+    wandb.log({"question": question })
+        
+    # Log training progress
+    
     
     result = chain(
         inputs={"question": question, "chat_history": chat_history},