default

app.py

@@ -1,25 +1,57 @@
 from pathlib import Path
 import streamlit as st
 import os
+import requests
 
-from form16_parser import build_parser
+from form16_parser import build_parser, UnsupportedForm16Error
+
+badges = """[![](https://img.shields.io/static/v1?label=Licence&message=MIT&color=darkgreen)](https://github.com/INF800/form16-parser)
+[![stability-alpha](https://img.shields.io/badge/stability-alpha-f4d03f.svg)](https://github.com/mkenney/software-guides/blob/master/STABILITY-BADGES.md#alpha)
+[![](https://img.shields.io/static/v1?label=Python&message=3.10+&color=indigo)](https://github.com/INF800/form16-parser) 
+[![im](https://img.shields.io/github/stars/INF800/form16-parser.svg?style=social&)]()
+"""
 
 def main():
     st.title("Form16 Parser")
 
+    st.markdown(f"{badges}")
+    st.write("Source code available @ https://github.com/INF800/form16-parser")
     uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
     Path("temp_files").mkdir(exist_ok=True, parents=True)
 
-    if uploaded_file is not None:
+    if uploaded_file is None:
+        # If no file is uploaded, load the default PDF file
+        default_pdf_url = "https://assets1.cleartax-cdn.com/cleartax/images/1655725194_sampleform16.pdf"
+        path = Path("temp_files/sampleform16.pdf")
+        # if not path.exists():
+        default_pdf_response = requests.get(default_pdf_url)
+
+        with open(str(path), "wb") as f:
+            f.write(default_pdf_response.content)
+
+        st.write(f"Parsing {default_pdf_url}.")
+        st.write("Try your own form 16 by uploading above.")
+
+        parser = build_parser()
+        parsed = parser.parse(str(path), return_output=True)
+
+        st.write("Parsed data (structured):")
+        st.write(parsed)
+    else:
         # Save the uploaded file to a temporary directory
         with open(os.path.join("temp_files", uploaded_file.name), "wb") as f:
             f.write(uploaded_file.getvalue())
 
         filepath = os.path.join("temp_files", uploaded_file.name)        
         parser = build_parser()
-        parsed = parser.parse(filepath, return_output=True)
 
-        st.write(parsed)
+        try:
+            parsed = parser.parse(filepath, return_output=True)
+            st.write("Parsed data (structured):")
+            st.write(parsed)
+        except UnsupportedForm16Error as e:
+            st.write(e)
+