Update app.py

app.py

@@ -76,30 +76,23 @@ def image_qa_app(kbvqa):
         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']:
+    if st.session_state.get('current_image') and not st.session_state.get('analysis_done', False):
         if st.button('Analyze Image'):
-            # Perform analysis on the image
             caption, detected_objects_str = analyze_image(st.session_state['current_image'], kbvqa)
+            st.session_state['caption'] = caption
+            st.session_state['detected_objects_str'] = detected_objects_str
             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'] and not st.session_state['answer_in_progress']:
+    if st.session_state.get('analysis_done', False):
         question = st.text_input("Ask a question about this image:")
         if st.button('Get Answer'):
-            st.session_state['answer_in_progress'] = True
-            answer = answer_question(st.session_state['processed_image'], question, caption, detected_objects_str, model=kbvqa)
+            answer = answer_question(st.session_state['current_image'], question, st.session_state.get('caption', ''), st.session_state.get('detected_objects_str', ''), 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.get('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']: