checkbox added

app.py

@@ -142,69 +142,85 @@ 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"
+                use_retrieval_tech = st.sidebar.checkbox(
+                    "Check the box if you want to see a sample retrieved information to download (only use this if this is a document-based task)!",
+                    value=False,
                 )
-                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? Just answer the question directly using key words, don't say a complete sentence.",
+                if use_retrieval_tech:
+                    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"]
+                            values = []
+                            for k in keys:
+                                updated_text_from_response = call_gemini_api(
+                                    image_base64,
+                                    api_key,
+                                    prompt=f"""
+                                        What is {k} in this document? Just answer the question directly with a word or two, don't say a complete sentence.
+
+                                        If there is any special characters, rewrite it w
+                                    """,
+                                )
+                                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}
                             )
-                            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}
-                        )
+                            # Display table
+                            with st.expander("Inspect table (before download)"):
+                                st.table(sample_payload_output)
 
-                        # Display table
-                        with st.expander("Inspect table (before download)"):
-                            st.table(sample_payload_output)
+                            # Convert DataFrame to CSV
+                            csv = sample_payload_output.to_csv(index=False)
 
-                        # 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")
+                            # 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.")
+                            # 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:",
+                use_gemini_to_chat = st.sidebar.checkbox(
+                    "Check the box if you want to chat with Gemini (do this if you want Gemini to answwer your questions)!",
+                    value=False,
                 )
-
-                # Display the entered question
-                if input_prompt:
-                    updated_text_from_response = call_gemini_api(
-                        image_base64, api_key, prompt=input_prompt
+                if use_gemini_to_chat:
+                    # Text input for the question
+                    input_prompt = st.text_input(
+                        "Type your question here:",
                     )
 
-                    if updated_text_from_response is not None:
-                        # Do something with the text
-                        updated_ans = updated_text_from_response["candidates"][0][
-                            "content"
-                        ]["parts"][0]["text"]
-                        with st.spinner("Wait for it..."):
-                            st.write(f"Gemini: {updated_ans}")
-                    else:
-                        st.warning("Check gemini's API.")
+                    # Display the entered question
+                    if input_prompt:
+                        updated_text_from_response = call_gemini_api(
+                            image_base64, api_key, prompt=input_prompt
+                        )
+
+                        if updated_text_from_response is not None:
+                            # Do something with the text
+                            updated_ans = updated_text_from_response["candidates"][0][
+                                "content"
+                            ]["parts"][0]["text"]
+                            with st.spinner("Wait for it..."):
+                                st.write(f"Gemini: {updated_ans}")
+                        else:
+                            st.warning("Check gemini's API.")
 
             except:
                 st.write("No response from API.")