Update app.py

app.py

@@ -6,6 +6,7 @@ import google.generativeai as genai
 from PIL import Image
 import pdfplumber
 import io
+import streamlit_pills as sp
 
 # Configure API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -67,8 +68,7 @@ sample_prompts = [
     'What are the Customter details?' , 
     'What is the information about the Products?'
 ]
-for prompt in sample_prompts:
-    st.write(f"- {prompt}")
+sp.pills(sample_prompts, label="Sample Prompts", default_index=0)
 
 uploaded_file = st.file_uploader("Upload an Invoice Image 🖼️ or PDF 📄", type=["jpg", "jpeg", "png", "pdf"])
 input_query = st.text_input("Ask a question about the invoice", key="input")