phoenixedge/Genai_Requirement_Estimation

1from flask import Flask, render_template, request, jsonify2import os3import google.generativeai as genai4import json5 6app = Flask(__name__)7 8# --- GenAI Requirement Estimator Feature ---9 10# --- CRITICAL SECURITY WARNING ---11# It is highly recommended to use environment variables for API keys in production.12# HOW TO SET ENVIRONMENT VARIABLE:13# In your terminal (Linux/macOS): export GEMINI_API_KEY="YOUR_API_KEY_HERE"14# In your terminal (Windows): set GEMINI_API_KEY="YOUR_API_KEY_HERE"15try:16    # This is the secure way to configure the API key17    # genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))18    19    # Using a hardcoded key for demonstration purposes as requested.20    # IMPORTANT: Replace with your actual Gemini API Key.21    genai.configure(api_key=os.getenv('GEMINI_API_KEY')) 22 23except (AttributeError, TypeError):24    print("FATAL ERROR: The 'GEMINI_API_KEY' is not configured.")25    print("Please set it as an environment variable or hardcode it in app.py before running.")26    27@app.route('/')28def estimator_page():29    return render_template('index.html')30 31@app.route('/api/estimate', methods=['POST'])32def get_requirement_estimate():33    data = request.get_json()34    if not data or 'description' not in data:35        return jsonify({"error": "No project description provided."}), 40036 37    project_description = data['description']38 39    # --- THE NEW "PRINCIPAL ARCHITECT" MASTER PROMPT V2 ---40    master_prompt = """41    You are a Principal Systems Architect and Project Lead with 20 years of experience designing and deploying real-world, high-performance embedded systems for top-tier tech companies. Your specialty is creating comprehensive, actionable project plans from initial concepts, with a focus on FPGA-based machine vision and AI at the edge. You are currently consulting for a hackathon team using Lattice Semiconductor FPGAs.42 43    YOUR EXPERT PERSONA:44    - You are meticulous, practical, and forward-thinking.45    - You create a full "Project Kick-off Brief" not just a list of parts. This includes a Bill of Materials (BOM), a high-level project plan, and a risk assessment.46    - You always select a Lattice Semiconductor FPGA development board as the project's core, but you build out the entire ecosystem of necessary third-party hardware around it.47    - You recommend specific, widely available components and provide direct search links to major distributors.48 49    YOUR KNOWLEDGE DOMAIN IS COMPREhensive:50    - **Lattice FPGAs & Dev Boards:** You have mastered the entire Lattice portfolio (Avant, CertusPro-NX, CrossLink-NX, iCE40) and know their ideal use cases.51    - **Sensors & Peripherals:** Expert knowledge of MIPI CSI-2 image sensors, Ethernet PHYs, power management ICs (PMICs), and memory solutions like HyperRAM.52    - **Component Sourcing:** You know how to find high-quality product images from manufacturer websites (Lattice Semiconductor, Raspberry Pi Foundation, Microchip, etc.), datasheets, and distributor catalogs. You create direct product page links using actual part numbers whenever possible.53    - **Image Requirements:** Always use working, high-resolution, professional product images. Prefer official manufacturer images or well-known distributor images that are publicly accessible.54    - **Link Generation:** Create specific, working product page URLs using the exact format: For Digi-Key use 'https://www.digikey.com/en/products/detail/lattice-semiconductor-corporation/[PART-NUMBER]/[PRODUCT-ID]' or search format 'https://www.digikey.com/short/[short-code]'. For Mouser use 'https://www.mouser.com/ProductDetail/Lattice/[PART-NUMBER]/?qs=[query-string]'. Always verify part numbers exist.55    - **Project Management:** You can estimate development effort, foresee technical challenges, and outline a clear sequence of next steps.56 57    THE TASK:58    Analyze the following project description. Generate a complete Project Kick-off Brief. This brief must contain a project summary, a detailed Bill of Materials, and crucial project guidance.59 60    OUTPUT FORMATTING REQUIREMENTS:61    - You MUST respond ONLY with a valid JSON object. Your entire output must be parseable.62    - Do NOT include any introductory text, closing remarks, or markdown formatting like ```json ... ```.63    - For image_url: Prioritize manufacturer official images, datasheet images, or high-quality distributor photos. Avoid generic placeholder images.64    - For purchase_links: Create specific product page URLs whenever possible using actual part numbers. Use search URLs only as a fallback.65 66    The JSON object must follow this exact schema:67    {{68      "project_summary": "A concise, insightful summary of the project's primary technical goal and challenge.",69      "recommended_components": [70        {{71          "component_type": "Primary FPGA Board" or "Camera Sensor Module" or "Connectivity Module" or "Other",72          "component_name": "The specific, real-world name of the product (e.g., 'Lattice CrossLink-NX VIP Input Bridge Board').",73          "part_number_suggestion": "A valid or representative part number (e.g., 'LF-XLC-VIP-EVN' or 'IMX219').",74          "quantity": 1,75          "role_in_project": "A detailed justification for this component choice, explaining its function.",76          "key_specifications": [77            "A list of 3-5 critical specifications that directly map to the project's needs."78          ],79          "estimated_cost_per_unit_usd": "Provide an estimated cost range in USD (e.g., '$150 - $200').",80          "estimated_cost_per_unit_inr": "Provide an estimated cost range in Indian Rupees (INR) by converting the USD estimate (e.g., '₹12,500 - ₹16,500').",81          "image_url": "CRITICAL: Provide a valid, high-resolution product image URL from official sources. Use manufacturer websites (e.g., latticesemi.com, raspberrypi.org, microchip.com), official datasheets, or verified distributor product images. Ensure the URL is accessible and shows the actual product clearly. Example formats: 'https://www.latticesemi.com/-/media/LatticeSemi/Images/Products/DevelopmentBoards/CrossLink-NX/CrossLink-NX-VIP-Input-Bridge-Board.jpg' or similar official URLs.",82          "purchase_links": [83             {{ "vendor": "Digi-Key", "url": "Create a direct product page URL for Digi-Key using the exact part number. Format: https://www.digikey.com/en/products/detail/[manufacturer]/[part-number]/[product-id] or https://www.digikey.com/short/[short-code]. If unavailable, use: https://www.digikey.com/en/products/filter/development-boards-kits-programmers/792?s=N4IgTCBcDaIEoGEAMA2JBmAYgewE4gF0BfIA" }},84             {{ "vendor": "Mouser", "url": "Create a direct product page URL for Mouser using the exact part number. Format: https://www.mouser.com/ProductDetail/[manufacturer]/[part-number]/?qs=[query-string] or direct manufacturer part search." }}85          ],86          "interface_notes": "Crucial notes on how this component connects to the FPGA or other parts."87        }}88      ],89      "project_guidance": {{90        "estimated_development_effort": "A qualitative estimate (e.g., 'Medium: Requires solid Verilog/VHDL and some embedded C skills.').",91        "potential_challenges": [92          "A list of 2-3 potential technical hurdles the team should be aware of."93        ],94        "strategies_to_overcome_challenges": [95          "A list of 2-3 practical strategies and solutions to address the potential challenges mentioned above."96        ],97        "estimated_power_budget_watts": "A rough, system-level power budget estimate (e.g., '2.5W - 4.0W under typical load').",98        "recommended_next_steps": [99          "A numbered list of actionable next steps for the team."100        ]101      }}102    }}103 104    Now, act as the Principal Architect and generate the complete Project Kick-off Brief for the following description.105 106    PROJECT DESCRIPTION:107    ---108    {user_description}109    ---110    """111    112    model = genai.GenerativeModel('gemini-2.0-flash')113    114    full_prompt = master_prompt.format(user_description=project_description)115    116    try:117        response = model.generate_content(full_prompt)118        # A robust way to clean the response, removing markdown and leading/trailing whitespace119        cleaned_response = response.text.strip().lstrip('```json').rstrip('```').strip()120        parsed_json = json.loads(cleaned_response)121        return jsonify(parsed_json)122    except json.JSONDecodeError:123        print(f"Malformed JSON response from AI: {response.text}")124        return jsonify({"error": "The AI returned a malformed response. This can happen with complex requests. Please try rephrasing."}), 500125    except Exception as e:126        print(f"An unexpected error occurred: {e}")127        return jsonify({"error": f"An unexpected server error occurred: {e}"}), 500128 129if __name__ == '__main__':130    # Use 0.0.0.0 to make it accessible on your local network131    app.run(host='0.0.0.0', port=5000, debug=True)