Update app.py

app.py

@@ -53,11 +53,11 @@ def calculate_confidence(ai_score):
 def calculate_risk(ai_score, confidence, emails, meetings):
     # Make sure to map the value to the Salesforce picklist options
     if ai_score > 0.75 and confidence > 75:
-        return "Low"  # Update to match Salesforce picklist value
+        return "Low"
     elif ai_score > 0.5:
-        return "Medium"  # Assuming "Medium" is another valid value
+        return "Medium"
     else:
-        return "High"  # Update to match Salesforce picklist value
+        return "High"
 
 def generate_recommendation(stage, emails, meetings, risk):
     # Example recommendation generation
@@ -117,8 +117,20 @@ def run_engine(amount, stage, industry, emails, meetings, close_gap, feedback=""
 
 # Gradio UI
 with gr.Blocks(title="AI Deal Qualification Engine") as app:
-    gr.Markdown("""
-    <h1 style="text-align:center;">🤖 AI-Powered Deal Qualification Engine</h1>
+    # Custom header with logout button
+    gr.HTML("""
+    <div style="display: flex; justify-content: space-between; align-items: center; 
+                background-color: #1a1a1a; color: white; padding: 12px 20px; border-radius: 8px;">
+        <div>
+            <h1 style="margin: 0; font-size: 22px;">🤖 AI-Powered Deal Qualification Engine</h1>
+        </div>
+        <div>
+            <button style="background-color: black; color: white; border: none; 
+                           padding: 8px 16px; border-radius: 6px; cursor: pointer;">
+                Logout
+            </button>
+        </div>
+    </div>
     <p style="text-align:center;">Intelligently qualify sales deals using engagement and pipeline signals.</p>
     """, elem_id="header")