Update app.py
Browse files
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 |
-
|
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")
|