Update app.py

app.py

@@ -1,53 +1,24 @@
-import os
-import requests
 import streamlit as st
-import pandas as pd
-from scraper import scrape_tariffs
-from sentence_transformers import SentenceTransformer
-
-# Replace the model loading approach with hf_hub_download to avoid cached_download error
-def download_model():
-    model_repo_id = "sentence-transformers/all-MiniLM-L6-v2"
-
-# Initialize Streamlit components
-st.title("Electricity Bill Estimator")
-st.sidebar.header("User Input")
-
-tariff_urls = {
-    "IESCO": "https://iesco.com.pk/index.php/customer-services/tariff-guide",
-    "FESCO": "https://fesco.com.pk/tariff",
-    "HESCO": "http://www.hesco.gov.pk/htmls/tariffs.htm",
-    "KE": "https://www.ke.com.pk/customer-services/tariff-structure/",
-    "LESCO": "https://www.lesco.gov.pk/ElectricityTariffs",
-    "PESCO": "https://pesconlinebill.pk/pesco-tariff/",
-    "QESCO": "http://qesco.com.pk/Tariffs.aspx",
-    "TESCO": "https://tesco.gov.pk/index.php/electricity-traiff"
-}
-
-def show_tariff_input():
-    # Display tariff rates selection
-    tariff_data = pd.read_csv("data/tariffs.csv")
-    tariff_types = tariff_data["category"].unique()
-    tariff_choice = st.selectbox("Select your tariff category:", tariff_types)
-    st.write(f"Selected Tariff: {tariff_choice}")
-
-def scrape_data():
-    # Scraping tariff data using provided URLs
-    scrape_tariffs(list(tariff_urls.values()))
-
-# Streamlit actions
-if st.sidebar.button("Scrape Data"):
-    scrape_data()
-
-if st.sidebar.button("Download Model"):
-    download_model()
-
-# User inputs for appliance and usage time (replace placeholders as needed)
-appliance_load = st.number_input("Enter appliance load in watts", min_value=10, max_value=5000, value=1000)
-usage_time = st.number_input("Enter usage time (in hours)", min_value=1, max_value=24, value=5)
-
-# Placeholder for electricity bill calculation and output display
-if appliance_load and usage_time:
-    bill_amount = appliance_load * usage_time * 0.25 # Add your own calculation based on tariffs
-    st.write(f"Your electricity bill: {bill_amount} PKR")
-
+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.")
+
+    # Display the fetch button
+    if st.button("Fetch 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)
+            else:
+                st.error("Failed to fetch tariff data or no data found.")
+
+if __name__ == "__main__":
+    main()