pratham0011 commited on
Commit
437210c
ยท
verified ยท
1 Parent(s): b32bf68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,6 +6,7 @@ import google.generativeai as genai
6
  from PIL import Image
7
  import pdfplumber
8
  import io
 
9
 
10
  # Configure API key
11
  genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -67,8 +68,7 @@ sample_prompts = [
67
  'What are the Customter details?' ,
68
  'What is the information about the Products?'
69
  ]
70
- for prompt in sample_prompts:
71
- st.write(f"- {prompt}")
72
 
73
  uploaded_file = st.file_uploader("Upload an Invoice Image ๐Ÿ–ผ๏ธ or PDF ๐Ÿ“„", type=["jpg", "jpeg", "png", "pdf"])
74
  input_query = st.text_input("Ask a question about the invoice", key="input")
 
6
  from PIL import Image
7
  import pdfplumber
8
  import io
9
+ import streamlit_pills as sp
10
 
11
  # Configure API key
12
  genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
68
  'What are the Customter details?' ,
69
  'What is the information about the Products?'
70
  ]
71
+ sp.pills(sample_prompts, label="Sample Prompts", default_index=0)
 
72
 
73
  uploaded_file = st.file_uploader("Upload an Invoice Image ๐Ÿ–ผ๏ธ or PDF ๐Ÿ“„", type=["jpg", "jpeg", "png", "pdf"])
74
  input_query = st.text_input("Ask a question about the invoice", key="input")