app.py

from flask import Flask, render_template, request, jsonify

app = Flask(__name__)

# Enhanced Knowledge base of Stacks with Pros/Cons and Budget info
STACKS = {
    "MERN": {
        "name": "MERN Stack (MongoDB, Express, React, Node)",
        "pros": [
            "Single language (JavaScript) everywhere",
            "Huge community and ecosystem",
            "Great for JSON data and APIs",
            "Fast development with reusable components",
            "Excellent for real-time applications"
        ],
        "cons": [
            "Can be complex for beginners",
            "Heavy for simple applications",
            "SEO challenges (though Next.js helps)",
            "Rapid ecosystem changes"
        ],
        "budget": "medium",
        "hosting_cost": "$10-50/month",
        "learning_curve": "medium",
        "best_for": ["realtime", "spa", "api"]
    },
    "Django": {
        "name": "Django Stack (Python, SQL, Django)",
        "pros": [
            "Batteries included framework",
            "Secure by default",
            "Excellent ORM and admin panel",
            "Great for rapid prototyping",
            "Strong community and documentation"
        ],
        "cons": [
            "Monolithic structure",
            "Can be overkill for simple projects",
            "Steeper learning curve",
            "Less flexible than microframeworks"
        ],
        "budget": "low",
        "hosting_cost": "$5-30/month",
        "learning_curve": "high",
        "best_for": ["cms", "enterprise", "data-heavy"]
    },
    "LAMP": {
        "name": "LAMP Stack (Linux, Apache, MySQL, PHP)",
        "pros": [
            "Extremely cheap hosting options",
            "Easy deployment and setup",
            "Mature and stable ecosystem",
            "Great for traditional web apps",
            "Huge talent pool"
        ],
        "cons": [
            "Inconsistent language syntax",
            "Performance issues at scale",
            "Considered outdated by some",
            "Security requires more attention"
        ],
        "budget": "low",
        "hosting_cost": "$3-15/month",
        "learning_curve": "low",
        "best_for": ["simple", "cms", "traditional"]
    },
    "Next.js": {
        "name": "Next.js Stack (React, Node, Vercel)",
        "pros": [
            "Excellent SEO with SSR/SSG",
            "Great developer experience",
            "Built-in optimization",
            "Easy deployment",
            "Modern React patterns"
        ],
        "cons": [
            "Vendor lock-in with Vercel",
            "Can be complex for simple sites",
            "Rapid updates and changes",
            "Learning curve for beginners"
        ],
        "budget": "medium",
        "hosting_cost": "$0-100/month",
        "learning_curve": "medium",
        "best_for": ["spa", "seo-critical", "modern"]
    }
}

def calculate_stack_score(stack_data, project_type, team_skill, budget):
    """Calculate a score for how well a stack matches the requirements"""
    score = 0
    
    # Project type matching
    if project_type in stack_data["best_for"]:
        score += 3
    
    # Budget matching
    budget_scores = {"low": 3, "medium": 2, "high": 1}
    if budget == "low" and stack_data["budget"] == "low":
        score += 3
    elif budget == "medium" and stack_data["budget"] in ["low", "medium"]:
        score += 2
    elif budget == "high":
        score += 1
    
    # Team skill matching
    skill_stack_match = {
        "js": ["MERN", "Next.js"],
        "python": ["Django"],
        "php": ["LAMP"],
        "mixed": ["LAMP", "Django"]
    }
    
    stack_name = stack_data["name"].split()[0]
    if stack_name in skill_stack_match.get(team_skill, []):
        score += 2
    
    return score

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/compare', methods=['POST'])
def compare():
    data = request.json
    project_type = data.get('projectType')
    team_skill = data.get('teamSkill')
    budget = data.get('budget', 'medium')
    
    # Calculate scores for all stacks
    stack_scores = []
    for stack_key, stack_data in STACKS.items():
        score = calculate_stack_score(stack_data, project_type, team_skill, budget)
        stack_scores.append((stack_key, stack_data, score))
    
    # Sort by score (highest first)
    stack_scores.sort(key=lambda x: x[2], reverse=True)
    
    # Get top 2 recommendations
    primary = stack_scores[0]
    secondary = stack_scores[1]
    
    # Generate intelligent verdict
    verdict = generate_verdict(primary, secondary, project_type, budget)
    
    response = {
        "optionA": {
            **primary[1],
            "score": primary[2],
            "match_reason": get_match_reason(primary[1], project_type, team_skill, budget)
        },
        "optionB": {
            **secondary[1],
            "score": secondary[2],
            "match_reason": get_match_reason(secondary[1], project_type, team_skill, budget)
        },
        "verdict": verdict
    }
    
    return jsonify(response)

def get_match_reason(stack_data, project_type, team_skill, budget):
    """Generate a reason why this stack matches the requirements"""
    reasons = []
    
    if project_type in stack_data["best_for"]:
        reasons.append(f"Perfect for {project_type} projects")
    
    if budget == "low" and stack_data["budget"] == "low":
        reasons.append("Budget-friendly option")
    
    return " • ".join(reasons) if reasons else "Good general-purpose choice"

def generate_verdict(primary, secondary, project_type, budget):
    """Generate an intelligent verdict based on the comparison"""
    primary_name = primary[1]["name"].split()[0]
    secondary_name = secondary[1]["name"].split()[0]
    
    if primary[2] > secondary[2] + 1:
        return f"{primary_name} is clearly the better choice for your {project_type} project with {budget} budget constraints."
    else:
        return f"Both {primary_name} and {secondary_name} are solid choices. {primary_name} edges out slightly, but consider your team's expertise and long-term goals."

if __name__ == '__main__':
    app.run(debug=True)