Update app.py

app.py

@@ -1,5 +1,8 @@
+# app.py
+
 import gradio as gr
 from model import score_opportunity
+from utils import validate_data
 
 def predict_deal(amount, stage, industry, lead_score, email_count, meeting_count, close_date_gap):
     input_data = {
@@ -11,27 +14,47 @@ def predict_deal(amount, stage, industry, lead_score, email_count, meeting_count
         "meeting_count": meeting_count,
         "close_date_gap": close_date_gap
     }
+
+    if not validate_data(input_data):
+        return 0, "Invalid", "⚠️ Please enter valid numeric values and select a stage."
+
     result = score_opportunity(input_data)
     return result['score'], result['risk'], result['recommendation']
 
 with gr.Blocks(title="AI Deal Qualification Engine") as demo:
-    gr.Markdown("# 🤖 AI-Powered B2B Deal Qualification Engine")
+    gr.Markdown(
+        """
+        # 🤖 AI-Powered Deal Qualification Engine
+        _Estimate the quality of your B2B opportunity using AI._  
+        Enter opportunity details below to predict:  
+        ✅ **Deal Score**  
+        ✅ **Risk Level**  
+        ✅ **AI Recommendation**
+        """
+    )
 
     with gr.Row():
-        amount = gr.Number(label="Amount")
-        stage = gr.Dropdown(["Prospecting", "Qualified", "Proposal", "Negotiation", "Closed Won", "Closed Lost"], label="Stage")
-        industry = gr.Textbox(label="Industry")
-        lead_score = gr.Number(label="Lead Score")
-        email_count = gr.Number(label="Email Count")
-        meeting_count = gr.Number(label="Meeting Count")
-        close_date_gap = gr.Number(label="Close Date Gap (days)")
+        with gr.Column():
+            amount = gr.Number(label="💰 Deal Amount (INR/USD)", scale=1)
+            stage = gr.Dropdown(
+                ["Prospecting", "Qualified", "Proposal", "Negotiation", "Closed Won", "Closed Lost"],
+                label="📊 Deal Stage"
+            )
+            industry = gr.Textbox(label="🏭 Industry (e.g., IT, Healthcare)")
 
-    submit_btn = gr.Button("Predict Deal Quality")
+        with gr.Column():
+            lead_score = gr.Number(label="📈 Lead Score (0–100)")
+            email_count = gr.Number(label="✉️ Email Interactions")
+            meeting_count = gr.Number(label="📅 Meeting Count")
+            close_date_gap = gr.Number(label="📆 Days Until Expected Close")
+
+    with gr.Row():
+        submit_btn = gr.Button("🔍 Predict Deal")
 
     with gr.Row():
-        score = gr.Number(label="Score (0–100)", interactive=False)
-        risk = gr.Textbox(label="Risk Level", interactive=False)
-        recommendation = gr.Textbox(label="AI Recommendation", lines=2, interactive=False)
+        score = gr.Number(label="🧮 Deal Score (0–100)", interactive=False)
+        risk = gr.Textbox(label="🚦 Risk Level", interactive=False)
+        recommendation = gr.Textbox(label="🧠 AI Recommendation", lines=2, interactive=False)
 
     submit_btn.click(fn=predict_deal,
                      inputs=[amount, stage, industry, lead_score, email_count, meeting_count, close_date_gap],