app updated

app.py

@@ -83,6 +83,9 @@ def main():
 
         # OCR by API Call of AWS Textract via Post Method
         if input_method == "Upload Image":
+            st.markdown(
+                "# OCR (Optical Character Recognition) - computer vision technology to locate letters/numbers from image."
+            )
             st.success("Running textract!")
             url = "https://2tsig211e0.execute-api.us-east-1.amazonaws.com/my_textract"
             payload = {"image": image_base64}
@@ -99,6 +102,7 @@ def main():
                 st.table(df)
 
             # Using an expander to hide the table
+            st.success("Bounding boxes drawn!")
             with st.expander("Show/Hide Annotation"):
                 try:
                     image = Image.open(image)
@@ -107,7 +111,6 @@ def main():
                     image_with_boxes = draw_bounding_boxes_for_textract(
                         image.copy(), result_dict
                     )
-                    st.success("Bounding boxes drawn.")
 
                     # Display annotated image
                     st.image(
@@ -121,6 +124,9 @@ def main():
 
         if api_key:
             # Make API call
+            st.markdown(
+                "# Gemini (Generative AI) - read the image content in a general form"
+            )
             st.success("Running Gemini!")
             with st.spinner("Wait for it..."):
                 response = call_gemini_api(image_base64, api_key)
@@ -136,6 +142,9 @@ def main():
                 with st.spinner("Wait for it..."):
                     st.write(text_from_response)
 
+                st.markdown(
+                    "# Information Retrieval - use Gemini to extract the values for the keys required by the stakeholders"
+                )
                 with st.spinner("Processing csv to download..."):
                     try:
                         keys = ["First Name", "Last Name", "Policy Number"]
@@ -204,7 +213,9 @@ def main():
 
     # YOLO
     if image is not None:
-        st.sidebar.success("Check the following box to run YOLO algorithm if desired!")
+        st.sidebar.success(
+            "Check the following box to run YOLO algorithm if desired (only do this if the task at hand is an object detection task)!"
+        )
         use_yolo = st.sidebar.checkbox("Use YOLO!", value=False)
         if use_yolo:
             yolo_option = st.selectbox(
@@ -222,9 +233,9 @@ def main():
                 "object-detection", model="eagle0504/detr-finetuned-balloon-v2"
             )
         else:
-            yolo_pipe = pipeline("object-detection", model="hustvl/yolos-small")
+            yolo_pipe = None
 
-        if use_yolo:
+        if yolo_pipe is not None:
             # Process image with YOLO
             image = Image.open(image)
             with st.spinner("Wait for it..."):