import gradio as gr
import requests
import json
import logging
import os

# Set up logging
logging.basicConfig(level=logging.INFO)

def get_model_info(model_id="Qwen/Qwen2-7B-Instruct", hf_token=os.getenv('HF_TOKEN')):
    url = f"https://huggingface.co/api/models"
    params = {
        "limit": 5,
        "pipeline_tag": "text-generation",
        "sort": "likes30d",
        "full": "True",
        "config": "True"
    }
    
    headers = {}
    if hf_token:
        headers["Authorization"] = f"Bearer {hf_token}"
    try:
        response = requests.get(url, params=params, headers=headers)
    except requests.exceptions.RequestException as e:
        print(f"Error fetching data for task text-generation: {e}")
        return None
    data = response.json()
    return json.dumps(data, indent=4)
    
    # url = f"https://huggingface.co/api/integrations/aws/v1/lookup/{model_id}"
    # headers = {
    #     "Authorization": f"Bearer {hf_token}",
    # }

    # logging.info(f"Requesting model info for model ID: {model_id}")
    # logging.info(f"URL: {url}")

    # response = requests.get(url, headers=headers)
    
    # logging.info(f"Response Status: {response.status_code}")
    # logging.info(f"Response Headers: {response.headers}")
    # logging.info(f"Response Text: {response.text}")

    # if response.status_code != 200:
    #     logging.error(f"Error: {response.status_code} - {response.text}")
    #     return f"Error: {response.status_code}\\nResponse: {response.text}"
    
    # data = response.json()
    # logging.info("Successfully retrieved model info.")
    # return json.dumps(data, indent=4)

iface = gr.Interface(
    fn=lambda model_id: get_model_info(model_id, hf_token=os.getenv('HF_TOKEN')),
    inputs=gr.Textbox(label="Model ID", placeholder="HuggingFaceH4/zephyr-7b-beta"),
    outputs=gr.Textbox(label="API Response", lines=20),
    title="Hugging Face Model Lookup",
    description="Enter a model ID to retrieve its AWS integration details from Hugging Face."
)

if __name__ == "__main__":
    iface.launch()