Update ailib.py

ailib.py

@@ -2,6 +2,8 @@ import openai
 import time
 
 OPENAI_TRANSCRIPTION_MODEL='whisper-1'
+OPENAI_COMPLETION_MODEL = "gpt-4o"
+MAX_TOKENS = 8192 
 openai_client = openai.OpenAI()
 
 def whisper_transcribe(fn, temperature=0):
@@ -16,3 +18,22 @@ def whisper_transcribe(fn, temperature=0):
     transcript = transcript.text
     print(f'Whisper transcribe [dt={dt} secs]')
     return transcript
+
+def summarize(text):
+    prompt = f"Resume todos los aspectos médicos de esta consulta: '{text}'"
+    
+    response = openai_client.chat.completions.create(
+            model = OPENAI_COMPLETION_MODEL,
+            messages = [{'role': 'user',
+                         'content': prompt}
+                       ],
+            temperature = 0, max_tokens = TOKEN_LIMIT,
+            top_p=1.0, frequency_penalty=0.0, 
+            presence_penalty=0.0)
+
+    resp_text = response.choices[0].message.content      # use others?
+
+    return resp_text
+
+
+