Update app.py

app.py

@@ -80,19 +80,24 @@ def process_rooftop_image(image, geographic_bounds):
 
 # Streamlit Interface
 st.title("Rooftop Usable Area for Solar Panels")
-st.write("Upload a rooftop image to calculate the usable area for solar panels.")
+st.write("Upload a rooftop image to calculate the usable area for solar panels, panel count, total cost, and energy generation potential.")
 
 # File uploader for the image
 uploaded_file = st.file_uploader("Upload an image (PNG or JPG)", type=["png", "jpg", "jpeg"])
 
 # Input fields for geographic bounds with default values
-lat_min = st.number_input("Latitude Min", value=89.34, step=0.0001)
+lat_min = st.number_input("Latitude Min", value=89.36, step=0.0001)
 lat_max = st.number_input("Latitude Max", value=89.37, step=0.0001)
 lon_min = st.number_input("Longitude Min", value=90.78, step=0.0001)
 lon_max = st.number_input("Longitude Max", value=90.80, step=0.0001)
 
+# Input fields for irradiance, panel efficiency, and sunlight hours
+irradiance = st.number_input("Irradiance (W/m²)", value=200, step=1)  # Default irradiance value (W/m²)
+panel_efficiency = st.number_input("Panel Efficiency (%)", value=18, step=1) / 100  # Default efficiency (18%)
+sunlight_hours = st.number_input("Sunlight Hours per Day", value=5, step=1)  # Default 5 hours of sunlight
+
 # Button to process the image
-if st.button("Calculate Usable Area"):
+if st.button("Calculate Usable Area and Energy Potential"):
     if uploaded_file is not None:
         # Read the uploaded image
         image = np.array(Image.open(uploaded_file).convert("RGB"))  # Convert PIL image to NumPy array
@@ -108,8 +113,20 @@ if st.button("Calculate Usable Area"):
         # Process the image
         final_image, usable_area = process_rooftop_image(image, geographic_bounds)
 
+        # Calculate panel count, total cost, and potential energy
+        panel_area = 5 * 5  # Each panel covers 25 square meters
+        panel_cost = 25000  # Cost per panel in INR
+        panel_count = usable_area // panel_area
+        total_cost = panel_count * panel_cost
+
+        # Calculate potential energy generation
+        energy_generation = usable_area * panel_efficiency * irradiance * sunlight_hours  # in Wh (Watt-hours)
+
         # Display results
         st.image(cv2.cvtColor(final_image, cv2.COLOR_BGR2RGB), caption="Processed Image with Boundaries")
         st.write(f"Usable Rooftop Area for Solar Panels (excluding shadows): **{usable_area:.2f} square meters**")
+        st.write(f"Number of 5×5m panels that can be used: **{int(panel_count)}**")
+        st.write(f"Total installation cost (₹25,000 per panel): **₹{int(total_cost):,}**")
+        st.write(f"Total potential energy generation per day: **{energy_generation:.2f} Wh**")
     else:
         st.error("Please upload an image to proceed.")