doc download added

app.py

@@ -104,11 +104,17 @@ def main():
                     image = Image.open(image)
 
                     # Draw bounding boxes and labels
-                    image_with_boxes = draw_bounding_boxes_for_textract(image.copy(), result_dict)
+                    image_with_boxes = draw_bounding_boxes_for_textract(
+                        image.copy(), result_dict
+                    )
                     st.success("Bounding boxes drawn.")
-        
+
                     # Display annotated image
-                    st.image(image_with_boxes, caption="Annotated Image", use_column_width=True)
+                    st.image(
+                        image_with_boxes,
+                        caption="Annotated Image",
+                        use_column_width=True,
+                    )
 
                 except:
                     st.warning("Check textract output!")
@@ -130,6 +136,42 @@ def main():
                 with st.spinner("Wait for it..."):
                     st.write(text_from_response)
 
+                with st.spinner("Processing csv to download..."):
+                    try:
+                        keys = ["First Name", "Last Name", "Policy Number"]
+                        values = []
+                        for k in keys:
+                            updated_text_from_response = call_gemini_api(
+                                image_base64,
+                                api_key,
+                                prompt=f"What is {k} in this document?",
+                            )
+                            value = updated_text_from_response["candidates"][0][
+                                "content"
+                            ]["parts"][0]["text"]
+                            values.append(value)
+
+                        # Create dataframe to download
+                        sample_payload_output = pd.DataFrame(
+                            {"Key": keys, "Values": values}
+                        )
+
+                        # Convert DataFrame to CSV
+                        csv = sample_payload_output.to_csv(index=False)
+
+                        # To convert to CSV and encode for the download link
+                        csv = sample_payload_output.to_csv(index=False).encode("utf-8")
+
+                        # Streamlit download button
+                        st.download_button(
+                            label="Download data as CSV",
+                            data=csv,
+                            file_name="data.csv",
+                            mime="text/csv",
+                        )
+                    except:
+                        st.warning("Please verify document source.")
+
                 # Text input for the question
                 input_prompt = st.text_input(
                     "Type your question here:",