import openai

# Set your Groq API Key
GROQ_API_KEY = "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"

def predict_hydrogen_production(params):
    """
    Uses Groq API (Llama 3) to predict hydrogen production based on user inputs.
    """
    client = openai.OpenAI(api_key=GROQ_API_KEY)

    prompt = f"""
    Given the electrolysis parameters:
    - Current Density: {params['current_density']} A/cm²
    - Voltage: {params['voltage']} V
    - Cell Temperature: {params['temperature']} °C
    - Membrane: {params['membrane']}
    - Electrodes: {params['electrodes']}

    Predict hydrogen production rate, efficiency, and cost.
    """

    response = client.chat.completions.create(
        model="llama-3-70b",
        messages=[{"role": "user", "content": prompt}]
    )

    return response.choices[0].message.content