Update app.py

app.py

@@ -219,6 +219,19 @@ CUSTOM_CSS = """
 """
 
 def recommend_preamble_solution(deployment_type, existing_ai, monthly_budget):
+    """
+    Updated recommendation logic:
+    - If budget < $27,000/mo:
+      - For existing AI: Recommend Guardrails
+      - For new platform: Recommend SaaS
+    - If budget >= $27,000/mo or On-Premises: Recommend Enterprise
+    """
+    try:
+        budget_value = float(monthly_budget) if monthly_budget is not None else 0
+    except (ValueError, TypeError):
+        budget_value = 0
+
+    # Always recommend Enterprise for on-premises deployment
     if deployment_type == "On-Premises":
         return (
             "Enterprise",
@@ -231,25 +244,39 @@ def recommend_preamble_solution(deployment_type, existing_ai, monthly_budget):
             Starting at $27,000/month"""
         )
     
-    if existing_ai == "No, we need a full platform" and 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"""
-        )
-        
+    # For budgets under $27,000/month
+    if budget_value < 27000:
+        if existing_ai == "Yes, we have existing AI applications":
+            return (
+                "Guardrails",
+                """### ⚡ Guardrails
+                Perfect for integrating with existing AI:
+                - Quick API integration
+                - Usage-based pricing
+                - $0.005 per API call
+                - Flexible scaling options"""
+            )
+        else:
+            return (
+                "SaaS",
+                """### 🚀 SaaS Platform
+                Perfect for teams getting started with AI:
+                - Full platform features
+                - Quick cloud deployment
+                - Regular updates
+                - $50 per user/month"""
+            )
+    
+    # For budgets $27,000/month and above
     return (
-        "Guardrails",
-        """### ⚡ Guardrails
-        Perfect for integrating with existing AI:
-        - Quick API integration
-        - Usage-based pricing
-        - $0.005 per API call
-        - Flexible scaling options"""
+        "Enterprise",
+        """### 🏢 Enterprise License
+        Perfect for organizations that need:
+        - Full enterprise features
+        - Complete compliance controls
+        - Dedicated support team
+        - Custom integrations
+        Starting at $27,000/month"""
     )
 
 def calculate_roi(inputs):
@@ -435,7 +462,7 @@ def create_app():
             should_show = (
                 deployment_type != "On-Premises" and
                 (existing_ai == "Yes, we have existing AI applications" or
-                 budget_value <= 5000)
+                 budget_value < 27000)
             )
             return gr.update(visible=should_show)
 
@@ -473,4 +500,9 @@ def create_app():
 app = create_app()
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()
+
+
+
+
+