# ai_assistant.py

from openai import OpenAI
import re

def initialize_openai_client(api_key, base_url):
    """Initialize the OpenAI client."""
    return OpenAI(api_key=api_key, base_url=base_url)

def generate_system_message(current_organism, current_antibiotic, current_was_positive):
    """Generate a dynamic system message for the AI Assistant."""
    system_message = """
    You are a clinical microbiology AI assistant. Analyze susceptibility results considering:
    1) Model predictions 2) CLSI/EUCAST guidelines 3) Local resistance patterns.
    Provide: Interpretation, therapy options, resistance mechanisms, and infection control measures.
    """
    
    if all([current_organism, current_antibiotic, current_was_positive is not None]):
        system_message += f"""
        CONTEXTUAL NOTES:
        - Current organism: {current_organism}
        - Current antibiotic: {current_antibiotic}
        - Culture positivity: {'Positive' if current_was_positive else 'Negative'}
        """
    
    return system_message

# In ai_assistant.py

def get_ai_response(client, prompt):
    """Get a response from the AI Assistant."""
    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": "You are a clinical microbiology AI assistant."},
            {"role": "user", "content": prompt}
        ]
    ).choices[0].message.content
    return response