from groq import Groq
import json
import sys
import os

# Add parent directory to path for imports
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from config.settings import settings
from utils.prompts import SYSTEM_PROMPT, create_user_prompt

class GroqService:
    def __init__(self):
        self.client = Groq(api_key=settings.GROQ_API_KEY)
        self.model_id = settings.MODEL_ID
    
    async def analyze_provider_notes(self, provider_notes: str) -> dict:
        """
        Analyze provider notes and return ICD and CPT codes with explanations
        """
        try:
            # Create the chat completion with system and user prompts
            chat_completion = self.client.chat.completions.create(
                messages=[
                    {
                        "role": "system",
                        "content": SYSTEM_PROMPT
                    },
                    {
                        "role": "user",
                        "content": create_user_prompt(provider_notes)
                    }
                ],
                model=self.model_id,
                temperature=0.1,  # Low temperature for consistent, factual outputs
                response_format={"type": "json_object"}  # Force JSON response
            )
            
            # Extract and parse the response
            response_content = chat_completion.choices[0].message.content
            parsed_response = json.loads(response_content)
            
            return parsed_response
            
        except json.JSONDecodeError as e:
            raise ValueError(f"Failed to parse JSON response from model: {str(e)}")
        except Exception as e:
            raise Exception(f"Error calling Groq API: {str(e)}")

groq_service = GroqService()