Update my_model/state_manager.py

my_model/state_manager.py

@@ -282,7 +282,7 @@ class StateManager:
         return caption, detected_objects_str, image_with_boxes
 
     
-    def add_to_qa_history(self, image_key, question, answer):
+    def add_to_qa_history(self, image_key, question, answer, prompt_length):
         """
         Adds a question-answer pair to the QA history of a specific image, to be used as hitory tracker.
     
@@ -292,7 +292,7 @@ class StateManager:
             answer (str): The answer generated by the KBVQA model.
         """
         if image_key in st.session_state['images_data']:
-            st.session_state['images_data'][image_key]['qa_history'].append((question, answer))
+            st.session_state['images_data'][image_key]['qa_history'].append((question, answer, prompt_length))
 
     
     def get_images_data(self):