Update app.py

app.py

@@ -69,10 +69,12 @@ def image_qa_app(kbvqa):
                 image_data['detected_objects_str'] = detected_objects_str
                 image_data['analysis_done'] = True
 
+        # Initialize qa_history for each image
+        qa_history = image_data.get('qa_history', [])
+
         if image_data['analysis_done']:
-            question = st.text_input("Ask a question about this image:", key=f'question_{image_key}')
+            question = st.text_input(f"Ask a question about this image ({image_key[-11:]}):", 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]:
                     answer = answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
                     qa_history.append((question, answer))
@@ -80,9 +82,9 @@ def image_qa_app(kbvqa):
                 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 each image
+        for q, a in qa_history:
+            st.text(f"Q: {q}\nA: {a}\n")
 
 
 def process_new_image(image_key, image, kbvqa):