Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -66,8 +66,8 @@ class InferenceRunner(StateManager):
         if uploaded_image is not None:
             self.process_new_image(uploaded_image.name, Image.open(uploaded_image))
 
-    def display_image_and_analysis(self, image_key, image_data):
-        nested_col21, nested_col22 = st.columns([0.65, 0.35])
+    def display_image_and_analysis(self, image_key, image_data, nested_col21, nested_col22):
+        
         image_for_display = self.resize_image(image_data['image'], 600)
         nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
         self.handle_analysis_button(image_key, image_data, nested_col22)
@@ -86,6 +86,7 @@ class InferenceRunner(StateManager):
             self.display_question_answering_interface(image_key, image_data, nested_col22)
 
     def display_question_answering_interface(self, image_key, image_data, nested_col22):
+        
         sample_questions = config.SAMPLE_QUESTIONS.get(image_key, [])
         selected_question = nested_col22.selectbox("Select a sample question or type your own:", ["Custom question..."] + sample_questions, key=f'sample_question_{image_key}')
         custom_question = nested_col22.text_input("Or ask your own question:", key=f'custom_question_{image_key}')
@@ -106,8 +107,11 @@ class InferenceRunner(StateManager):
         self.display_session_state()
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
-                with st.container():
-                    self.display_image_and_analysis(image_key, image_data)
+                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:]}')
+                    self.display_image_and_analysis(image_key, image_data, nested_col21, nested_col22)
                     self.handle_question_answering(image_key, image_data, nested_col22)