app.py

import streamlit as st
import os

from form16_parser import build_parser

def main():
    st.title("Form16 Parser")

    uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")

    if uploaded_file is not None:
        # 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)

        

if __name__ == "__main__":
    main()