Update my_model/utilities/st_utils.py

my_model/utilities/st_utils.py

@@ -1,6 +1,8 @@
+import pandas as pd
 import streamlit as st
 from my_model.tabs.run_inference import run_inference
 
+
 class UIManager:
     def __init__(self):
         self.tabs = {
@@ -40,7 +42,6 @@ class UIManager:
         st.write("This is a Place Holder until the contents are uploaded.")
 
     def display_run_inference(self):
-        
         run_inference()
 
     def display_dissertation_report(self):
@@ -67,61 +68,48 @@ class UIManager:
 
 
 
+class StateManager:
+    def __init__(self):
+        self.initialize_state()
+
+    def initialize_state(self):
+        if 'images_data' not in st.session_state:
+            st.session_state['images_data'] = {}
+        if 'model_settings' not in st.session_state:
+            st.session_state['model_settings'] = {'detection_model': None, 'confidence_level': None}
+        if 'kbvqa' not in st.session_state:
+            st.session_state['kbvqa'] = None
+        if 'selected_method' not in st.session_state:
+            st.session_state['selected_method'] = None
+
+    def update_model_settings(self, detection_model=None, confidence_level=None, selected_method=None):
+        if detection_model is not None:
+            st.session_state['model_settings']['detection_model'] = detection_model
+        if confidence_level is not None:
+            st.session_state['model_settings']['confidence_level'] = confidence_level
+        if selected_method is not None:
+            st.session_state['selected_method'] = selected_method
+
+    def check_settings_changed(self, current_detection_model, current_confidence_level):
+        return (st.session_state['model_settings']['detection_model'] != current_detection_model or
+                st.session_state['model_settings']['confidence_level'] != current_confidence_level)
+
+    def display_model_settings(self):
+        st.write("### Current Model Settings:")
+        st.table(pd.DataFrame(st.session_state['model_settings'], index=[0]))
+
+    def display_session_state(self):
+        st.write("### Current Session State:")
+        data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items()]
+        df = pd.DataFrame(data)
+        st.table(df)
+
+
+
+
+
 
 
 
 
-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.")