from __future__ import annotations

import json
import requests

from ...typing import CreateResult, Messages
from ..base_provider import ProviderModelMixin, AbstractProvider
from ..helper import format_prompt

class NexraChatGptWeb(AbstractProvider, ProviderModelMixin):
    label = "Nexra ChatGPT Web"
    url = "https://nexra.aryahcr.cc/documentation/chatgpt/en"
    working = True
    
    default_model = "gptweb"
    models = [default_model]
    model_aliases = {"gpt-4": "gptweb"}
    api_endpoints = {"gptweb": "https://nexra.aryahcr.cc/api/chat/gptweb"}

    @classmethod
    def get_model(cls, model: str) -> str:
        if model in cls.models:
            return model
        elif model in cls.model_aliases:
            return cls.model_aliases[model]
        else:
            return cls.default_model
            
    @classmethod
    def create_completion(
        cls,
        model: str,
        messages: Messages,
        markdown: bool = False,
        **kwargs
    ) -> CreateResult:
        model = cls.get_model(model)
        api_endpoint = cls.api_endpoints.get(model, cls.api_endpoints[cls.default_model])

        headers = {
            'Content-Type': 'application/json'
        }
        
        data = {
            "prompt": format_prompt(messages),
            "markdown": markdown
        }
        
        response = requests.post(api_endpoint, headers=headers, json=data)

        return cls.process_response(response)

    @classmethod
    def process_response(cls, response):
        if response.status_code == 200:
            try:
                content = response.text.lstrip('_')
                json_response = json.loads(content)
                return json_response.get('gpt', '')
            except json.JSONDecodeError:
                return "Error: Unable to decode JSON response"
        else:
            return f"Error: {response.status_code}"