Update app.py

app.py

@@ -1,97 +1,73 @@
 import gradio as gr
 import pandas as pd
 
-# Preamble brand colors from the image
-PREAMBLE_COLORS = {
-    'white': '#081423',     # Light background
-    'yellow': '#FFC700',    # Yellow accent
-    'blue': '#4556E4',      # Primary blue
-    'navy': '#1F2C6D',      # Navy blue
-    'black': '#ECF0F6'      # Dark background
+# Preamble brand colors from the brand guide
+BRAND_COLORS = {
+    'light_gray': '#081423',
+    'yellow': '#1F2C6D',
+    'blue': '#4556E4',
+    'navy': '#FFC700',
+    'dark_gray': '#ECF0F6'
 }
 
+# Default compliance data
+default_compliance_df = pd.DataFrame({
+    "Regulation": [
+        "SEC Reg S-P (Privacy)", 
+        "GLBA (Financial)",
+        "FINRA Rules",
+        "Bank Secrecy Act",
+        "Dodd-Frank Act"
+    ],
+    "Expected Violations": [1, 1, 1, 1, 1],
+    "Penalty": [100000, 100000, 150000, 200000, 250000],
+    "Attorney Cost": [50000, 45000, 75000, 100000, 125000]
+})
+
 CUSTOM_CSS = """
 .gradio-container {
     max-width: 1200px !important;
     margin: auto !important;
     padding: 2rem !important;
     background-color: #081423 !important;
-    color: #FFFFFF !important;
+    color: #ECF0F6 !important;
 }
 
-/* Header styling */
 .header {
-    background-color: #1F2C6D;
+    background-color: #4556E4;
     padding: 1.5rem;
     border-radius: 12px;
     margin-bottom: 2rem;
-    color: white;
 }
 
-/* Product selector styling */
 .product-selector {
-    background: linear-gradient(145deg, #4556E4, #1F2C6D);
+    background: #1F2C6D;
     border-radius: 12px;
     padding: 1.5rem;
     margin-bottom: 2rem;
-    color: white;
 }
 
-/* Results styling */
 .results-container {
-    background: #ECF0F6;
+    background: #4556E4;
     border-radius: 12px;
     padding: 1.5rem;
     margin-bottom: 2rem;
 }
 
-.result-card {
-    background-color: #FFFFFF;
-    border-radius: 8px;
-    padding: 1rem;
-    margin: 0.5rem;
-    color: #081423;
-}
-
-/* Button styling */
 .primary-button {
     background-color: #FFC700 !important;
     color: #081423 !important;
     font-weight: bold !important;
-    padding: 0.75rem 1.5rem !important;
-    border-radius: 8px !important;
-    transition: all 0.2s !important;
 }
 
-.primary-button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 4px 12px rgba(255, 199, 0, 0.3);
-}
-
-/* Tab styling */
-.tab-nav {
-    background-color: #1F2C6D !important;
-    border-radius: 8px !important;
-}
-
-.tab-nav button {
-    color: white !important;
-}
-
-.tab-nav button.selected {
-    background-color: #4556E4 !important;
-    color: white !important;
-}
-
-/* Input styling */
-.input-row {
-    background-color: rgba(69, 86, 228, 0.1);
+.result-card {
+    background-color: #ECF0F6;
+    color: #081423;
     border-radius: 8px;
     padding: 1rem;
-    margin-bottom: 1rem;
+    margin: 0.5rem;
 }
 
-/* Recommendation panel */
 .recommendation-panel {
     background-color: #FFC700;
     color: #081423;
@@ -102,10 +78,93 @@ CUSTOM_CSS = """
 }
 """
 
-# [Previous code for compliance_df and calculate_roi remains the same]
+def recommend_preamble_solution(deployment_type, existing_ai, monthly_budget):
+    """
+    Logic for recommending Preamble solution:
+    - If on-prem is needed -> Enterprise
+    - If no existing AI:
+        - If budget > 5000 -> SaaS Platform
+        - If budget <= 5000 -> Guardrails
+    - If has existing AI -> Recommend Guardrails first
+    """
+    if deployment_type == "On-Premises":
+        return ("Enterprise", """### 🏢 Enterprise License
+        **Perfect for organizations that need:**
+        - On-premises deployment
+        - Complete compliance controls
+        - Dedicated support team
+        - Custom integrations
+        Starting at $27,000/month""")
+    
+    if existing_ai == "No, we need a full platform":
+        if monthly_budget > 5000:
+            return ("SaaS", """### 🚀 SaaS Platform
+            **Perfect for teams getting started with AI:**
+            - Full platform features
+            - Quick cloud deployment
+            - Regular updates
+            - $50 per user/month""")
+        else:
+            return ("Guardrails", """### ⚡ Guardrails
+            **Start with our foundational offering:**
+            - Basic AI safety controls
+            - API-first implementation
+            - $0.005 per API call
+            - Usage-based pricing""")
+    
+    return ("Guardrails", """### ⚡ Guardrails Recommended
+    **Perfect for existing AI applications:**
+    - Quick integration with your AI
+    - API-first implementation
+    - $0.005 per API call
+    - Usage-based pricing""")
+
+def calculate_roi(inputs):
+    [num_employees, hours_saved_per_week, hourly_wage,
+     initial_platform_cost, num_ai_hires, avg_salary,
+     ai_maintenance_cost, ai_security_cost,
+     revenue_increase,
+     monthly_budget, deployment_type, existing_ai, estimated_api_calls,
+     compliance_data] = inputs
+
+    # Calculate savings and costs
+    total_hours_saved = num_employees * hours_saved_per_week * 52
+    labor_cost_savings = total_hours_saved * hourly_wage
+
+    # Compliance savings
+    compliance_savings = sum(
+        row["Expected Violations"] * (row["Penalty"] + row["Attorney Cost"])
+        for _, row in compliance_data.iterrows()
+    )
+
+    # Total benefits
+    total_benefits = labor_cost_savings + revenue_increase + compliance_savings
+
+    # In-house costs
+    total_costs_build = (
+        initial_platform_cost + 
+        (num_ai_hires * avg_salary) + 
+        ai_maintenance_cost + 
+        ai_security_cost
+    )
+
+    # Preamble costs
+    solution_type, _ = recommend_preamble_solution(deployment_type, existing_ai, monthly_budget)
+    if solution_type == "Enterprise":
+        total_costs_preamble = 27000 * 12
+    elif solution_type == "SaaS":
+        total_costs_preamble = num_employees * 50 * 12
+    else:  # Guardrails
+        total_costs_preamble = estimated_api_calls * 0.005 * 12
+
+    # ROI calculations
+    roi_build = ((total_benefits - total_costs_build) / total_costs_build) * 100 if total_costs_build else 0
+    roi_preamble = ((total_benefits - total_costs_preamble) / total_costs_preamble) * 100 if total_costs_preamble else 0
+
+    return roi_build, roi_preamble, total_benefits, total_costs_build, total_costs_preamble
 
 def create_app():
-    with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="orange")) as roi_app:
+    with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="blue")) as roi_app:
         # Header
         with gr.Row(elem_classes="header"):
             gr.Markdown(
@@ -113,90 +172,142 @@ def create_app():
                 "[Visit Preamble.com](https://preamble.com) to learn more"
             )
         
-        # Product Selection Section - Moved to top
+        # Product Selection
         with gr.Row(elem_classes="product-selector"):
             with gr.Column():
-                gr.Markdown("## 🎯 Find Your Ideal Preamble Solution")
-                with gr.Row(elem_classes="input-row"):
-                    deployment_type = gr.Radio(
-                        ["Cloud/SaaS", "On-Premises"],
-                        label="How would you like to deploy?",
-                        value="Cloud/SaaS"
-                    )
-                    existing_ai = gr.Radio(
-                        ["Yes, we have existing AI applications", "No, we need a full platform"],
-                        label="Do you have existing AI applications?",
-                        value="No, we need a full platform"
-                    )
-                
-                with gr.Row(elem_classes="input-row"):
-                    monthly_budget = gr.Number(
-                        label="Monthly Budget ($)",
-                        value=5000,
-                        info="What's your monthly budget for AI governance?"
-                    )
-                    estimated_api_calls = gr.Number(
-                        label="Estimated Monthly API Calls",
-                        value=10000,
-                        info="Required for Guardrails pricing"
-                    )
-                
-                recommendation_box = gr.Markdown(
-                    "Complete the information above to get your personalized solution recommendation",
-                    elem_classes="recommendation-panel"
+                gr.Markdown("## Find Your Ideal Preamble Solution")
+                deployment_type = gr.Radio(
+                    ["Cloud/SaaS", "On-Premises"],
+                    label="Deployment Preference",
+                    value="Cloud/SaaS"
+                )
+                existing_ai = gr.Radio(
+                    ["Yes, we have existing AI applications", "No, we need a full platform"],
+                    label="AI Implementation Status",
+                    value="No, we need a full platform"
+                )
+                monthly_budget = gr.Number(
+                    label="Monthly Budget ($)",
+                    value=5000
+                )
+                estimated_api_calls = gr.Number(
+                    label="Estimated Monthly API Calls",
+                    value=10000,
+                    visible=False
                 )
+                recommendation_box = gr.Markdown(elem_classes="recommendation-panel")
 
         # Results section
         with gr.Row(elem_classes="results-container"):
             with gr.Column():
                 build_roi_box = gr.Markdown(
-                    "### 🏗️ Building In-House\n**ROI:** 0%\n**Total Costs:** $0",
+                    "### Building In-House\n**ROI:** --%\n**Total Costs:** $0",
                     elem_classes="result-card"
                 )
             with gr.Column():
                 preamble_roi_box = gr.Markdown(
-                    "### 🚀 Using Preamble\n**ROI:** 0%\n**Total Costs:** $0",
+                    "### Using Preamble\n**ROI:** --%\n**Total Costs:** $0",
                     elem_classes="result-card"
                 )
 
-        # ROI Calculator Section
-        gr.Markdown("## 📈 Calculate Your ROI")
-        
-        # [Rest of the tabs and calculator remain the same]
-        
-        # Update functions
-        def update_recommendation(monthly_budget, deployment_type, existing_ai, num_employees):
-            if deployment_type == "On-Premises":
-                return """### 🏢 Enterprise (On-Premises)
-                **Perfect for:** Large enterprises with complex compliance needs
-                - Full platform deployment
-                - Complete compliance controls
-                - Dedicated support team
-                - Custom integrations
-                Starting at $27,000/month"""
+        # Calculator Inputs
+        with gr.Tabs():
+            with gr.Tab("Organization Details"):
+                num_employees = gr.Slider(
+                    label="Number of Employees",
+                    minimum=1, maximum=1000, value=215
+                )
+                hours_saved_per_week = gr.Slider(
+                    label="Hours Saved per Week per Employee",
+                    minimum=0, maximum=40, value=6.3
+                )
+                hourly_wage = gr.Slider(
+                    label="Average Hourly Wage ($)",
+                    minimum=10, maximum=200, value=62
+                )
+
+            with gr.Tab("Build Costs"):
+                initial_platform_cost = gr.Number(label="Initial Development Cost ($)", value=100000)
+                num_ai_hires = gr.Number(label="Number of AI Hires", value=1)
+                avg_salary = gr.Number(label="Average Annual Salary ($)", value=150000)
+                ai_maintenance_cost = gr.Number(label="Annual Maintenance ($)", value=20000)
+                ai_security_cost = gr.Number(label="Security & Compliance ($)", value=10000)
+
+            with gr.Tab("Compliance"):
+                compliance_data = gr.Dataframe(
+                    value=default_compliance_df,
+                    headers=["Regulation", "Expected Violations", "Penalty", "Attorney Cost"],
+                    datatype="pandas"
+                )
+
+            with gr.Tab("Benefits"):
+                revenue_increase = gr.Number(
+                    label="Estimated Annual Revenue Increase ($)",
+                    value=50000
+                )
+
+        calculate_button = gr.Button("Calculate ROI", elem_classes="primary-button")
+
+        def update_results(*inputs):
+            roi_build, roi_preamble, benefits, costs_build, costs_preamble = calculate_roi(inputs)
             
-            if existing_ai == "Yes, we have existing AI applications" and monthly_budget <= 5000:
-                return """### ⚡ Guardrails
-                **Perfect for:** Teams with existing AI applications
-                - API-first implementation
-                - $0.005 per API call
-                - Quick integration
-                - Usage-based pricing
-                Estimated monthly cost: ${:,.2f}""".format(estimated_api_calls * 0.005)
+            build_text = (
+                f"### 🏗️ Building In-House\n"
+                f"**ROI:** {roi_build:,.1f}%\n"
+                f"**Total Costs:** ${costs_build:,.2f}\n"
+                f"**Total Benefits:** ${benefits:,.2f}"
+            )
             
-            return """### 🚀 SaaS Platform
-            **Perfect for:** Teams getting started with AI
-            - Full platform features
-            - Quick cloud deployment
-            - Regular updates
-            - Simple per-user pricing
-            $50 per user/month (estimated: ${:,.2f}/month)""".format(num_employees * 50)
+            preamble_text = (
+                f"### 🚀 Using Preamble\n"
+                f"**ROI:** {roi_preamble:,.1f}%\n"
+                f"**Total Costs:** ${costs_preamble:,.2f}\n"
+                f"**Total Benefits:** ${benefits:,.2f}"
+            )
+            
+            return build_text, preamble_text
+
+        def update_api_calls_visibility(existing_ai, deployment_type):
+            # Show API calls input only for Guardrails candidates
+            should_show = (
+                deployment_type != "On-Premises" and 
+                (existing_ai == "Yes, we have existing AI applications" or 
+                 monthly_budget <= 5000)
+            )
+            return gr.update(visible=should_show)
+
+        # Update recommendation when inputs change
+        for input_elem in [deployment_type, existing_ai, monthly_budget]:
+            input_elem.change(
+                fn=lambda d, e, b: recommend_preamble_solution(d, e, b)[1],
+                inputs=[deployment_type, existing_ai, monthly_budget],
+                outputs=[recommendation_box]
+            )
+            
+            # Update API calls visibility
+            if input_elem in [deployment_type, existing_ai]:
+                input_elem.change(
+                    fn=update_api_calls_visibility,
+                    inputs=[existing_ai, deployment_type],
+                    outputs=[estimated_api_calls]
+                )
 
-        # [Rest of the event handlers remain the same]
+        # Connect calculate button
+        calculate_button.click(
+            fn=update_results,
+            inputs=[
+                num_employees, hours_saved_per_week, hourly_wage,
+                initial_platform_cost, num_ai_hires, avg_salary,
+                ai_maintenance_cost, ai_security_cost,
+                revenue_increase,
+                monthly_budget, deployment_type, existing_ai, estimated_api_calls,
+                compliance_data
+            ],
+            outputs=[build_roi_box, preamble_roi_box]
+        )
 
-    return roi_app
+        return roi_app
 
-# Create and launch the app
 app = create_app()
 
 if __name__ == "__main__":