Update app.py

app.py

@@ -1,29 +1,24 @@
 import streamlit as st
-import pandas as pd
-from tariff_scraper import TARIFF_URLS, scrape_multiple_sections_to_csv
+from tariff_scraper import TARIFF_URLS, scrape_tariff_data
 
 def main():
-    st.title("PESCO Tariff Rates Viewer")
-    st.write("This app fetches and displays the latest PESCO tariff rates.")
+    st.title("IESCO Tariff Rates Viewer")
+    st.write("This app fetches and displays the latest IESCO tariff rates.")
 
-    # Button to fetch and save tariff rates
-    if st.button("Fetch and Save PESCO Tariff Rates"):
-        url = TARIFF_URLS["PESCO"]
+    # Display the fetch button
+    if st.button("Fetch IESCO Tariff Rates"):
+        url = TARIFF_URLS["IESCO"]
         st.write(f"Fetching data from {url}...")
 
         with st.spinner("Scraping data..."):
-            output_file = scrape_multiple_sections_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}")
+            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)
             else:
-                st.error(output_file)
+                st.error("Failed to fetch tariff data or no data found.")
 
 if __name__ == "__main__":
     main()