Update app.py

app.py

@@ -1,47 +1,40 @@
 import gradio as gr
-from mock_model import predict, get_lead_score_for_stage
-
-STAGES = ["Prospecting", "Proposal/Price Quote", "Negotiation", "Closed Won", "Closed Lost"]
-
-def run_app(amount, stage, industry, emails, meetings, close_gap):
-    lead_score = get_lead_score_for_stage(stage)
-    input_data = {
-        "amount": amount,
-        "stage": stage,
-        "industry": industry,
-        "lead_score": lead_score,
-        "emails_last_7_days": emails,
-        "meetings_last_30_days": meetings,
-        "close_date_gap": close_gap
-    }
-    result = predict(input_data, None, None, None)
-    return (
-        lead_score,
-        result["score"],
-        result["confidence"],
-        result["risk"],
-        result["recommendation"]
-    )
-
-demo = gr.Interface(
-    fn=run_app,
-    title="AI-Powered Deal Qualification Engine",
-    inputs=[
-        gr.Number(label="Amount (USD)", value=50000),
-        gr.Dropdown(STAGES, label="Stage"),
-        gr.Textbox(label="Industry", value="Software"),
-        gr.Number(label="Emails in Last 7 Days", value=3),
-        gr.Number(label="Meetings in Last 30 Days", value=2),
-        gr.Number(label="Close Date Gap (days)", value=10)
-    ],
-    outputs=[
-        gr.Number(label="Generated Lead Score"),
-        gr.Number(label="AI Score (0–100)"),
-        gr.Number(label="Confidence (0–1)"),
-        gr.Textbox(label="Risk Level"),
-        gr.Textbox(label="Recommendation")
-    ]
-)
-
-if __name__ == "__main__":
-    demo.launch()
+from scorer import get_lead_score, calculate_score, calculate_confidence, calculate_risk
+from recommender import generate_recommendation
+
+def run_engine(amount, stage, industry, emails, meetings, close_gap):
+    lead_score = get_lead_score(stage, emails, meetings, close_gap)
+    ai_score = calculate_score(lead_score, emails, meetings, close_gap)
+    confidence = calculate_confidence(ai_score)
+    risk = calculate_risk(ai_score, confidence, emails, meetings)
+    recommendation = generate_recommendation(stage, emails, meetings, risk)
+
+    return lead_score, ai_score, confidence, risk, recommendation
+
+with gr.Blocks(title="B2B Deal Qualification Engine") as demo:
+    gr.Markdown("## 🤖 AI-Powered Deal Qualification Engine")
+
+    with gr.Row():
+        amount = gr.Number(label="Deal Amount (USD)", value=50000)
+        stage = gr.Dropdown(["Prospecting", "Proposal/Price Quote", "Negotiation", "Closed Won", "Closed Lost"], label="Stage")
+        industry = gr.Textbox(label="Industry", value="Software")
+
+    with gr.Row():
+        emails = gr.Number(label="Emails in Last 7 Days", value=3)
+        meetings = gr.Number(label="Meetings in Last 30 Days", value=2)
+        close_gap = gr.Number(label="Days Until Close Date", value=10)
+
+    submit = gr.Button("Score Deal")
+
+    with gr.Row():
+        lead_score_out = gr.Number(label="Generated Lead Score")
+        ai_score_out = gr.Number(label="AI Score (0–100)")
+        confidence_out = gr.Number(label="Confidence Level (0–1)")
+        risk_out = gr.Textbox(label="Risk Assessment")
+        reco_out = gr.Textbox(label="AI Recommendation")
+
+    submit.click(fn=run_engine,
+                 inputs=[amount, stage, industry, emails, meetings, close_gap],
+                 outputs=[lead_score_out, ai_score_out, confidence_out, risk_out, reco_out])
+
+demo.launch()