Update app.py

app.py

@@ -1,24 +1,29 @@
 import streamlit as st
-from tariff_scraper import TARIFF_URLS, scrape_tariff_data
+import pandas as pd
+from tariff_scraper import TARIFF_URLS, scrape_tariff_data_to_csv
 
 def main():
     st.title("PESCO Tariff Rates Viewer")
     st.write("This app fetches and displays the latest PESCO tariff rates.")
 
-    # Display the fetch button
-    if st.button("Fetch PESCO Tariff Rates"):
+    # Button to fetch and save tariff rates
+    if st.button("Fetch and Save PESCO Tariff Rates"):
         url = TARIFF_URLS["PESCO"]
         st.write(f"Fetching data from {url}...")
 
         with st.spinner("Scraping data..."):
-            data = scrape_tariff_data(url)
-            if data and isinstance(data, list) and len(data) > 0:
-                st.success("Data fetched successfully!")
-                st.write("### Tariff Rates:")
-                for row in data:
-                    st.write(row)
+            output_file = scrape_tariff_data_to_csv(url)
+            if output_file.endswith(".csv"):
+                st.success(f"Data successfully saved to {output_file}!")
+                st.write("### Preview of Tariff Rates:")
+                try:
+                    # Read the CSV file and display it in tabular format
+                    data = pd.read_csv(output_file)
+                    st.dataframe(data)
+                except Exception as e:
+                    st.error(f"Error reading the CSV file: {e}")
             else:
-                st.error("Failed to fetch tariff data or no data found.")
+                st.error(output_file)
 
 if __name__ == "__main__":
     main()