Update app.py

app.py

@@ -23,32 +23,40 @@ menu = ["txt", "htm", "md", "py"]
 choice = st.sidebar.selectbox("Output File Type:", menu)
 model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301'))
 
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:45]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
+def chat_with_model(prompt, document_section):
+    model = model_choice
+    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
+    conversation.append({'role': 'user', 'content': prompt})
+    conversation.append({'role': 'assistant', 'content': document_section})
+    response = openai.ChatCompletion.create(model=model, messages=conversation)
+    return response['choices'][0]['message']['content']
+
 def transcribe_audio(openai_key, file_path, model):
     OPENAI_API_URL = "https://api.openai.com/v1/audio/transcriptions"
     headers = {
         "Authorization": f"Bearer {openai_key}",
     }
-
     with open(file_path, 'rb') as f:
         data = {'file': f}
         response = requests.post(OPENAI_API_URL, headers=headers, files=data, data={'model': model})
-
     if response.status_code == 200:
         st.write(response.json())
+        response2 = chat_with_model(response.json().get('text'), '')
+        st.write('Responses:')
+        #st.write(response)
+        st.write(response2)
         return response.json().get('text')
     else:
         st.write(response.json())
         st.error("Error in API call.")
         return None
 
-
 def save_and_play_audio(audio_recorder):
     audio_bytes = audio_recorder()
     if audio_bytes:
@@ -64,14 +72,7 @@ if filename is not None:
     if st.button("Transcribe"):
         transcription = transcribe_audio(openai.api_key, filename, "whisper-1")
         st.write(transcription)
-
-def chat_with_model(prompt, document_section):
-    model = model_choice
-    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
-    conversation.append({'role': 'user', 'content': prompt})
-    conversation.append({'role': 'assistant', 'content': document_section})
-    response = openai.ChatCompletion.create(model=model, messages=conversation)
-    return response['choices'][0]['message']['content']
+        chat_with_model(transcription, '') # push transcript through as prompt
 
 def create_file(filename, prompt, response):
     if filename.endswith(".txt"):