Initial commit

medical_summarizer.py

@@ -0,0 +1,73 @@
+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)