Update app.py

app.py

@@ -11,7 +11,6 @@ from my_model.captioner.image_captioning import get_caption
 from my_model.utilities import free_gpu_resources
 
 
-
 # Placeholder for undefined functions
 def load_caption_model():
     st.write("Placeholder for load_caption_model function")
@@ -29,6 +28,9 @@ def get_caption(image):
 def free_gpu_resources():
     pass
 
+# Sample images (assuming these are paths to your sample images)
+sample_images = ["path/to/sample1.jpg", "path/to/sample2.jpg", "path/to/sample3.jpg"]
+
 # Main function
 def main():
     st.sidebar.title("Navigation")
@@ -47,37 +49,21 @@ def main():
     elif selection == "Object Detection":
         run_object_detection()
 
-def display_home():
-    st.title("MultiModal Learning for Knowledge-Based Visual Question Answering")
-    st.write("Home page content goes here...")
-
-def display_dissertation_report():
-    st.title("Dissertation Report")
-    st.write("Click the link below to view the PDF.")
-    st.download_button(
-        label="Download PDF",
-        data=open("Files/Dissertation Report.pdf", "rb"),
-        file_name="example.pdf",
-        mime="application/octet-stream"
-    )
-
-def display_evaluation_results():
-    st.title("Evaluation Results")
-    st.write("This is a Place Holder until the contents are uploaded.")
-
-def display_dataset_analysis():
-    st.title("OK-VQA Dataset Analysis")
-    st.write("This is a Place Holder until the contents are uploaded.")
+# Other display functions...
 
 def run_inference():
-    st.title("Image-based Q&A App")
+    st.title("Run Inference")
+    # Image-based Q&A and Object Detection functionality
+    image_qa_and_object_detection()
+
+def image_qa_and_object_detection():
     # Image-based Q&A functionality
+    st.subheader("Image-based Q&A")
     image_qa_app()
 
-def run_object_detection():
-    st.title("Object Detection")
     # Object Detection functionality
-    # ... Implement your code for this section ...
+    st.subheader("Object Detection")
+    object_detection_app()
 
 def image_qa_app():
     # Initialize session state for storing images and their Q&A histories
@@ -89,46 +75,28 @@ def image_qa_app():
         st.session_state['images_qa_history'] = []
         st.experimental_rerun()
 
+    # Display sample images
+    st.write("Or choose from 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)
+
     # Image uploader
     uploaded_image = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg"])
-
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
-        current_image_key = uploaded_image.name  # Use image name 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")
-
-            # 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 for the current image
-                question = st.text_input("Ask a question about this image:", key=question_key)
-
-                # Get Answer button for the current image
-                if st.button('Get Answer', key=button_key):
-                    # Process the image and question
-                    answer = get_answer(image_info['image'], question)  # Implement this function
-                    image_info['qa_history'].append((question, answer))
-                    st.experimental_rerun()  # Rerun to update the display
-
-def get_answer(image, question):
-    # Implement the logic to process the image and question, and return the answer
-    return "Sample answer based on the image and question."
+        process_uploaded_image(image)
+
+def process_uploaded_image(image):
+    current_image_key = image.filename  # Use image filename as a unique key
+    # ... rest of the image processing code ...
+
+# Object Detection App
+def object_detection_app():
+    # ... Implement your code for object detection ...
+
+# Other functions...
 
 if __name__ == "__main__":
     main()