Update app.py

app.py

@@ -75,8 +75,9 @@ def image_qa_app(kbvqa):
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
         st.session_state['current_image'] = image
-        st.session_state['processed_image'] = copy.deepcopy(image)  # Create a copy for processing
+        st.session_state['processed_image'] = copy.deepcopy(image)
         st.session_state['qa_history'] = []
+        print("Image uploaded and processed image set.")
 
     # Display the current image (unaltered)
     if st.session_state.get('current_image') is not None:
@@ -85,19 +86,17 @@ def image_qa_app(kbvqa):
         # Question input and processing
         question = st.text_input("Ask a question about this image:")
         if st.button('Get Answer'):
-            # Use the processed image for object detection and other processing
             processed_image = st.session_state.get('processed_image')
             if processed_image:
-                # Perform object detection or other processing on processed_image
-                # For example: processed_image = perform_object_detection(processed_image)
+                print("Processing image for question:", question)
                 answer = answer_question(processed_image, question, model=kbvqa)
                 st.session_state['qa_history'].append((question, answer))
+                print("Answer generated:", answer)
 
             # Display all Q&A
             for q, a in st.session_state['qa_history']:
                 st.text(f"Q: {q}\nA: {a}\n")
 
-            # Update the processed image in session state
             st.session_state['processed_image'] = processed_image