Update app.py

app.py

@@ -1,24 +1,42 @@
 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("IESCO Tariff Rates Viewer")
-    st.write("This app fetches and displays the latest IESCO tariff rates.")
+    st.title("Electricity Tariff Data Viewer")
+    st.sidebar.header("Tariff Data Scraper")
 
-    # Display the fetch button
-    if st.button("Fetch IESCO Tariff Rates"):
-        url = TARIFF_URLS["IESCO"]
-        st.write(f"Fetching data from {url}...")
+    # Dropdown for selecting a company
+    selected_company = st.sidebar.selectbox("Select a Company", options=list(TARIFF_URLS.keys()))
+    st.sidebar.write("Selected Company:", selected_company)
 
+    # Button to fetch and scrape data
+    if st.sidebar.button("Fetch Tariff Data"):
+        url = TARIFF_URLS[selected_company]
+        st.sidebar.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)
+            result = scrape_tariff_data_to_csv(url, output_file=f"{selected_company}_tariff_data.csv")
+            if result.endswith(".csv"):
+                st.sidebar.success(f"Data successfully saved to {result}")
+                st.session_state["csv_file"] = result
             else:
-                st.error("Failed to fetch tariff data or no data found.")
+                st.sidebar.error(f"Error: {result}")
+
+    # Display the preview of the data if available
+    if "csv_file" in st.session_state and st.session_state["csv_file"]:
+        st.subheader("Preview of Tariff Data")
+        try:
+            csv_file = st.session_state["csv_file"]
+            data = pd.read_csv(csv_file)
+            st.write(data)
+            st.download_button(
+                label="Download Tariff Data as CSV",
+                data=open(csv_file, "rb").read(),
+                file_name=csv_file,
+                mime="text/csv"
+            )
+        except Exception as e:
+            st.error(f"An error occurred while loading the CSV file: {e}")
 
 if __name__ == "__main__":
     main()