Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -61,7 +61,7 @@ class InferenceRunner(StateManager):
         
         
         # Display sample images as clickable thumbnails
-        st.write("D")
+   
         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):
@@ -81,40 +81,37 @@ class InferenceRunner(StateManager):
         self.display_session_state()
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
-                st.write("E")
+                
                 with st.container():
                     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:]}')
-                    st.write(image_data['analysis_done'] , self.settings_changed , self.confidance_change)
+                    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
-                        st.write("F")
+                     
                         nested_col22.text("Please click 'Analyze Image'..")
                         free_gpu_resources()
                         with nested_col22:
-                            st.write("G")
+                        
                             analyze_button_key = f'analyze_{image_key}_{st.session_state.detection_model}_{st.session_state.confidence_level}'  # unique key for each click
-                            st.write(analyze_button_key)
-                            st.write("H")
+                            
                             if st.button('Analyze Image', key=analyze_button_key, on_click=self.disable_widgets, disabled=self.is_widget_disabled):
-                                st.text("AAAAAAAAAAAAAAAAAAAAA")
                                 caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'])
-                                st.text("BBBBBBBBBBBBBBBBBBBB")
-                                st.write(detected_objects_str)
+                              
                                 self.update_image_data(image_key, caption, detected_objects_str, True)
                             st.session_state['loading_in_progress'] = False
                             free_gpu_resources()
-                            st.write("II")
+                            
         
                     # Initialize qa_history for each image
                     qa_history = image_data.get('qa_history', [])
-                    st.write("J")
+                 
                     if image_data['analysis_done']:
-                        st.write("K")
+                    
                         free_gpu_resources()
                         if self.confidance_change:
-                            st.write("L")
-                            nested_col22.warning("Confidence level changed, please click analyze again.")
+                        
+                            nested_col22.warning("If you change the Confidence level, please click analyze again.")
                             
                         st.session_state['loading_in_progress'] = False
                         sample_questions = config.SAMPLE_QUESTIONS.get(image_key, [])
@@ -130,15 +127,12 @@ class InferenceRunner(StateManager):
                         # Use the selected sample question or the custom question
                         question = custom_question if selected_question == "Custom question..." else selected_question
                         
-                     #   if not question:
-                     #       nested_col22.warning("Please select or enter a question.")
-                     #   else:
                         if question in [q for q, _, _ in qa_history] and not self.settings_changed and not self.confidance_change:
                             nested_col22.warning("This question has already been answered.")
                         else:
                             if nested_col22.button('Get Answer', key=f'answer_{image_key}', disabled=self.is_widget_disabled):
                                 free_gpu_resources()
-                                st.write("M") 
+                           
                                 answer, prompt_length = self.answer_question(image_data['caption'], image_data['detected_objects_str'], question)
                                 st.session_state['loading_in_progress'] = False
                                 self.add_to_qa_history(image_key, question, answer, prompt_length)
@@ -147,7 +141,7 @@ class InferenceRunner(StateManager):
                     for num, (q, a, p) in enumerate(qa_history):
                         nested_col22.text(f"Q{num+1}: {q}\nA{num+1}: {a}\nPrompt Length: {p}\n")
                         free_gpu_resources()
-                    st.write("N")    
+                     
                     
 
         
@@ -159,20 +153,20 @@ class InferenceRunner(StateManager):
         """
         
         self.set_up_widgets()
-        st.write("A")
+  
         load_fine_tuned_model = False
         fine_tuned_model_already_loaded = False
         reload_detection_model = False
         force_reload_full_model = False
         
-        #st.session_state['settings_changed'] = self.has_state_changed()
+     
         if self.is_model_loaded and self.settings_changed:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
             self.update_prev_state()
-            st.write("B")
+         
            
-        st.session_state.button_label = "Reload Model" if self.is_model_loaded and self.settings_changed else "Load Model"
-        st.write("C")
+        st.session_state.button_label = "Reload Model" if self.is_model_loaded and st.session_state.kvbqa.detection_model != st.session_state['detection_model'] else "Load Model"
+      
         with self.col1:
             if st.session_state.method == "Fine-Tuned Model":
                 with st.container():