Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -30,7 +30,6 @@ class InferenceRunner(StateManager):
         """
         
         super().__init__()
-       # self.initialize_state()
 
 
     def answer_question(self, caption, detected_objects_str, question):
@@ -41,8 +40,6 @@ class InferenceRunner(StateManager):
             caption (str): The caption generated for the image.
             detected_objects_str (str): String representation of objects detected in the image.
             question (str): The user's question about the image.
-  
-
         Returns:
             str: The generated answer to the question.
         """
@@ -53,15 +50,8 @@ class InferenceRunner(StateManager):
         return answer, prompt_length
 
 
-    def image_qa_app(self):
-        """
-        Main application interface for image-based question answering. It handles displaying
-        of sample images, uploading of new images, and facilitates the QA process.
-        """
-        
-        
+    def display_sample_images(self):
         # Display sample images as clickable thumbnails
-   
         self.col1.write("Choose from sample images:")
         cols = self.col1.columns(len(config.SAMPLE_IMAGES))
         for idx, sample_image_path in enumerate(config.SAMPLE_IMAGES):
@@ -72,13 +62,20 @@ class InferenceRunner(StateManager):
                 if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image)
 
+    def image_qa_app(self):
+        """
+        Main application interface for image-based question answering. It handles displaying
+        of sample images, uploading of new images, and facilitates the QA process.
+        """
+        
+        self.display_sample_images()
         # Image uploader
         uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
         if uploaded_image is not None:
             self.process_new_image(uploaded_image.name, Image.open(uploaded_image))
 
         # Display and interact with each uploaded/selected image
-        self.display_session_state()
+        
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
                 
@@ -86,7 +83,7 @@ class InferenceRunner(StateManager):
                     nested_col21, nested_col22 = st.columns([0.65, 0.35])
                     image_for_display = self.resize_image(image_data['image'], 600)
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
-                    nested_col21.write(image_data['analysis_done'] , self.settings_changed , self.confidance_change)
+                    
                     if not image_data['analysis_done'] or self.settings_changed or self.confidance_change:  # if not done analysis before or even done but settings changed, then we need to analyze again
                      
                         nested_col22.text("Please click 'Analyze Image'..")