Update app.py

app.py

@@ -5,7 +5,6 @@ import os
 from dotenv import load_dotenv
 import plotly.graph_objects as go
 
-
 # Load environment variables from .env file
 load_dotenv()
 
@@ -35,16 +34,13 @@ def build_prompt(location, project_type, roof_size=None, desired_kwh=None, elect
     if project_type == "Rooftop Solar":
         prompt = f"""
 You are an AI-based solar project estimator.
-
 Use the following calculation methods:
-
 - Estimated system size (kW) = Roof size (sq meters) × 0.15
 - Estimated daily solar output (kWh) = System size (kW) × Average GHI (kWh/m²/day)
 - Total system cost (₹) = System size (kW) × Solar system cost per kW
 - Assume tariff rate = ₹7/kWh
 - Monthly savings (₹) = Estimated daily output × 30 × 7
 - Payback period (years) = Total system cost ÷ (Monthly savings × 12)
-
 Inputs:
 - Project Type: Rooftop Solar
 - Location: {location}
@@ -52,7 +48,6 @@ Inputs:
 - Monthly Electricity Bill: ₹{electricity_bill}
 - Average GHI: {ghi} kWh/m²/day
 - Solar System Cost per kW: ₹{solar_cost_per_kw}
-
 Now, calculate and return strictly in this format:
 Estimated solar system size in kW: <value>
 Estimated daily solar output in kWh: <value>
@@ -63,16 +58,13 @@ Payback period in years: <value>
     else:
         prompt = f"""
 You are an AI-based solar project estimator.
-
 Use the following calculation methods:
-
 - Required system size (kW) = Desired monthly solar production ÷ (30 × Average GHI)
 - Estimated daily solar output (kWh) = System size (kW) × Average GHI (kWh/m²/day)
 - Total system cost (₹) = System size (kW) × Solar system cost per kW
 - Assume tariff rate = ₹7/kWh
 - Monthly savings (₹) = Estimated daily output × 30 × 7
 - Payback period (years) = Total system cost ÷ (Monthly savings × 12)
-
 Inputs:
 - Project Type: Ground Mount Solar
 - Location: {location}
@@ -80,7 +72,6 @@ Inputs:
 - Monthly Electricity Bill: ₹{electricity_bill}
 - Average GHI: {ghi} kWh/m²/day
 - Solar System Cost per kW: ₹{solar_cost_per_kw}
-
 Now, calculate and return strictly in this format:
 Required solar system size in kW: <value>
 Estimated daily solar output in kWh: <value>
@@ -124,6 +115,14 @@ if submitted and location:
         ghi = state_data['Avg_GHI (kWh/m²/day)']
         solar_cost_per_kw = state_data['Solar_Cost_per_kW (₹)']
 
+        # Check roof size limit for rooftop solar
+        if project_type == "Rooftop Solar" and roof_size:
+            max_allowed_kw = roof_size * 0.15
+            if max_allowed_kw > 5:  # Maximum 5 kW limit
+                st.warning(f"Roof size exceeds the maximum allowed capacity of 5 kW. The system size will be limited to 5 kW.")
+                max_allowed_kw = 5
+            roof_size = max_allowed_kw / 0.15  # Adjust roof size to fit within the limit
+
         prompt_text = build_prompt(location, project_type, roof_size=roof_size, desired_kwh=desired_kwh, electricity_bill=electricity_bill, ghi=ghi, solar_cost_per_kw=solar_cost_per_kw)
         
         # Call Gemini API