Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -55,21 +55,23 @@ class InferenceRunner(StateManager):
                     if not image_data['analysis_done']:
                         nested_col22.text("Please click 'Analyze Image'..")
                         with nested_col22:
-                            if st.button('Analyze Image', key=f'analyze_{image_key}'):
-                                st.session_state['loading_in_progress'] = True
+                            if st.button('Analyze Image', key=f'analyze_{image_key}', on_click=self.disable_widgets, disabled=self.is_widget_disabled):
+
                                 caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'], kbvqa)
                                 self.update_image_data(image_key, caption, detected_objects_str, True)
-                            st.session_state['loading_in_progress'] = False
+                                st.session_state['loading_in_progress'] = False
+
         
                     # Initialize qa_history for each image
                     qa_history = image_data.get('qa_history', [])
         
                     if image_data['analysis_done']:
                         question = nested_col22.text_input(f"Ask a question about this image ({image_key[-11:]}):", key=f'question_{image_key}')
-                        if nested_col22.button('Get Answer', key=f'answer_{image_key}'):
-                            st.session_state['loading_in_progress'] = True
+                        if nested_col22.button('Get Answer', key=f'answer_{image_key}', on_click=self.disable_widgets, disabled=self.is_widget_disabled):
+
                             if question not in [q for q, _ in qa_history]:
                                 answer = self.answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
+                                st.session_state['loading_in_progress'] = False
                                 self.add_to_qa_history(image_key, question, answer)
                             else: nested_col22.warning("This questions has already been answered.")
                                 
@@ -140,6 +142,7 @@ class InferenceRunner(StateManager):
 
         
         if self.is_model_loaded():
+            st.session_state['loading_in_progress'] = False
             free_gpu_resources()
             self.image_qa_app(self.get_model())