Update app.py

app.py

@@ -2,45 +2,86 @@ import streamlit as st
 from audiorecorder import audiorecorder
 import openai
 import os
-import tempfile
-
-# Nastavení OpenAI klíče
 openai.api_key = os.environ['OPENAI_API_KEY']
 
-def transcribe(audio_bytes):
-    # Vytvoření dočasného souboru pro uložení audio dat
-    with tempfile.NamedTemporaryFile(suffix='.mp3', delete=False) as temp_audio_file:
-        temp_audio_file.write(audio_bytes)
-        temp_audio_file_path = temp_audio_file.name
-
-    # Otevření dočasného souboru a jeho předání modelu Whisper
-    with open(temp_audio_file_path, "rb") as audio_file:
-        transcript = openai.Audio.transcribe(
-            file=audio_file,
-            model="whisper-1"
-        )
-        return transcript["choices"][0]["text"]
-
-# Inicializace session state
+
+def get_completion(messages, model="gpt-3.5-turbo"):
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=messages,
+        temperature=0)
+    return response.choices[0].message["content"]
+
+
+def transcribe(audio_path):
+    audio_file = open(audio_path, "rb")
+    transcript = openai.Audio.transcribe("whisper-1", audio_file, filename = '1.mp3')
+    return transcript["text"]
+
+
+def get_ddx(vignette):
+    messages_ddx = [
+        {'role': 'system', 'content': 'Jste nástroj AI asistenta lékaře. Napište sadu příkazů pro pacienta k odlišení mezi stavy. Napište pouze příkazy a zdůvodnění. Nepište žádné další informace. Nepište žádný úvod.'},
+        {'role': 'user', 'content': vignette}]
+    ddx = get_completion(messages_ddx)
+    return ddx
+
+
+def get_orders(vignette, ddx):
+    messages_orders = [
+        {'role': 'system', 'content': 'Jste nástroj AI asistenta lékaře. Napište sadu příkazů pro pacienta k odlišení mezi stavy. Napište pouze příkazy a zdůvodnění. Nepište žádné další informace. Nepište žádný úvod.'},
+        {'role': 'user', 'content': f'Informace o pacientovi: {vignette}. Rozdílné diagnózy: {ddx}'}]
+    orders = get_completion(messages_orders)
+    return orders
+
+
 if 'vignette' not in st.session_state:
     st.session_state['vignette'] = ''
 
+if 'ddx' not in st.session_state:
+    st.session_state['ddx'] = ''
+
+if 'orders' not in st.session_state:
+    st.session_state['orders'] = ''
+
 if 'length' not in st.session_state:
     st.session_state['length'] = 0
 
-# Hlavní titulek aplikace
-st.title("AI Transcription for Healthcare Providers")
-st.markdown("Record your patient presentation and get the transcription.")
+
+st.title("AI loop for healthcare providers")
+st.markdown(
+    "Record your patient presentation and get the differential diagnoses and orders.")
 st.divider()
 
-# Nahrávání audia
 audio = audiorecorder("Record", "Stop")
 
-# Zpracování nahrávky
-if len(audio) != st.session_state['length']:
+
+if (len(audio) != st.session_state['length']):
     st.session_state['length'] = len(audio)
-    transcript = transcribe(audio.tobytes())
-    st.session_state['vignette'] += transcript
+    # wav_file = open("audio.mp3", "wb")
+    # wav_file.write(audio.tobytes())
+    transcript = openai.Audio.translate_raw("whisper-1", audio.tobytes(), filename = '1.mp3')
+    transcript["text"]
+    st.session_state['vignette'] += transcript["text"]
+
+
+st.session_state['vignette'] = st.text_area(
+    "Vignette", value=st.session_state['vignette'])
+
+
+if st.button("Get DDX and Orders"):
+    vignette = st.session_state['vignette']
+    ddx = get_ddx(vignette)
+    st.session_state['ddx'] = ddx
+    st.session_state['orders'] = get_orders(vignette, ddx)
+
+
+col1, col2 = st.columns(2)
+
+with col1:
+    st.markdown(
+        f"**DDX**\n\n{st.session_state['ddx']}", unsafe_allow_html=True)
 
-# Zobrazení transkripce
-st.session_state['vignette'] = st.text_area("Transcription", value=st.session_state['vignette'])
+with col2:
+    st.markdown(
+        f"**ORDERS**\n\n{st.session_state['orders']}", unsafe_allow_html=True)