Update app.py

app.py

@@ -70,34 +70,21 @@ def image_qa_app(kbvqa):
                 image_data['analysis_done'] = True
 
         if image_data['analysis_done']:
-            # Initialize qa_history
-            qa_history = []
-        
-            # Check if the question has already been asked
-            current_image_key = st.session_state.get('current_image_key')
-            question = st.text_input("Ask a question about this image:")
-            if st.button('Get Answer'):
-                # Ensure that the image data exists in the session state
-                if current_image_key not in st.session_state['images_data']:
-                    st.session_state['images_data'][current_image_key] = {'qa_history': []}
-        
-                qa_history = st.session_state['images_data'][current_image_key].get('qa_history', [])
+            question = st.text_input("Ask a question about this image:", key=f'question_{image_key}')
+            if st.button('Get Answer', key=f'answer_{image_key}'):
+                qa_history = image_data.get('qa_history', [])
                 if question not in [q for q, _ in qa_history]:
-                    # Assuming caption and detected_objects_str are already set
-                    caption = st.session_state.get('caption', '')
-                    detected_objects_str = st.session_state.get('detected_objects_str', '')
-                    answer = answer_question(caption, detected_objects_str, question, kbvqa)
+                    answer = answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
                     qa_history.append((question, answer))
-                    st.session_state['images_data'][current_image_key]['qa_history'] = qa_history
+                    image_data['qa_history'] = qa_history
                 else:
                     st.info("This question has already been asked.")
 
-    # Display Q&A history
-    for q, a in qa_history:
-        st.text(f"Q: {q}\nA: {a}\n")
-    # Display Q&A history
-    for q, a in qa_history:
-        st.text(f"Q: {q}\nA: {a}\n")
+            # Display Q&A history
+            for q, a in qa_history:
+                st.text(f"Q: {q}\nA: {a}\n")
+
+
 def process_new_image(image_key, image, kbvqa):
     """Process a new image and update the session state."""
     if image_key not in st.session_state['images_data']: