Update app.py

app.py

@@ -1,66 +1,93 @@
+import os
 import streamlit as st
-from tariff_scraper import scrape_tariff_data
+from tariff_scraper import fetch_tariff_data
+import pandas as pd
+from groq import Groq
 
-def main():
-    st.title("Electricity Bill Calculator with Tariff Scraper")
-    
-    # Sidebar: Tariff Rates
-    st.sidebar.subheader("Tariff Rates")
-    url = st.sidebar.text_input("Tariff URL", "https://iesco.com.pk/index.php/customer-services/tariff-guide")
+# Initialize Groq client
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-    if st.sidebar.button("Fetch Tariff Rates"):
-        with st.spinner("Fetching tariff rates..."):
-            data = scrape_tariff_data(url)
-            if isinstance(data, dict):  # Check if valid tariff data is returned
-                st.session_state["tariff_rates"] = data
-                st.sidebar.success("Tariff rates fetched successfully!")
-            else:
-                st.sidebar.error(data)
+# 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
+}
 
-    # Use Fetched Tariff Rates
-    tariff_categories = st.session_state.get("tariff_rates", {})
-    if tariff_categories:
-        load_type = st.sidebar.selectbox("Select Load Type", options=tariff_categories.keys())
-        tariff_rate = tariff_categories.get(load_type, 0.0)
-    else:
-        st.sidebar.warning("Using default rate. Fetch tariff rates to use live data.")
-        load_type = "Default"
-        tariff_rate = st.sidebar.number_input("Default Tariff Rate (PKR/unit)", min_value=0.0, value=20.0)
+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
 
-    # Sidebar: Appliance Usage
-    st.sidebar.subheader("Appliance Usage")
-    appliances = {
-        "Fan": 75, "Air Conditioner (1 Ton)": 1500, "Refrigerator": 150,
-        "LED Bulb (20W)": 20, "Iron": 1000, "Microwave Oven": 1200
-    }
-    appliance = st.sidebar.selectbox("Select Appliance", options=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)
+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
 
-    if st.sidebar.button("Add Appliance"):
-        st.session_state.setdefault("appliances", []).append({
-            "appliance": appliance,
-            "load": appliances[appliance],
-            "quantity": quantity,
-            "usage_hours": usage_hours
-        })
-        st.sidebar.success(f"{quantity} x {appliance} added.")
+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
 
-    # Main App: Display Appliances
-    st.write("### Selected Appliances")
-    if "appliances" in st.session_state:
-        for app in st.session_state["appliances"]:
-            st.write(f"{app['quantity']} x {app['appliance']} ({app['usage_hours']} hours/day)")
-        
-        # Calculate Bill
-        if st.button("Calculate Bill"):
-            total_units = sum(
-                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 for {load_type}: PKR {bill:.2f}")
-    else:
-        st.write("No appliances added yet.")
+st.title("Dynamic Tariff Electricity Bill Calculator")
 
-if __name__ == "__main__":
-    main()
+# 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("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!")
+
+# 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.")
+
+# 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}**")