Update app.py

app.py

@@ -48,6 +48,8 @@ def image_qa_app(kbvqa):
         st.session_state['qa_history'] = []
     if 'analysis_done' not in st.session_state:
         st.session_state['analysis_done'] = False
+    if 'answer_in_progress' not in st.session_state:
+        st.session_state['answer_in_progress'] = False
 
     # Display sample images as clickable thumbnails
     st.write("Choose from sample images:")
@@ -60,6 +62,7 @@ def image_qa_app(kbvqa):
                 st.session_state['current_image'] = image
                 st.session_state['qa_history'] = []
                 st.session_state['analysis_done'] = False
+                st.session_state['answer_in_progress'] = False
 
     # Image uploader
     uploaded_image = st.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
@@ -68,6 +71,7 @@ def image_qa_app(kbvqa):
         st.session_state['current_image'] = image
         st.session_state['qa_history'] = []
         st.session_state['analysis_done'] = False
+        st.session_state['answer_in_progress'] = False
 
     # Analyze Image button
     if st.session_state.get('current_image') and not st.session_state['analysis_done']:
@@ -75,21 +79,27 @@ def image_qa_app(kbvqa):
             # Perform analysis on the image
             analyze_image(st.session_state['current_image'], kbvqa)
             st.session_state['analysis_done'] = True
+            st.session_state['processed_image'] = copy.deepcopy(st.session_state['current_image'])
 
     # Display the current image (unaltered)
     if st.session_state.get('current_image'):
         st.image(st.session_state['current_image'], caption='Uploaded Image.', use_column_width=True)
 
     # Get Answer button
-    if st.session_state['analysis_done']:
+    if st.session_state['analysis_done'] and not st.session_state['answer_in_progress']:
         question = st.text_input("Ask a question about this image:")
         if st.button('Get Answer'):
-            answer = answer_question(st.session_state['current_image'], question, model=kbvqa)
+            st.session_state['answer_in_progress'] = True
+            answer = answer_question(st.session_state['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")
+    # Display all Q&A
+    for q, a in st.session_state['qa_history']:
+        st.text(f"Q: {q}\nA: {a}\n")
+
+    # Reset the answer_in_progress flag after displaying the answer
+    if st.session_state['answer_in_progress']:
+        st.session_state['answer_in_progress'] = False
 
 def run_inference():
     st.title("Run Inference")