import requests
import pandas as pd
from config import NREL_API_KEY, IEA_API_KEY, IRENA_API_KEY, DOE_API_KEY, GROQ_API_KEY
from groq import Groq

# Function to fetch hydrogen data from different APIs
def fetch_hydrogen_data(source):
    if source == "NREL":
        url = f"https://developer.nrel.gov/api/hydrogen/v1/stations.json?api_key={NREL_API_KEY}"
    elif source == "IEA":
        url = f"https://api.iea.org/hydrogen?api_key={IEA_API_KEY}"
    elif source == "IRENA":
        url = f"https://api.irena.org/hydrogen?api_key={IRENA_API_KEY}"
    elif source == "DOE":
        url = f"https://api.doe.gov/hydrogen?api_key={DOE_API_KEY}"
    else:
        return pd.DataFrame()

    response = requests.get(url)
    if response.status_code == 200:
        data = response.json()
        return pd.DataFrame(data["results"])
    else:
        return pd.DataFrame()

# Function for AI-powered analysis using Groq API
def groq_ai_analysis(data):
    groq_api = Groq(api_key=GROQ_API_KEY)
    
    query = f"Analyze hydrogen electrolysis data: {data.to_dict()}"
    
    response = groq_api.chat.completions.create(
        messages=[{"role": "user", "content": query}],
        model="llama3-70b-8192"
    )
    
    return response.choices[0].message["content"]