Update app.py

app.py

@@ -1,36 +1,88 @@
 import streamlit as st
 from tariff_scraper import TARIFF_URLS, scrape_tariff_data
 
+def calculate_bill(units_consumed, tariff_data):
+    """
+    Calculate the bill based on units consumed and tariff rates.
+    Assumes the tariff data contains rates per unit in a structured format.
+    """
+    for row in tariff_data:
+        if "to" in row.lower():  # Check if the row contains range information
+            try:
+                parts = row.split("|")
+                unit_range = parts[0].strip()
+                rate = float(parts[-1].strip())  # Assuming rate is the last column
+                
+                # Parse the unit range (e.g., "1-100" or ">300")
+                if "-" in unit_range:
+                    low, high = map(int, unit_range.split("-"))
+                    if low <= units_consumed <= high:
+                        return units_consumed * rate
+                elif ">" in unit_range:
+                    threshold = int(unit_range.strip(">").strip())
+                    if units_consumed > threshold:
+                        return units_consumed * rate
+            except:
+                pass
+    return 0.0  # Default to 0 if no matching rate is found
+
 def main():
-    st.title("Electricity Tariff Scraper")
-    st.write("Select a company to fetch tariff data:")
+    st.set_page_config(page_title="Electricity Bill Calculator", layout="wide")
+    st.title("Electricity Bill Calculator")
 
-    # Dropdown menu for selecting the company
-    selected_company = st.selectbox("Select Company", options=list(TARIFF_URLS.keys()))
+    # Sidebar for company selection
+    with st.sidebar:
+        st.header("Select Company")
+        selected_company = st.selectbox("Company", options=list(TARIFF_URLS.keys()))
+        st.write("Selected Company:", selected_company)
 
-    if st.button("Fetch Tariff Rates"):
+        # Fetch tariff data from the selected company
         url = TARIFF_URLS[selected_company]
-        st.write(f"Fetching data from {selected_company} ({url})...")
-
-        with st.spinner("Scraping data..."):
-            data = scrape_tariff_data(url)
-            if data:
-                st.success(f"Data scraped successfully for {selected_company}!")
-                st.write("### Preview of Tariff Data:")
-                st.write("Here is the scraped data (showing the first 10 rows):")
-                for row in data[:10]:  # Show only the first 10 rows for readability
-                    st.write(row)
-                st.write("### Download Full Data:")
-                if st.button("Download as Text"):
-                    full_data = "\n".join(data)
-                    st.download_button(
-                        label="Download Tariff Data",
-                        data=full_data,
-                        file_name=f"{selected_company}_tariff_data.txt",
-                        mime="text/plain"
-                    )
-            else:
-                st.error("Failed to scrape data or no data available.")
+        with st.spinner("Fetching tariff data..."):
+            tariff_data = scrape_tariff_data(url)
+
+    # Main app interface for bill calculation
+    st.header("Appliances and Usage")
+    st.write("Add your appliances and usage details below:")
+
+    # Predefined appliances
+    appliances = {
+        "Fan": 75,  # Power in watts
+        "Bulb": 40,
+        "Refrigerator": 150,
+        "Air Conditioner": 1500,
+        "Washing Machine": 500,
+        "Microwave": 1200
+    }
+
+    # Allow users to add appliances and customize
+    selected_appliances = st.multiselect("Select Appliances", options=list(appliances.keys()))
+    custom_appliance_name = st.text_input("Add Custom Appliance Name")
+    custom_appliance_power = st.number_input("Custom Appliance Power (Watts)", min_value=1, step=1)
+
+    if custom_appliance_name and custom_appliance_power:
+        appliances[custom_appliance_name] = custom_appliance_power
+        st.success(f"Added custom appliance: {custom_appliance_name} ({custom_appliance_power}W)")
+
+    # Input fields for usage details
+    total_units = 0
+    for appliance in selected_appliances:
+        st.subheader(f"{appliance} ({appliances[appliance]}W)")
+        num_units = st.number_input(f"How many {appliance}s?", min_value=1, step=1, key=f"{appliance}_units")
+        usage_hours = st.number_input(f"Daily usage hours for {appliance}:", min_value=1, step=1, key=f"{appliance}_hours")
+        monthly_units = (appliances[appliance] / 1000) * num_units * usage_hours * 30  # Convert watts to kWh
+        total_units += monthly_units
+        st.write(f"Estimated Monthly Units for {appliance}: {monthly_units:.2f} kWh")
+
+    st.write("### Total Estimated Monthly Units Consumed:", f"{total_units:.2f} kWh")
+
+    # Calculate the bill
+    if st.button("Calculate Bill"):
+        bill = calculate_bill(total_units, tariff_data)
+        if bill > 0:
+            st.success(f"Estimated Monthly Bill: Rs. {bill:.2f}")
+        else:
+            st.error("Could not calculate the bill. Please check the tariff data or try again.")
 
 if __name__ == "__main__":
     main()