Update app.py

app.py

@@ -1,93 +1,53 @@
 import os
+import requests
 import streamlit as st
-from tariff_scraper import fetch_tariff_data
 import pandas as pd
-from groq import Groq
-
-# Initialize Groq client
-client = Groq(api_key=os.environ.get('GroqApi'))
-
-# Predefined appliance list with typical loads in watts
-APPLIANCE_OPTIONS = {
-    "Fan": 75,
-    "Air Conditioner (1 Ton)": 1500,
-    "Air Conditioner (1.5 Ton)": 2200,
-    "Refrigerator": 150,
-    "LED Bulb (20W)": 20,
-    "Tube Light": 40,
-    "Iron": 1000,
-    "Microwave Oven": 1200,
-    "Washing Machine": 500,
-    "Electric Heater": 1500,
-    "Laptop": 50,
-    "Desktop Computer": 200,
-    "Television (LCD/LED)": 120,
-    "Water Pump": 1000,
-    "Geyser (Electric)": 3000
+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"
 }
 
-if "appliance_data" not in st.session_state:
-    st.session_state.appliance_data = []
-if "tariff_rate" not in st.session_state:
-    st.session_state.tariff_rate = None
-
-def calculate_total_units():
-    """Calculate total units consumed in kWh."""
-    total_energy_wh = 0
-    for data in st.session_state.appliance_data:
-        total_energy_wh += data["quantity"] * data["load"] * data["usage_hours"]
-    return total_energy_wh / 1000  # Convert watt-hours to kilowatt-hours
+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 fetch_tariff_from_groq(units):
-    """Fetch tariff rate based on units consumed using Groq API."""
-    try:
-        response = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": f"Provide tariff rate for {units} units consumed in PKR."
-                }
-            ],
-            model="llama3-8b-8192",
-        )
-        return float(response.choices[0].message.content.strip())
-    except Exception as e:
-        st.error(f"Error fetching tariff from Groq: {e}")
-        return None
+def scrape_data():
+    # Scraping tariff data using provided URLs
+    scrape_tariffs(list(tariff_urls.values()))
 
-st.title("Dynamic Tariff Electricity Bill Calculator")
+# Streamlit actions
+if st.sidebar.button("Scrape Data"):
+    scrape_data()
 
-# Add Appliances
-st.sidebar.subheader("Add Appliances")
-appliance = st.sidebar.selectbox("Select Appliance", options=list(APPLIANCE_OPTIONS.keys()))
-default_load = APPLIANCE_OPTIONS[appliance]
-quantity = st.sidebar.number_input("Number of Appliances", min_value=1, value=1)
-load = st.sidebar.number_input("Load (Watts per Appliance)", min_value=1, value=default_load)
-usage_hours = st.sidebar.number_input("Usage Time (Hours per Day)", min_value=0.0, value=6.0)
+if st.sidebar.button("Download Model"):
+    download_model()
 
-if st.sidebar.button("Add Appliance"):
-    st.session_state.appliance_data.append({
-        "appliance": appliance,
-        "load": load,
-        "quantity": quantity,
-        "usage_hours": usage_hours
-    })
-    st.sidebar.success(f"Added {quantity} {appliance}(s) to the list!")
+# 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)
 
-# Display Appliance List
-st.subheader("Appliance List")
-if st.session_state.appliance_data:
-    for idx, data in enumerate(st.session_state.appliance_data, start=1):
-        st.write(f"{idx}. {data['appliance']} - {data['quantity']} units, {data['load']}W each, {data['usage_hours']} hours/day")
-else:
-    st.write("No appliances added.")
+# 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")
 
-# Calculate Monthly Units and Electricity Bill
-if st.button("Calculate Monthly Bill"):
-    total_units_kwh = calculate_total_units() * 30  # Assuming 30 days in a month
-    st.write(f"**Total Units Consumed (kWh): {total_units_kwh:.2f}**")
-    tariff_rate = fetch_tariff_from_groq(total_units_kwh)
-    if tariff_rate:
-        monthly_bill = total_units_kwh * tariff_rate
-        st.subheader("Electricity Bill")
-        st.write(f"**Total Monthly Bill: PKR {monthly_bill:.2f}**")