Update app.py

app.py

@@ -3,25 +3,27 @@ from tariff_scraper import scrape_tariff_data
 
 def main():
     st.title("Electricity Bill Calculator with Tariff Scraper")
-    st.sidebar.subheader("Tariff Data Scraper")
-
-    # Tariff Scraper Section
-    st.sidebar.write("Enter the URL of the electricity tariff page:")
-    url = st.sidebar.text_input("Tariff URL", "https://iesco.com.pk/index.php/customer-services/tariff-guide")
     
-    if st.sidebar.button("Scrape Tariff Data"):
-        if url:
-            with st.spinner("Scraping data..."):
-                data = scrape_tariff_data(url)
-                if isinstance(data, list):
-                    st.success("Tariff data fetched successfully!")
-                    st.write("### Scraped Tariff Data")
-                    for row in data[:10]:  # Display first 10 rows
-                        st.write(row)
-                else:
-                    st.error(data)
-        else:
-            st.sidebar.error("Please provide a valid URL.")
+    # Default URL for scraping tariff rates
+    url = "https://iesco.com.pk/index.php/customer-services/tariff-guide"
+    
+    # Scrape tariff rates
+    tariff_rate = 0.0
+    with st.spinner("Fetching tariff rates..."):
+        data = scrape_tariff_data(url)
+        if isinstance(data, list) and data:
+            # Extract tariff rate from the scraped data
+            # Assuming tariff rate is mentioned as a numeric value in PKR/unit
+            for row in data:
+                if "PKR" in row:  # Replace with specific logic matching the structure of the scraped data
+                    try:
+                        tariff_rate = float(row.split()[1])  # Adjust this parsing logic based on your data
+                        break
+                    except (ValueError, IndexError):
+                        continue
+        if tariff_rate == 0.0:
+            st.warning("Could not fetch tariff rates. Using default rate of PKR 20.0/unit.")
+            tariff_rate = 20.0  # Default rate
 
     # Electricity Bill Calculator Section
     st.write("## Electricity Bill Calculator")
@@ -31,7 +33,7 @@ def main():
         "Fan": 75, "Air Conditioner (1 Ton)": 1500, "Refrigerator": 150,
         "LED Bulb (20W)": 20, "Iron": 1000, "Microwave Oven": 1200
     }
-    tariff_rate = st.sidebar.number_input("Tariff Rate (PKR/unit)", min_value=0.0, value=20.0)
+    
     appliance = st.sidebar.selectbox("Select Appliance", options=list(appliances.keys()))
     quantity = st.sidebar.number_input("Quantity", min_value=1, value=1)
     usage_hours = st.sidebar.number_input("Usage Hours/Day", min_value=1.0, value=6.0)
@@ -60,7 +62,7 @@ def main():
                 app["quantity"] * app["load"] * app["usage_hours"] / 1000 for app in st.session_state["appliances"]
             )
             bill = total_units * 30 * tariff_rate
-            st.success(f"Estimated Monthly Bill: PKR {bill:.2f}")
+            st.success(f"Estimated Monthly Bill: PKR {bill:.2f} (Tariff Rate: PKR {tariff_rate:.2f}/unit)")
         else:
             st.warning("No appliances added for calculation.")