Update app.py

app.py

@@ -4,20 +4,60 @@ import pandas as pd
 from calculations import calculate_roi
 from ai_synopsis import generate_synopsis
 from pdf_generator import generate_pdf
+from email_sender import send_email
+
+
+# ====================================
+# PAGE CONFIG
+# ====================================
 
 st.set_page_config(
     page_title="AI ROI Calculator",
     layout="wide"
 )
 
+
+# ====================================
+# TITLE
+# ====================================
+
 st.title("AI ROI Calculator")
 
-st.markdown("Generate ROI savings reports with AI-powered synopsis")
+st.markdown(
+    "Generate ROI savings reports with AI-powered synopsis"
+)
 
 st.divider()
 
+
+# ====================================
+# CUSTOMER DETAILS
+# ====================================
+
+st.subheader("Customer Details")
+
+customer_name = st.text_input("Full Name")
+
+customer_email = st.text_input("Email Address")
+
+company_name = st.text_input("Company Name")
+
+phone_number = st.text_input("Phone Number")
+
+st.divider()
+
+
+# ====================================
+# MAIN LAYOUT
+# ====================================
+
 col1, col2 = st.columns(2)
 
+
+# ====================================
+# LEFT COLUMN
+# ====================================
+
 with col1:
 
     st.subheader("Business Inputs")
@@ -82,6 +122,11 @@ with col1:
         value=1500,
     )
 
+
+# ====================================
+# RIGHT COLUMN
+# ====================================
+
 with col2:
 
     st.subheader("ROI Results")
@@ -109,42 +154,105 @@ with col2:
         value=f"INR {roi_data['Total Savings']}"
     )
 
+
 st.divider()
 
+
+# ====================================
+# GENERATE REPORT BUTTON
+# ====================================
+
 if st.button("Generate AI Report"):
 
-    with st.spinner("Generating AI Synopsis..."):
-
-        user_data = {
-            "Industry": industry,
-            "Number of Assets": num_assets,
-            "Average Asset Value": asset_value,
-            "Employees": num_employees,
-        }
-
-        ai_data = {
-            "industry": industry,
-            "num_assets": num_assets,
-            "asset_value": asset_value,
-            "num_employees": num_employees,
-            "total_savings": roi_data['Total Savings'],
-        }
-
-        synopsis = generate_synopsis(ai_data)
-
-        st.subheader("AI Generated Synopsis")
-
-        st.write(synopsis)
-
-        pdf = generate_pdf(
-            user_data,
-            roi_data,
-            synopsis,
-        )
-
-        st.download_button(
-            label="Download Report",
-            data=pdf,
-            file_name="ai_roi_report.pdf",
-            mime="application/pdf",
-        )
+    # Validation
+    if (
+        customer_name == ""
+        or customer_email == ""
+        or company_name == ""
+        or phone_number == ""
+    ):
+
+        st.error("Please fill all customer details.")
+
+    else:
+
+        with st.spinner("Generating AI Synopsis..."):
+
+            # ====================================
+            # USER DATA
+            # ====================================
+
+            user_data = {
+                "Customer Name": customer_name,
+                "Customer Email": customer_email,
+                "Company Name": company_name,
+                "Phone Number": phone_number,
+                "Industry": industry,
+                "Number of Assets": num_assets,
+                "Average Asset Value": asset_value,
+                "Employees": num_employees,
+            }
+
+            # ====================================
+            # AI DATA
+            # ====================================
+
+            ai_data = {
+                "industry": industry,
+                "num_assets": num_assets,
+                "asset_value": asset_value,
+                "num_employees": num_employees,
+                "total_savings": roi_data['Total Savings'],
+            }
+
+            # ====================================
+            # GENERATE AI SYNOPSIS
+            # ====================================
+
+            synopsis = generate_synopsis(ai_data)
+
+            st.subheader("AI Generated Synopsis")
+
+            st.write(synopsis)
+
+            # ====================================
+            # GENERATE PDF
+            # ====================================
+
+            pdf = generate_pdf(
+                user_data,
+                roi_data,
+                synopsis,
+            )
+
+            # ====================================
+            # SEND EMAIL TO COMPANY
+            # ====================================
+
+            email_status = send_email(
+                user_data,
+                roi_data
+            )
+
+            if email_status:
+
+                st.success(
+                    "Lead details sent successfully!"
+                )
+
+            else:
+
+                st.error(
+                    "Failed to send lead email."
+                )
+
+            # ====================================
+            # DOWNLOAD BUTTON
+            # ====================================
+
+            st.download_button(
+                label="Download Report",
+                data=pdf,
+                file_name="ai_roi_report.pdf",
+                mime="application/pdf",
+            )