Update app.py

app.py

@@ -1,8 +1,16 @@
 import os
 import streamlit as st
+import urllib3
+from bs4 import BeautifulSoup
+import pandas as pd
+import tempfile
 from groq import Groq
-import requests
-from utils import calculate_bill
+
+# Initialize Groq client
+client = Groq(api_key=os.environ.get('GroqApi'))
+
+# Tariff URL for scraping
+TARIFF_URL = "https://iesco.com.pk/index.php/customer-services/tariff-guide"
 
 # Predefined appliance list with typical loads in watts
 APPLIANCE_OPTIONS = {
@@ -23,42 +31,49 @@ APPLIANCE_OPTIONS = {
     "Geyser (Electric)": 3000
 }
 
-# Initialize session state for appliance data
+# Initialize session state
 if "appliance_data" not in st.session_state:
     st.session_state.appliance_data = []
+if "tariff_data" not in st.session_state:
+    st.session_state.tariff_data = {"rate_per_unit": 25}  # Default fallback rate
 
-# Groq Client Setup
-client = Groq(api_key=os.environ.get('GroqApi'))
+def scrape_tariff_data(url):
+    """Scrape tariff data from the specified URL."""
+    try:
+        http = urllib3.PoolManager()
+        response = http.request('GET', url)
+        if response.status == 200:
+            soup = BeautifulSoup(response.data, 'html.parser')
+            text = soup.get_text()
+            # Simulating tariff rate extraction (replace with accurate extraction logic)
+            rate = 25  # Placeholder value, replace with extraction from `text`
+            return {"rate_per_unit": rate}
+        else:
+            st.warning(f"Failed to fetch tariff data. Status code: {response.status}")
+    except Exception as e:
+        st.error(f"Error scraping tariff data: {e}")
+    return {"rate_per_unit": 25}  # Default fallback rate
 
-# App Configuration
+# Streamlit App Configuration
 st.title("Pakistani Electricity Bill Calculator")
-st.sidebar.header("Add Appliances")
-st.sidebar.write("Select appliances, quantity, and usage details to calculate electricity consumption.")
 
-# Tariff Data
-def fetch_tariff_data():
-    """Fetch tariff data from the WAPDA website."""
-    url = "https://nepra.org.pk/consumer%20affairs/Electricity%20Bill.php"  # Replace with actual URL
-    try:
-        response = requests.get(url, timeout=10)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        print(f"Error fetching tariff data: {e}")
-        return {"rate_per_unit": 25}  # Return mock data as fallback
+# Step 1: Fetch Tariff Data
+st.header("Step 1: Fetch Tariff Data")
+if st.button("Fetch Tariff Data"):
+    st.session_state.tariff_data = scrape_tariff_data(TARIFF_URL)
+    st.success("Tariff data updated successfully!")
 
-st.write("Fetching real-time tariff data...")
-tariff_data = fetch_tariff_data()
-st.success("Tariff data fetched successfully!")
+# Display current tariff data
+st.write(f"**Current Tariff Rate:** PKR {st.session_state.tariff_data['rate_per_unit']} per kWh")
 
-# Appliance Selection
+# Step 2: Add Appliances
+st.sidebar.header("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)
 
-# Add Appliance Button
 if st.sidebar.button("Add Appliance"):
     st.session_state.appliance_data.append({
         "appliance": appliance,
@@ -69,41 +84,36 @@ if st.sidebar.button("Add Appliance"):
     st.sidebar.success(f"Added {quantity} {appliance}(s) to the list!")
 
 # Display Appliance List
-st.header("Appliance List")
+st.header("Step 3: Appliance List")
 if st.session_state.appliance_data:
     total_load = 0
-    total_usage_hours = 0
-    total_energy_wh = 0  # Total energy in watt-hours
+    total_energy_wh = 0
     for idx, data in enumerate(st.session_state.appliance_data):
-        st.write(f"{idx+1}. {data['appliance']} - {data['quantity']} units, {data['load']}W each, {data['usage_hours']} hours/day")
+        st.write(f"{idx + 1}. {data['appliance']} - {data['quantity']} units, {data['load']}W each, {data['usage_hours']} hours/day")
         total_load += data["quantity"] * data["load"]
         total_energy_wh += data["quantity"] * data["load"] * data["usage_hours"]
     st.write(f"**Total Load: {total_load} Watts**")
 else:
     st.write("No appliances added yet.")
 
-# Calculate Monthly Bill Button
+# Step 4: Calculate Monthly Bill
 if st.button("Calculate Monthly Bill"):
-    # Convert total energy consumption from watt-hours (Wh) to kilowatt-hours (kWh)
-    total_energy_kwh = total_energy_wh / 1000  # Convert energy to kWh
-
-    # Calculate total monthly energy consumption by multiplying by 30
-    monthly_energy_kwh = total_energy_kwh * 30
-
-    # Get the rate from tariff data or use a default rate if not found
-    rate_per_unit = tariff_data.get("rate_per_unit", 25)  # Default rate of 25 PKR/kWh
-
-    # Correct Monthly Bill Calculation: Total Monthly Energy in kWh * Rate per kWh (Rate in PKR)
-    monthly_bill = monthly_energy_kwh * rate_per_unit
+    if not st.session_state.appliance_data:
+        st.warning("Please add appliances before calculating the bill.")
+    else:
+        total_energy_kwh = total_energy_wh / 1000  # Convert energy to kWh
+        monthly_energy_kwh = total_energy_kwh * 30
+        rate_per_unit = st.session_state.tariff_data.get("rate_per_unit", 25)
+        monthly_bill = monthly_energy_kwh * rate_per_unit
 
-    # Query AI Model (optional, if needed for insights)
-    query = f"Calculate the monthly electricity bill for appliances with a total energy consumption of {monthly_energy_kwh:.2f} kWh per month."
-    response = client.chat.completions.create(
-        messages=[{"role": "user", "content": query}],
-        model="llama3-8b-8192",
-    )
+        # Query AI Model (optional, for additional insights)
+        query = f"Calculate the monthly electricity bill for appliances with a total energy consumption of {monthly_energy_kwh:.2f} kWh per month."
+        response = client.chat.completions.create(
+            messages=[{"role": "user", "content": query}],
+            model="llama3-8b-8192",
+        )
 
-    # Display Results
-    st.write(f"**Total Monthly Electricity Bill: PKR {monthly_bill:.2f}**")
-    st.write("**AI Model Insight:**")
-    st.write(response.choices[0].message.content)
+        # Display Results
+        st.write(f"**Total Monthly Electricity Bill: PKR {monthly_bill:.2f}**")
+        st.write("**AI Model Insight:**")
+        st.write(response.choices[0].message.content)