Spaces:
Sleeping
Sleeping
File size: 2,094 Bytes
59537b7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import streamlit as st
import pdfplumber
import re
# Function to extract text from PDF
def extract_text_from_pdf(pdf_path):
try:
with pdfplumber.open(pdf_path) as pdf:
text = "".join([page.extract_text() for page in pdf.pages])
return text
except Exception as e:
return str(e)
# Function to extract numeric rates
def extract_rates(text):
rates = re.findall(r"\b\d+(?:\.\d+)?\b", text) # Regex for numbers, including decimals
return rates
# Function to calculate electricity bill
def calculate_bill(units, rate_per_unit):
return units * rate_per_unit
# Streamlit App
def main():
st.title("Tariff Rate Extractor and Electricity Bill Calculator")
st.sidebar.title("Options")
app_mode = st.sidebar.selectbox(
"Choose the mode:",
["Extract Rates from PDF", "Calculate Electricity Bill"]
)
if app_mode == "Extract Rates from PDF":
st.header("Upload a PDF to Extract Rates")
pdf_file = st.file_uploader("Upload PDF", type=["pdf"])
if pdf_file is not None:
with open(f"/tmp/{pdf_file.name}", "wb") as f:
f.write(pdf_file.getbuffer())
# Extract text and rates
text = extract_text_from_pdf(f"/tmp/{pdf_file.name}")
rates = extract_rates(text)
st.subheader("Extracted Numeric Rates")
if rates:
st.write(rates)
else:
st.write("No numeric rates found in the document.")
elif app_mode == "Calculate Electricity Bill":
st.header("Electricity Bill Calculator")
units = st.number_input(
"Enter the number of units consumed (kWh):", min_value=0.0, step=0.1
)
rate_per_unit = st.number_input(
"Enter the rate per unit (e.g., 5.5):", min_value=0.0, step=0.1
)
if st.button("Calculate Bill"):
total_bill = calculate_bill(units, rate_per_unit)
st.success(f"Your total electricity bill is: ₹{total_bill:.2f}")
if __name__ == "__main__":
main()
|