Update app.py

app.py

@@ -91,58 +91,62 @@ def main():
         # Convert the resized image to base64
         image_base64 = convert_image_to_base64(resized_image)
 
+        col1, col2 = st.columns(2)
+
         # 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}
-            result_dict = post_request_and_parse_response(url, payload)
-            output_data = extract_line_items(result_dict)
-            df = pd.DataFrame(output_data)
-
-            # Using an expander to hide the json
-            with st.expander("Show/Hide Raw Json"):
-                st.write(result_dict)
-
-            # Using an expander to hide the table
-            with st.expander("Show/Hide Table"):
-                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)
-
-                    # Draw bounding boxes and labels
-                    image_with_boxes = draw_bounding_boxes_for_textract(
-                        image.copy(), result_dict
-                    )
-
-                    # Display annotated image
-                    st.image(
-                        image_with_boxes,
-                        caption="Annotated Image",
-                        use_column_width=True,
-                    )
-
-                except:
-                    st.warning("Check textract output!")
+            with col1:
+                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}
+                result_dict = post_request_and_parse_response(url, payload)
+                output_data = extract_line_items(result_dict)
+                df = pd.DataFrame(output_data)
+    
+                # Using an expander to hide the json
+                with st.expander("Show/Hide Raw Json"):
+                    st.write(result_dict)
+    
+                # Using an expander to hide the table
+                with st.expander("Show/Hide Table"):
+                    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)
+    
+                        # Draw bounding boxes and labels
+                        image_with_boxes = draw_bounding_boxes_for_textract(
+                            image.copy(), result_dict
+                        )
+    
+                        # Display annotated image
+                        st.image(
+                            image_with_boxes,
+                            caption="Annotated Image",
+                            use_column_width=True,
+                        )
+    
+                    except:
+                        st.warning("Check textract output!")
 
         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)
-
-            with st.expander("Raw output from Gemini"):
-                st.write(response)
+            with col2:
+                # 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)
+    
+                with st.expander("Raw output from Gemini"):
+                    st.write(response)
 
             # Display the response
             try: