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 = {
        "amount": amount,
        "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 valid numeric values and select a stage."

    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
    _Estimate the quality of your B2B opportunity using AI._

    Enter opportunity details below to predict:
    ✅ Deal Score  
    ✅ Risk Level  
    ✅ Confidence  
    ✅ AI Recommendation
    """)

    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 (e.g., Software, Healthcare)")

        with gr.Column():
            lead_score = gr.Number(label="📈 Lead Score (0–100)")
            emails_last_7_days = gr.Number(label="✉️ Emails Last 7 Days")
            meetings_last_30_days = gr.Number(label="📅 Meetings Last 30 Days")

    submit_btn = gr.Button("🔍 Predict Deal")

    with gr.Row():
        score = gr.Number(label="🧮 Deal 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="🧠 AI Recommendation", lines=2, interactive=False)

    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.launch()