| import streamlit as st | |
| class StateManager: | |
| def __init__(self): | |
| self.reset_state() | |
| def reset_state(self): | |
| """Resets the state to its initial values.""" | |
| self.current_image = None | |
| self.qa_history = [] | |
| self.analysis_done = False | |
| self.answer_in_progress = False | |
| self.caption = "" | |
| self.detected_objects_str = "" | |
| def set_current_image(self, image): | |
| """Sets the current image and resets relevant state variables.""" | |
| try: | |
| self.current_image = image | |
| self.qa_history = [] | |
| self.analysis_done = False | |
| self.answer_in_progress = False | |
| self.caption = "" | |
| self.detected_objects_str = "" | |
| except Exception as e: | |
| print(f"Error setting current image: {e}") | |
| def add_to_qa_history(self, question, answer): | |
| """Adds a question-answer pair to the history.""" | |
| if question and answer: | |
| self.qa_history.append((question, answer)) | |
| else: | |
| print("Invalid question or answer. Cannot add to history.") | |
| def set_analysis_done(self, status=True): | |
| """Sets the analysis status.""" | |
| self.analysis_done = status | |
| def set_answer_in_progress(self, status=True): | |
| """Sets the answer in progress status.""" | |
| self.answer_in_progress = status | |
| def set_caption(self, caption): | |
| """Sets the image caption.""" | |
| if caption: | |
| self.caption = caption | |
| else: | |
| print("Invalid caption. Cannot set caption.") | |
| def set_detected_objects_str(self, detected_objects_str): | |
| """Sets the detected objects string.""" | |
| if detected_objects_str: | |
| self.detected_objects_str = detected_objects_str | |
| else: | |
| print("Invalid detected objects string. Cannot set detected objects.") | |