pratham0011 commited on
Commit
b32bf68
ยท
verified ยท
1 Parent(s): 332ced6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -7,8 +7,9 @@ from PIL import Image
7
  import pdfplumber
8
  import io
9
 
 
10
  genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
11
- # load gemini model
12
  model = genai.GenerativeModel("gemini-1.5-flash")
13
 
14
  def get_gemini_response(input, image, prompt):
@@ -56,10 +57,21 @@ def input_file_setup(uploaded_file):
56
 
57
  st.set_page_config(page_title="Invoice Information Extractor", page_icon="๐Ÿ”ฎ")
58
  st.title("Invoice Information Extractor ๐Ÿ“ƒ")
59
- # st.write("Upload your invoice")
 
 
 
 
 
 
 
 
 
 
 
60
 
61
  uploaded_file = st.file_uploader("Upload an Invoice Image ๐Ÿ–ผ๏ธ or PDF ๐Ÿ“„", type=["jpg", "jpeg", "png", "pdf"])
62
- input_query = st.text_input("Ask a question", key="input")
63
 
64
  if uploaded_file is not None:
65
  if uploaded_file.type.startswith('image'):
 
7
  import pdfplumber
8
  import io
9
 
10
+ # Configure API key
11
  genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
12
+ # Load gemini model
13
  model = genai.GenerativeModel("gemini-1.5-flash")
14
 
15
  def get_gemini_response(input, image, prompt):
 
57
 
58
  st.set_page_config(page_title="Invoice Information Extractor", page_icon="๐Ÿ”ฎ")
59
  st.title("Invoice Information Extractor ๐Ÿ“ƒ")
60
+
61
+ st.write("Upload your invoice (image or PDF) and ask a question related to the invoice. Here are some example prompts:")
62
+
63
+ # Display sample prompts
64
+ st.subheader("Sample Prompts:")
65
+ sample_prompts = [
66
+ "What is the total amount?",
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")
75
 
76
  if uploaded_file is not None:
77
  if uploaded_file.type.startswith('image'):