Update app.py

app.py

@@ -61,23 +61,13 @@ def transcribe_audio(openai_key, file_path, model):
         st.write('Reasoning with your transcription..')
         st.write(response.json())
         response2 = chat_with_model(response.json().get('text'), '') # send transcript to ChatGPT
+        create_file(filename, response, response2)
         return response2
     else:
         st.write(response.json())
         st.error("Error in API call.")
         return None
         
-def create_file(filename, prompt, response):
-    if filename.endswith(".txt"):
-        with open(filename, 'w') as file:
-            file.write(f"Prompt:\n{prompt}\nResponse:\n{response}")
-    elif filename.endswith(".htm"):
-        with open(filename, 'w') as file:
-            file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
-    elif filename.endswith(".md"):
-        with open(filename, 'w') as file:
-            file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
-
 def save_and_play_audio(audio_recorder):
     audio_bytes = audio_recorder()
     if audio_bytes:
@@ -95,9 +85,6 @@ if filename is not None:
     transcription = transcribe_audio(openai.api_key, filename, "whisper-1")
     st.markdown('### Transcription:')
     st.write(transcription)
-    #chat_with_model(transcription, '') # push transcript through as prompt
-    #response2 = chat_with_model(transcription.json().get('text'), '')
-
 
 def truncate_document(document, length):
     return document[:length]