Update app.py

app.py

@@ -3,14 +3,8 @@ import requests
 import streamlit as st
 import pandas as pd
 from scraper import scrape_tariffs
-from transformers import AutoTokenizer, AutoModel
-import torch
 
-# Load pre-trained transformer model for query embeddings
-tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
-model = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
-
-# Streamlit App: Electricity Bill Estimator
+# Streamlit App: Electricity Bill & Carbon Footprint Estimator
 st.title("🔌 Electricity Bill & Carbon Footprint Estimator")
 st.sidebar.header("⚙️ User Input")
 
@@ -59,6 +53,18 @@ def calculate_carbon_footprint(monthly_energy_kwh):
 if st.sidebar.button("Scrape Tariff Data"):
     scrape_data()
 
+# Sidebar: Tariff Selection
+st.sidebar.subheader("💡 Select Tariff")
+try:
+    tariff_data = pd.read_csv("data/tariffs.csv")
+    tariff_types = tariff_data["category"].unique()
+    selected_tariff = st.sidebar.selectbox("Select your tariff category:", tariff_types)
+    rate_per_kwh = tariff_data[tariff_data["category"] == selected_tariff]["rate"].iloc[0]
+    st.sidebar.write(f"Rate per kWh: **{rate_per_kwh} PKR**")
+except FileNotFoundError:
+    st.sidebar.error("⚠️ Tariff data not found. Please scrape the data first.")
+    rate_per_kwh = 0
+
 # Sidebar: User Inputs for Appliances
 st.sidebar.subheader("🏠 Add Appliances")
 selected_appliance = st.sidebar.selectbox("Select an appliance:", list(appliances.keys()))
@@ -95,13 +101,13 @@ if st.session_state["appliance_list"]:
         )
 
 # Electricity Bill and Carbon Footprint Calculation
-if st.session_state["appliance_list"]:
+if st.session_state["appliance_list"] and rate_per_kwh > 0:
     total_daily_energy_kwh = sum(
         (appliance["power"] * appliance["quantity"] * appliance["hours"]) / 1000
         for appliance in st.session_state["appliance_list"]
     )
     monthly_energy_kwh = total_daily_energy_kwh * 30  # Assume 30 days in a month
-    bill_amount = monthly_energy_kwh * 0.25  # Simplified bill calculation
+    bill_amount = monthly_energy_kwh * rate_per_kwh  # Dynamic tariff rate
     carbon_footprint = calculate_carbon_footprint(monthly_energy_kwh)
 
     st.subheader("💵 Electricity Bill & 🌍 Carbon Footprint")