Update app.py

app.py

@@ -36,7 +36,18 @@ def chat_with_model(prompt, document_section):
     conversation.append({'role': 'assistant', 'content': document_section})
     response = openai.ChatCompletion.create(model=model, messages=conversation)
     return response['choices'][0]['message']['content']
-
+    
+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}")
+    
 # Updated to auto process transcript to chatgpt in AI pipeline from Whisper to ChatGPT
 def transcribe_audio(openai_key, file_path, model):
     OPENAI_API_URL = "https://api.openai.com/v1/audio/transcriptions"
@@ -50,7 +61,18 @@ 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)
+        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}")
+
+(filename, response, response2)
         return response2
     else:
         st.write(response.json())
@@ -77,16 +99,6 @@ if filename is not None:
     #chat_with_model(transcription, '') # push transcript through as prompt
     #response2 = chat_with_model(transcription.json().get('text'), '')
 
-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 truncate_document(document, length):
     return document[:length]