Update my_model/utilities/st_utils.py

my_model/utilities/st_utils.py

@@ -0,0 +1,57 @@
+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.")
+