Update main.py

main.py

@@ -163,6 +163,12 @@ class InvoiceProcessingApp:
     
         files = self.get_available_files()
         chosen = st.sidebar.multiselect("Select invoices to process", files)
+
+        st.sidebar.markdown("---")
+        st.sidebar.markdown("## 🧾 Purchase Order Details")
+    
+        POs = self.get_available_PO()
+        PO = st.sidebar.multiselect("Select PO to process", POs)
     
         st.session_state.selected_files = chosen
     
@@ -199,6 +205,16 @@ class InvoiceProcessingApp:
         ]
         return sorted(files)
 
+    def get_available_PO(self) -> List[str]:
+        po_dir = "data"
+        os.makedirs(po_dir, exist_ok=True)
+        files = [
+            os.path.join(po_dir, f)
+            for f in os.listdir(po_dir)
+            if f.lower().endswith(".csv")
+        ]
+        return sorted(files)
+
     # WORKFLOW EXECUTION
     # def _get_stages_for_workflow(self, workflow_type: str) -> list[str]:
     #     """Return dynamic stage flow for given workflow type."""