import os
import sys
import ollama
from openai import OpenAI

# Configuration
OLLAMA_MODEL = "llama3:8b" # Modèle performant et léger
# Si vous mettez une clé OpenAI, elle sera utilisée
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

PROMPT_SYSTEM = """
Tu es un expert médical assistant. Tu dois analyser la transcription d'une réunion médicale ou d'une consultation.
Ta mission est de produire une synthèse structurée incluant :
1. Objet de la réunion / Motif de consultation.
2. Éléments clés de la discussion (Symptômes, antécédents, examens évoqués).
3. Décisions prises ou Diagnostic provisoire.
4. Plan d'action (Traitements prescrits, examens complémentaires, prochain rendez-vous).

Règle d'or : Sois précis, utilise la terminologie médicale correcte, et sois synthétique.
"""

def summarize_transcript(transcript, use_openai=False):
    """
    Génère une synthèse médicale du transcript fourni.
    """
    if use_openai and OPENAI_API_KEY:
        print("Utilisation d'OpenAI pour la synthèse...")
        client = OpenAI(api_key=OPENAI_API_KEY)
        response = client.chat.completions.create(
            model="gpt-4o",
            messages=[
                {"role": "system", "content": PROMPT_SYSTEM},
                {"role": "user", "content": f"Voici le transcript à analyser :

{transcript}"}
            ]
        )
        return response.choices[0].message.content
    else:
        print(f"Utilisation d'Ollama ({OLLAMA_MODEL}) pour la synthèse...")
        try:
            response = ollama.chat(
                model=OLLAMA_MODEL,
                messages=[
                    {"role": "system", "content": PROMPT_SYSTEM},
                    {"role": "user", "content": f"Voici le transcript à analyser :

{transcript}"}
                ]
            )
            return response['message']['content']
        except Exception as e:
            return f"Erreur avec Ollama : {e}. Assurez-vous qu'Ollama est lancé et que le modèle {OLLAMA_MODEL} est téléchargé."

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Usage: python medical_summarizer.py <transcript_file.txt>")
    else:
        file_path = sys.argv[1]
        with open(file_path, "r", encoding="utf-8") as f:
            content = f.read()
        
        summary = summarize_transcript(content)
        
        output_file = file_path.rsplit('.', 1)[0] + "_summary.md"
        with open(output_file, "w", encoding="utf-8") as f:
            f.write(summary)
        
        print(f"Synthèse sauvegardée dans : {output_file}")
        print("
--- SYNTHÈSE MÉDICALE ---
")
        print(summary)