Update app.py

app.py

@@ -70,14 +70,28 @@ def image_qa_app(kbvqa):
                 image_data['analysis_done'] = True
 
         if image_data['analysis_done']:
-            question = st.text_input(f"Ask a question about this image ({image_key}):", key=f'question_{image_key}')
-            if st.button('Get Answer', key=f'answer_{image_key}'):
-                answer = answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
-                image_data['qa_history'].append((question, answer))
-
-            for q, a in image_data['qa_history']:
-                st.text(f"Q: {q}\nA: {a}\n")
-
+            # 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', [])
+                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)
+                    qa_history.append((question, answer))
+                    st.session_state['images_data'][current_image_key]['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")
 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']:
@@ -140,7 +154,7 @@ def main():
 
         
     st.sidebar.title("Navigation")
-    selection = st.sidebar.radio("Go to", ["Home", "Dataset Analysis", "Evaluation Results", "Finetuning and Evaluation Results", "Run Inference", "Dissertation Report"])
+    selection = st.sidebar.radio("Go to", ["Home", "Dataset Analysis", "Finetuning and Evaluation Results", "Run Inference", "Dissertation Report"])
     st.sidebar.write("More Pages will follow .. ")
 
     if selection == "Home":