Update app.py

app.py

@@ -5,22 +5,13 @@ import os
 from dotenv import load_dotenv
 import plotly.graph_objects as go
 
-# Load environment variables from .env file
+
+# Load environment variables
 load_dotenv()
 
 # Set page configuration
 st.set_page_config(page_title="☀️AI-Based Solar Project Estimation Tool", layout="centered")
 
-# Initialize Gemini with the API key
-api_key = os.getenv("GOOGLE_API_KEY")
-if api_key:
-    genai.configure(api_key=api_key)
-else:
-    st.error("API key is missing. Please set the GOOGLE_API_KEY environment variable.")
-
-# Use Gemini-1.5-Pro model
-model = genai.GenerativeModel("gemini-1.5-pro")
-
 # Load solar data
 @st.cache_data
 def load_data():
@@ -29,68 +20,16 @@ def load_data():
 
 df = load_data()
 
-# Build prompt for Gemini
-# Build prompt for Gemini
-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 an AI-based solar project estimator.
-Use the following calculation methods:
-- Estimated system size (kW) = Roof size (sq meters) × 0.10
-- 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}
-- 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}
-Now, calculate and return strictly in this format:
-Estimated solar system size in kW: <value>
-Estimated daily solar output in kWh: <value>
-Total system cost in ₹: <value>
-Monthly savings in ₹: <value>
-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}
-- 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}
-Now, calculate and return strictly in this format:
-Required solar system size in kW: <value>
-Estimated daily solar output in kWh: <value>
-Total system cost in ₹: <value>
-Monthly savings in ₹: <value>
-Payback period in years: <value>
-"""
-    return prompt
-
-
-
-# UI - Form for user input
+# Constants
+TARIFF_RATE = 7  # ₹7 per kWh
+ROOFTOP_CONVERSION_FACTOR = 0.10  # 0.10 kW per sq meter
+
+# UI - Form
 st.title("☀️AI-Based Solar Project Estimation Tool")
 st.write("### Enter Your Details Below:")
 
 with st.form("solar_form"):
     state_options = df['State'].dropna().unique()
-    
     location = st.selectbox("Select your State", options=sorted(state_options))
     
     project_type = st.radio(
@@ -109,116 +48,53 @@ with st.form("solar_form"):
 
     submitted = st.form_submit_button("Get Estimate")
 
-# Generate the solar project estimate via Gemini
+# Calculate directly
 if submitted and location:
     state_data = df[df['State'].str.contains(location, case=False)].iloc[0]
-    
+
     if state_data is not None:
         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
-        with st.spinner("Generating solar estimate with Gemini..."):
-            response = model.generate_content(prompt_text)
-        
-        # Display structured output
-        st.subheader("🔹 Solar Project Estimate")
-        
-        estimated_data = response.text.strip().split("\n")
-        
-        system_size_kw = None
-        monthly_savings_rs = None
-        total_system_cost = None
-        payback_period_years = None
-        daily_output_kwh = None
-
-        for point in estimated_data:
-            if ":" in point:
-                try:
-                    key, value = point.split(":", 1)
-                    key = key.strip()
-                    value = value.strip()
-                    
-                    st.write(f"**{key}**: {value}")
-
-                    if "Estimated solar system size" in key or "Required solar system size" in key:
-                        system_size_kw = float(value.split()[0])
-                    if "Monthly savings" in key:
-                        monthly_savings_rs = float(value.split()[0])
-                    if "Total system cost" in key:
-                        total_system_cost = float(value.split()[0])
-                    if "Payback period" in key:
-                        payback_period_years = float(value.split()[0])
-
-                except ValueError:
-                    st.warning("There was an issue processing the response. Please try again.")
-
-        # Calculate Daily Output in kWh
-        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)
-            """)
+            system_size_kw = round(roof_size * ROOFTOP_CONVERSION_FACTOR, 2)
+            estimated_daily_output = round(system_size_kw * ghi, 2)
         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")
-            
-            fig = go.Figure(data=[ 
-                go.Bar(
-                    name="System Size (kW)",
-                    x=["Solar System Size", "Daily Output", "Total Cost", "Monthly Savings", "Payback Period"],
-                    y=[system_size_kw, daily_output_kwh , total_system_cost, monthly_savings_rs, payback_period_years],
-                    marker_color='#636EFA'
-                ),
-                go.Bar(
-                    name="Financials",
-                    x=["Solar System Size", "Daily Output", "Total Cost", "Monthly Savings", "Payback Period"],
-                    y=[0, 0, total_system_cost, monthly_savings_rs, payback_period_years],
-                    marker_color='#00CC96'
-                )
-            ])
-            
-            fig.update_layout(
-                barmode='group',
-                title="Comparison of Solar System Parameters",
-                yaxis_title="Values",
-                xaxis_title="Parameters"
+            system_size_kw = round(desired_kwh / (30 * ghi), 2)
+            estimated_daily_output = round(system_size_kw * ghi, 2)
+
+        total_system_cost = round(system_size_kw * solar_cost_per_kw, 2)
+        monthly_savings = round(estimated_daily_output * 30 * TARIFF_RATE, 2)
+        payback_period = round(total_system_cost / (monthly_savings * 12), 2)
+
+        # Display Results
+        st.subheader("🔹 Solar Project Estimate")
+        st.write(f"**Estimated solar system size in kW**: {system_size_kw}")
+        st.write(f"**Estimated daily solar output in kWh**: {estimated_daily_output}")
+        st.write(f"**Total system cost in ₹**: {total_system_cost}")
+        st.write(f"**Monthly savings in ₹**: {monthly_savings}")
+        st.write(f"**Payback period in years**: {payback_period}")
+
+        # Visual Summary
+        st.subheader("📊 Visual Summary")
+        fig = go.Figure(data=[ 
+            go.Bar(
+                name="System Parameters",
+                x=["System Size (kW)", "Daily Output (kWh)", "Total Cost (₹)", "Monthly Savings (₹)", "Payback (Years)"],
+                y=[system_size_kw, estimated_daily_output, total_system_cost, monthly_savings, payback_period],
+                marker_color='#636EFA'
             )
-            st.plotly_chart(fig, use_container_width=True)
+        ])
+        fig.update_layout(
+            title="Solar System Estimation Overview",
+            yaxis_title="Values",
+            xaxis_title="Parameters"
+        )
+        st.plotly_chart(fig, use_container_width=True)
+
+        st.info("Note: Tariff assumed ₹7/kWh. Actual payback may vary based on location, grid policy, and maintenance.")
 
     else:
-        st.error("Sorry, the location entered does not match any available data.")
+        st.error("State data not found. Please try a valid state.")
 else:
-    st.warning("Please fill out all fields to see your solar project estimate.")
+    st.warning("Please complete all fields to get your estimate.")