Update app.py

app.py

@@ -2,86 +2,35 @@ import streamlit as st
 from audiorecorder import audiorecorder
 import openai
 import os
-openai.api_key = os.environ['OPENAI_API_KEY']
-
-
-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"]
 
+# Nastavení OpenAI klíče
+openai.api_key = os.environ['OPENAI_API_KEY']
 
 def transcribe(audio_path):
     audio_file = open(audio_path, "rb")
-    transcript = openai.Audio.translate_raw("whisper-1", audio_file, filename = '1.mp3')
+    transcript = openai.Audio.translate_raw("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
-
-
+# Inicializace session state
 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 loop for healthcare providers")
-st.markdown(
-    "Record your patient presentation and get the differential diagnoses and orders.")
+st.markdown("Record your patient presentation and get the transcription.")
 st.divider()
 
+# Nahrávání audia
 audio = audiorecorder("Record", "Stop")
 
-
-if (len(audio) != st.session_state['length']):
+# Zpracování nahrávky
+if len(audio) != st.session_state['length']:
     st.session_state['length'] = len(audio)
-    # 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)
+    transcript = transcribe(audio.tobytes())
+    st.session_state['vignette'] += transcript
 
-with col2:
-    st.markdown(
-        f"**ORDERS**\n\n{st.session_state['orders']}", unsafe_allow_html=True)
+# Zobrazení transkripce
+st.session_state['vignette'] = st.text_area("Vignette", value=st.session_state['vignette'])