Update app.py

app.py

@@ -68,29 +68,30 @@ def image_qa_app(kbvqa_model):
     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['current_image'] = sample_image_path
                 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"])
-
     if uploaded_image is not None:
-        st.session_state['current_image'] = Image.open(uploaded_image)
+        st.session_state['current_image'] = uploaded_image
         st.session_state['qa_history'] = []
         st.session_state['analysis_done'] = False
         st.session_state['answer_in_progress'] = False
 
     # Display the image if it's in the session state
-    if 'current_image' in st.session_state:
-        st,image(st.session_state['current_image'], use_column_width=True)
-
+    if 'current_image' in st.session_state and st.session_state['current_image'] is not None:
+        if isinstance(st.session_state['current_image'], str):
+            # If it's a file path from sample images
+            image_to_display = Image.open(st.session_state['current_image'])
+        else:
+            # If it's an uploaded file
+            image_to_display = Image.open(st.session_state['current_image'])
+        st.image(image_to_display, use_column_width=True)
     else:
-        st.write("image not uploaded properly")
-
+        st.write("No image selected or uploaded.")