Update app.py

app.py

@@ -73,26 +73,32 @@ def image_qa_app(kbvqa):
     # Image uploader
     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)
+        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['qa_history'] = []
 
-    # Display the current image
-    if st.session_state['current_image'] is not None:
+    # Display the current image (unaltered)
+    if st.session_state.get('current_image') is not None:
         st.image(st.session_state['current_image'], caption='Uploaded Image.', use_column_width=True)
 
-        # Question input
+        # Question input and processing
         question = st.text_input("Ask a question about this image:")
-
-        # Get Answer button
         if st.button('Get Answer'):
-            # Process the question
-            answer = answer_question(st.session_state['current_image'], question, model=kbvqa)
-            free_gpu_resources()
-            st.session_state['qa_history'].append((question, answer))
-
-        # Display all Q&A
-        for q, a in st.session_state['qa_history']:
-            st.text(f"Q: {q}\nA: {a}\n")
+            # 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)
+                answer = answer_question(processed_image, question, model=kbvqa)
+                st.session_state['qa_history'].append((question, 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
 
             
 # Main function