Update app.py

app.py

@@ -15,20 +15,14 @@ import my_model.utilities.st_config as st_config
 
 
 
-class ImageHandler:
-    @staticmethod
-    def analyze_image(image, model, show_processed_image=False):
-        img = copy.deepcopy(image)
-        caption = model.get_caption(img)
-        image_with_boxes, detected_objects_str = model.detect_objects(img)
-        if show_processed_image:
-            st.image(image_with_boxes, use_column_width=True)
-        return caption, detected_objects_str
 
-    @staticmethod
-    def free_gpu_resources():
-        # Implementation for freeing GPU resources
-        free_gpu_resources()
+def analyze_image(image, model, show_processed_image=False):
+    img = copy.deepcopy(image)
+    caption = model.get_caption(img)
+    image_with_boxes, detected_objects_str = model.detect_objects(img)
+    if show_processed_image:
+        st.image(image_with_boxes, use_column_width=True)
+    return caption, detected_objects_str
 
 
 
@@ -42,23 +36,6 @@ class QuestionAnswering:
         st.write(detected_objects_str)
         return answer
 
-class UIComponents:
-    @staticmethod
-    def display_image_selection(sample_images):
-        cols = st.columns(len(sample_images))
-        for idx, sample_image_path in enumerate(sample_images):
-            with cols[idx]:
-                image = Image.open(sample_image_path)
-                st.image(image, use_column_width=True)
-                if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
-                    st.session_state['current_image'] = image
-                    st.session_state['qa_history'] = []
-                    st.session_state['analysis_done'] = False
-                    st.session_state['answer_in_progress'] = False
-
-    @staticmethod
-    def display_image(img):
-        st.image(img, use_column_width=True)
 
 def load_kbvqa_model(detection_model):
     """Load KBVQA Model based on the selected detection model."""
@@ -86,7 +63,18 @@ def set_model_confidence(detection_model):
 def image_qa_app(kbvqa_model):
     """Streamlit app interface for image QA."""
     sample_images = st_config.SAMPLE_IMAGES
-    UIComponents.display_image_selection(sample_images)
+
+    cols = st.columns(len(sample_images))
+    for idx, sample_image_path in enumerate(sample_images):
+        with cols[idx]:
+            image = Image.open(sample_image_path)
+            st.image(image, use_column_width=True)
+            if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
+                st.session_state['current_image'] = image
+                st.session_state['qa_history'] = []
+                st.session_state['analysis_done'] = False
+                st.session_state['answer_in_progress'] = False
+   
 
     uploaded_image = st.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
 
@@ -98,7 +86,7 @@ def image_qa_app(kbvqa_model):
 
     # Display the image if it's in the session state
     if 'current_image' in st.session_state:
-        UIComponents.display_image(st.session_state['current_image'])
+        st,image(st.session_state['current_image'], use_column_width=True)
 
     else:
         st.write("image not uploaded properly")
@@ -109,7 +97,7 @@ def image_qa_app(kbvqa_model):
 
     if st.session_state.get('current_image') and not st.session_state.get('analysis_done', False):
         if st.button('Analyze Image'):
-            caption, detected_objects_str = ImageHandler.analyze_image(st.session_state['current_image'], kbvqa_model)
+            caption, detected_objects_str = analyze_image(st.session_state['current_image'], kbvqa_model)
             st.session_state['caption'] = caption
             st.session_state['detected_objects_str'] = detected_objects_str
             st.session_state['analysis_done'] = True