Update app.py
Browse files
app.py
CHANGED
@@ -54,19 +54,19 @@ def input_file_setup(uploaded_file):
|
|
54 |
else:
|
55 |
raise FileNotFoundError("File not found")
|
56 |
|
57 |
-
st.set_page_config(page_title="Invoice Extractor", page_icon="🔮")
|
58 |
-
st.title("Invoice Extractor")
|
59 |
st.write("Upload your invoice📃")
|
60 |
|
61 |
input_query = st.text_input("Ask a question", key="input")
|
62 |
-
uploaded_file = st.file_uploader("Upload an
|
63 |
|
64 |
if uploaded_file is not None:
|
65 |
if uploaded_file.type.startswith('image'):
|
66 |
image = Image.open(uploaded_file)
|
67 |
-
st.image(image, caption="
|
68 |
elif uploaded_file.type.startswith('application/pdf'):
|
69 |
-
st.write("PDF uploaded.
|
70 |
|
71 |
submit = st.button("Submit")
|
72 |
|
|
|
54 |
else:
|
55 |
raise FileNotFoundError("File not found")
|
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 |
input_query = st.text_input("Ask a question", key="input")
|
62 |
+
uploaded_file = st.file_uploader("Upload an Image 🖼️ or PDF 📄", type=["jpg", "jpeg", "png", "pdf"])
|
63 |
|
64 |
if uploaded_file is not None:
|
65 |
if uploaded_file.type.startswith('image'):
|
66 |
image = Image.open(uploaded_file)
|
67 |
+
st.image(image, caption="Image uploaded.", use_column_width=True)
|
68 |
elif uploaded_file.type.startswith('application/pdf'):
|
69 |
+
st.write("PDF uploaded.")
|
70 |
|
71 |
submit = st.button("Submit")
|
72 |
|