Update app.py

app.py

@@ -5,6 +5,13 @@ import os
 from dotenv import load_dotenv
 import plotly.graph_objects as go
 
+import streamlit as st
+import pandas as pd
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
+import plotly.graph_objects as go
+
 # Load environment variables from .env file
 load_dotenv()
 
@@ -33,21 +40,13 @@ df = load_data()
 def build_prompt(location, project_type, roof_size=None, desired_kwh=None, electricity_bill=None, ghi=None, solar_cost_per_kw=None):
     if project_type == "Rooftop Solar":
         prompt = f"""
-You are a solar project estimator tool. Based on the following details, calculate and return only the values and formulas used, without any extra description:
+You are a solar project estimator tool. Based on the following details, calculate and return only the values without any extra description:
 Project Type: Rooftop Solar
 Location: {location}
 Roof size: {roof_size} sq meters
 Monthly electricity bill: ₹{electricity_bill}
 Average GHI: {ghi} kWh/m²/day
 Solar system cost per kW: ₹{solar_cost_per_kw}
-
-Formulas:
-1. Estimated solar system size in kW = (Desired monthly production in kWh) / (GHI * Roof Area)
-2. Estimated daily solar output in kWh = System size (kW) * GHI
-3. Total system cost in ₹ = System size (kW) * Solar cost per kW (₹)
-4. Monthly savings in ₹ = Monthly bill * (Solar output in kWh / Monthly consumption)
-5. Payback period in years = Total system cost / (Monthly savings * 12)
-
 Respond strictly in this format (do not add anything extra):
 Estimated solar system size in kW: <value>
 Estimated daily solar output in kWh: <value>
@@ -57,21 +56,13 @@ Payback period in years: <value>
 """
     else:  # Ground Mount Solar
         prompt = f"""
-You are a solar project estimator tool. Based on the following details, calculate and return only the values and formulas used, without any extra description:
+You are a solar project estimator tool. Based on the following details, calculate and return only the values without any extra description:
 Project Type: Ground Mount Solar
 Location: {location}
 Desired monthly solar production: {desired_kwh} kWh
 Monthly electricity bill: ₹{electricity_bill}
 Average GHI: {ghi} kWh/m²/day
 Solar system cost per kW: ₹{solar_cost_per_kw}
-
-Formulas:
-1. Required solar system size in kW = Desired monthly production (kWh) / GHI
-2. Estimated daily solar output in kWh = System size (kW) * GHI
-3. Total system cost in ₹ = System size (kW) * Solar cost per kW (₹)
-4. Monthly savings in ₹ = Monthly bill * (Solar output in kWh / Monthly consumption)
-5. Payback period in years = Total system cost / (Monthly savings * 12)
-
 Respond strictly in this format (do not add anything extra):
 Required solar system size in kW: <value>
 Estimated daily solar output in kWh: <value>
@@ -129,7 +120,7 @@ if submitted and location:
         monthly_savings_rs = None
         total_system_cost = None
         payback_period_years = None
-        daily_output_kwh = None  # Add this variable
+        daily_output_kwh = None
 
         for point in estimated_data:
             if ":" in point:
@@ -156,11 +147,34 @@ if submitted and location:
         if system_size_kw is not None and ghi is not None:
             daily_output_kwh = system_size_kw * ghi  # Estimate daily output in kWh
         
+        # Show the formulas
+        st.subheader("🧮 Formulas Used:")
+
+        if project_type == "Rooftop Solar":
+            st.markdown(f"""
+            **For Rooftop Solar:**
+            - Estimated Solar System Size (kW) = Roof Size (m²) × 0.15
+            - Estimated Daily Solar Output (kWh) = System Size (kW) × Average GHI (kWh/m²/day)
+            - Total System Cost (₹) = System Size (kW) × Solar Cost per kW (₹)
+            - Monthly Savings (₹) = (Estimated Daily Output × 30 × Tariff Rate per kWh) [Approximate]
+            - Payback Period (years) = Total System Cost ÷ (Monthly Savings × 12)
+            """)
+        else:
+            st.markdown(f"""
+            **For Ground Mount Solar:**
+            - Required Solar 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 Cost per kW (₹)
+            - Monthly Savings (₹) = (Estimated Daily Output × 30 × Tariff Rate per kWh) [Approximate]
+            - Payback Period (years) = Total System Cost ÷ (Monthly Savings × 12)
+            """)
+        
+        st.info("Note: Tariff rate is assumed based on your state electricity average or ₹7/kWh (default).")
+
         # Visual Summary - Grouped Bar Chart
         if system_size_kw is not None and total_system_cost is not None:
             st.subheader("📊 Visual Summary")
             
-            # Grouped Bar chart for all the values
             fig = go.Figure(data=[ 
                 go.Bar(
                     name="System Size (kW)",
@@ -176,7 +190,6 @@ if submitted and location:
                 )
             ])
             
-            # Update layout
             fig.update_layout(
                 barmode='group',
                 title="Comparison of Solar System Parameters",