Update app.py

app.py

@@ -2,86 +2,26 @@ 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"]
-
-
-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
-
-
-st.title("AI loop for healthcare providers")
-st.markdown(
-    "Nahraj pacientův projev.")
-st.divider()
-
-audio = audiorecorder("Record", "Stop")
-
-
-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.transcribe("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)
+# Nastavení klíče API pro OpenAI
+openai.api_key = os.environ['OPENAI_API_KEY']
 
-with col2:
-    st.markdown(
-        f"**ORDERS**\n\n{st.session_state['orders']}", unsafe_allow_html=True)
+def transcribe(audio_data):
+    """
+    Transkribuje audio data pomocí Whisper API.
+    """
+    response = openai.Audio.transcribe("whisper-1", audio_data, filename='1.mp3')
+    return response["text"]
+
+# Streamlit aplikace
+st.title("Audio Transkriptor")
+
+# Nahrávání audia
+st.markdown("Nahrajte svůj hlasový záznam.")
+audio_data = audiorecorder("Record", "Stop")
+
+# Zpracování a zobrazení transkripce
+if audio_data is not None and len(audio_data) > 0:
+    with st.spinner('Probíhá transkripce...'):
+        transcript = transcribe(audio_data.tobytes())
+        st.text_area("Transkript", value=transcript, height=300)