74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
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)
|