Update ailib.py

ailib.py

@@ -7,6 +7,28 @@ TOKEN_LIMIT = 4096
 
 openai_client = openai.OpenAI()
 
+def openai_query(prompt):
+    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      
+    return resp_text
+
+def nueva_ficha(resumen_historial, resumen_hoy):
+    
+    new_prompt = f'''Escribe una nueva ficha médica, considerando el historial médico: "{resumen_historial}",
+                    y también lo que ocurrió hoy: "{resumen_hoy}"
+                    '''
+    nueva = openai_query(new_prompt)
+    
+    return nueva
+    
+
 def whisper_transcribe(fn, temperature=0):
     t0 = time.time()
     audio_file = open(fn,'rb')  # redundant?
@@ -22,19 +44,5 @@ def whisper_transcribe(fn, temperature=0):
 
 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
-
-
-
+    resp_text = openai_query(prompt)
+    return resp_text