Update app.py

app.py

@@ -44,61 +44,46 @@ def image_qa_and_object_detection():
     object_detection_app()
 
 def image_qa_app():
-    # Initialize session state for storing images and their Q&A histories
-    if 'images_qa_history' not in st.session_state:
-        st.session_state['images_qa_history'] = []
-
-    # Button to clear all data
-    if st.button('Clear All'):
-        st.session_state['images_qa_history'] = []
-        st.experimental_rerun()
-
-    # Image uploader
-    uploaded_image = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg"])
-
-    # Display sample images
-    st.write("Or choose from sample images:")
+    # Initialize session state for storing the current image and its Q&A history
+    if 'current_image' not in st.session_state:
+        st.session_state['current_image'] = None
+    if 'qa_history' not in st.session_state:
+        st.session_state['qa_history'] = []
+
+    # Display sample images as clickable thumbnails
+    st.write("Choose from sample images:")
+    cols = st.columns(len(sample_images))
     for idx, sample_image_path in enumerate(sample_images):
-        if st.button(f"Use Sample Image {idx+1}", key=f"sample_{idx}"):
-            uploaded_image = Image.open(sample_image_path)
-            process_uploaded_image(uploaded_image)
+        with cols[idx]:
+            image = Image.open(sample_image_path)
+            if st.image(image, use_column_width=True):
+                st.session_state['current_image'] = image
+                st.session_state['qa_history'] = []
 
+    # Image uploader
+    uploaded_image = st.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
     if uploaded_image is not None:
-        image = Image.open(uploaded_image)
-        process_uploaded_image(image)
-
-def process_uploaded_image(image):
-    current_image_key = image.filename  # Use image filename as a unique key
-    # Check if the image is already in the history
-    if not any(info['image_key'] == current_image_key for info in st.session_state['images_qa_history']):
-        st.session_state['images_qa_history'].append({
-            'image_key': current_image_key,
-            'image': image,
-            'qa_history': []
-        })
-
-    # Display all images and their Q&A histories
-    for image_info in st.session_state['images_qa_history']:
-        st.image(image_info['image'], caption='Uploaded Image.', use_column_width=True)
-        for q, a in image_info['qa_history']:
-            st.text(f"Q: {q}\nA: {a}\n")
+        st.session_state['current_image'] = Image.open(uploaded_image)
+        st.session_state['qa_history'] = []
+
+    # Display the current image
+    if st.session_state['current_image'] is not None:
+        st.image(st.session_state['current_image'], caption='Uploaded Image.', use_column_width=True)
 
-        # If the current image is being processed
-        if image_info['image_key'] == current_image_key:
-            # Unique keys for each widget
-            question_key = f"question_{current_image_key}"
-            button_key = f"button_{current_image_key}"
+        # Question input
+        question = st.text_input("Ask a question about this image:")
 
-            # Question input for the current image
-            question = st.text_input("Ask a question about this image:", key=question_key)
+        # Get Answer button
+        if st.button('Get Answer'):
+            # Process the question
+            answer = answer_question(st.session_state['current_image'], question)
+            st.session_state['qa_history'].append((question, answer))
 
-            # Get Answer button for the current image
-            if st.button('Get Answer', key=button_key):
-                # Process the image and question
-                answer = answer_question(image_info['image'], question, None, None)  # Implement this function
-                image_info['qa_history'].append((question, answer))
-                st.experimental_rerun()  # Rerun to update the display
+        # Display all Q&A
+        for q, a in st.session_state['qa_history']:
+            st.text(f"Q: {q}\nA: {a}\n")
 
+            
 # Object Detection App
 def object_detection_app():
     # ... Implement your code for object detection ...