Update app.py

app.py

@@ -2,18 +2,15 @@ import streamlit as st
 import pandas as pd
 import google.generativeai as genai
 import os
-from io import StringIO
-import csv
 from dotenv import load_dotenv
 
-
-# Load environment variables from .env file
+# Load environment variables
 load_dotenv()
 
-# Set page configuration first
+# Set page configuration
 st.set_page_config(page_title="AI-based Solar Project Estimation Tool", layout="centered")
 
-# Initialize Gemini with the API key loaded from the .env file
+# Initialize Gemini
 api_key = os.getenv("GOOGLE_API_KEY")
 if api_key:
     genai.configure(api_key=api_key)
@@ -30,67 +27,85 @@ def load_data():
 
 df = load_data()
 
-# UI - Form for user input
+# Solar Calculation Function (not relying on Gemini for calculation)
+def calculate_solar_estimate(roof_size, monthly_bill, electricity_price, ghi, cost_per_kw):
+    daily_consumption_inr = monthly_bill / 30
+    daily_consumption_kwh = daily_consumption_inr / electricity_price
+
+    # Assume system size based on consumption
+    estimated_system_size_kw = 3  # Fixed for now as realistic estimate
+    peak_sun_hours = ghi  # Use GHI as sun hours (approximation)
+    
+    daily_solar_output_kwh = estimated_system_size_kw * peak_sun_hours * 0.75  # considering derating factor
+    
+    total_system_cost = estimated_system_size_kw * cost_per_kw
+    
+    monthly_generation_kwh = daily_solar_output_kwh * 30
+    monthly_savings_inr = monthly_generation_kwh * electricity_price
+    
+    annual_savings_inr = monthly_savings_inr * 12
+    payback_period_years = total_system_cost / annual_savings_inr
+
+    return {
+        "Estimated solar system size in kW": round(estimated_system_size_kw, 2),
+        "Estimated daily solar output in kWh": round(daily_solar_output_kwh, 2),
+        "Total system cost in ₹": int(total_system_cost),
+        "Monthly savings in ₹": int(monthly_savings_inr),
+        "Payback period in years": round(payback_period_years, 2)
+    }
+
+# UI - Form
 st.title("AI-based Solar Project Estimation Tool")
-st.write("### Enter Your Details Below:")
+st.write("### Enter Your Details:")
 
 with st.form("solar_form"):
     state_options = df['State'].dropna().unique()
     
     location = st.selectbox("Select your State", options=sorted(state_options))
     roof_size = st.number_input("Enter your roof size (in sq meters)", min_value=1)
-    electricity_bill = st.number_input("Enter your monthly electricity bill (₹)", min_value=0)
+    monthly_bill = st.number_input("Enter your monthly electricity bill (₹)", min_value=0)
     
     submitted = st.form_submit_button("Get Estimate")
 
-# Build the prompt for Gemini
-def build_prompt(location, roof_size, electricity_bill, ghi, solar_cost_per_kw):
-    prompt = f"""
-    Estimate the solar system for the location '{location}' based on the following details:
-    - Roof size: {roof_size} sq meters
-    - Monthly electricity bill: ₹{electricity_bill}
-    - Average GHI (solar radiation) for {location}: {ghi} kWh/m²/day
-    - Solar system cost per kW in {location}: ₹{solar_cost_per_kw}
-    Provide the following:
-    1. Estimated solar system size in kW
-    2. Estimated daily solar output in kWh
-    3. Total system cost in ₹
-    4. Monthly savings in ₹
-    5. Payback period in years
-    """
-    return prompt
-
-# Generate the solar project estimate via Gemini
-if submitted and location and roof_size > 0 and electricity_bill >= 0:
+if submitted and location and roof_size > 0 and monthly_bill >= 0:
     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 (₹)']
+        electricity_price = 8  # Assume ₹8/kWh
         
-        prompt_text = build_prompt(location, roof_size, electricity_bill, ghi, solar_cost_per_kw)
+        # Calculate estimates
+        estimates = calculate_solar_estimate(roof_size, monthly_bill, electricity_price, ghi, solar_cost_per_kw)
+
+        # Build clean prompt for Gemini to verify the calculation
+        prompt = f"""
+        ONLY output these 5 points based on inputs:
+        1. Estimated solar system size in kW
+        2. Estimated daily solar output in kWh
+        3. Total system cost in ₹
+        4. Monthly savings in ₹
+        5. Payback period in years
         
-        # Call Gemini API once for all the batch generation
-        with st.spinner("Generating solar estimate with Gemini..."):
-            response = model.generate_content(prompt_text)
+        Roof size = {roof_size} sq meters
+        Monthly bill = ₹{monthly_bill}
+        GHI = {ghi} kWh/m²/day
+        Solar system cost per kW = ₹{solar_cost_per_kw}
         
-        # Display structured output with only the requested points
+        Use no description. Only numeric values clearly.
+        """
+
+        # Call Gemini API (just to double-check/validate if you want)
+        with st.spinner("Generating final estimate..."):
+            gemini_response = model.generate_content(prompt)
+            final_response = gemini_response.text.strip()
+
+        # Display calculated values directly (without trusting Gemini text output)
         st.subheader("Solar Project Estimate")
-        
-        # Break down the response into structured points
-        estimated_data = response.text.strip().split("\n")
-        
-        # Display only the required points: system size, cost, savings, and payback period
-        for point in estimated_data:
-            if ":" in point:  # Only process lines with a colon
-                try:
-                    # Extract the value after the colon
-                    key, value = point.split(":", 1)  # Split into two parts only
-                    st.write(f"{key.strip()}: {value.strip()}")
-                except ValueError:
-                    # Handle cases where the line doesn't split into two parts
-                    st.warning("There was an issue processing the response.")
+
+        for key, value in estimates.items():
+            st.write(f"{key}: {value}")
     else:
-        st.error("Sorry, the location entered does not match any available data.")
+        st.error("Location data not found. Please select a valid state.")
 else:
-    st.warning("Please fill out all fields to see your solar project estimate.")
+    st.warning("Please fill all the fields.")