Update app.py

app.py

@@ -1,57 +1,64 @@
-# app.py
-
 import gradio as gr
 from model import score_opportunity
-from utils import validate_data
 
-def predict_deal(amount, stage, industry, lead_score, emails_last_7_days, meetings_last_30_days):
-    input_data = {
+def predict_deal(amount, close_date, stage, industry, lead_score, emails_last_7_days, meetings_last_30_days):
+    result = score_opportunity({
         "amount": amount,
+        "close_date": close_date,
         "stage": stage,
         "industry": industry,
         "lead_score": lead_score,
         "emails_last_7_days": emails_last_7_days,
         "meetings_last_30_days": meetings_last_30_days
-    }
-
-    if not validate_data(input_data):
-        return 0, 0.0, "Invalid", "⚠️ Please enter all fields correctly."
-
-    result = score_opportunity(input_data)
-    return result['score'], result['confidence'], result['risk'], result['recommendation']
+    })
 
-with gr.Blocks(title="AI Deal Qualification Engine") as demo:
-    gr.Markdown("""
-    # 🤖 AI-Powered Deal Qualification Engine  
-    _Predict B2B deal success using AI-based scoring._
-    """)
+    score = result["score"]
+    confidence = result["confidence"]
 
-    with gr.Row():
-        with gr.Column():
-            amount = gr.Number(label="💰 Deal Amount (INR/USD)")
-            stage = gr.Dropdown(
-                ["Prospecting", "Qualified", "Proposal", "Proposal/Price Quote", "Negotiation", "Closed Won", "Closed Lost"],
-                label="📊 Deal Stage"
-            )
-            industry = gr.Textbox(label="🏭 Industry")
+    if score >= 80 and confidence >= 0.85:
+        recommendation = "✅ Great potential. Proceed confidently with the deal."
+        risk = "Low"
+    elif 60 <= score < 80:
+        if confidence >= 0.75:
+            recommendation = "🟡 Moderate chance. Strengthen customer engagement."
+            risk = "Medium"
+        else:
+            recommendation = "🤔 50/50. Customer interest is unclear. Clarify further."
+            risk = "Medium"
+    elif score < 60 and (emails_last_7_days + meetings_last_30_days) >= 5:
+        recommendation = "🔍 Investigate – engagement high, but low interest shown."
+        risk = "High"
+    else:
+        recommendation = "⚠️ Low potential. Reassess or de-prioritize."
+        risk = "High"
 
-        with gr.Column():
-            lead_score = gr.Number(label="📈 Lead Score (0–100)")
-            emails_last_7_days = gr.Number(label="✉️ Emails in Last 7 Days")
-            meetings_last_30_days = gr.Number(label="📅 Meetings in Last 30 Days")
-
-    submit_btn = gr.Button("🔍 Predict")
-
-    with gr.Row():
-        score = gr.Number(label="🧮 Score (0–100)", interactive=False)
-        confidence = gr.Number(label="📊 Confidence (0–1)", interactive=False)
-        risk = gr.Textbox(label="🚦 Risk Level", interactive=False)
-        recommendation = gr.Textbox(label="🧠 Recommendation", lines=2, interactive=False)
+    return {
+        "Score": score,
+        "Confidence": confidence,
+        "Risk": risk,
+        "Recommendation": recommendation
+    }
 
-    submit_btn.click(
-        fn=predict_deal,
-        inputs=[amount, stage, industry, lead_score, emails_last_7_days, meetings_last_30_days],
-        outputs=[score, confidence, risk, recommendation]
-    )
+demo = gr.Interface(
+    fn=predict_deal,
+    inputs=[
+        gr.Number(label="Deal Amount"),
+        gr.Textbox(label="Close Date (YYYY-MM-DD)"),
+        gr.Dropdown(["Prospecting", "Qualification", "Proposal/Price Quote", "Negotiation/Review", "Closed Won", "Closed Lost"], label="Stage"),
+        gr.Textbox(label="Industry"),
+        gr.Slider(0, 100, step=1, label="Lead Score"),
+        gr.Number(label="Emails in Last 7 Days"),
+        gr.Number(label="Meetings in Last 30 Days"),
+    ],
+    outputs=[
+        gr.Number(label="Score"),
+        gr.Number(label="Confidence"),
+        gr.Textbox(label="Risk"),
+        gr.Textbox(label="Recommendation"),
+    ],
+    title="AI Deal Qualification Engine",
+    description="Enter opportunity details to get deal score and recommendation."
+)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()