Update backupapp.py

backupapp.py

@@ -23,32 +23,53 @@ 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 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"
     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())
-        return response.json().get('text')
+        st.write('Reasoning with your transcription..')
+        transcript=response.json().get('text')
+        st.write(transcript)
+        gptResponse = chat_with_model(transcript, '') # send transcript to ChatGPT
+        filename = generate_filename(transcript, choice) # auto name file with date and prompt per output file type
+        create_file(filename, transcript, gptResponse) # write output file 
+        return gptResponse
     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:
@@ -59,30 +80,13 @@ def save_and_play_audio(audio_recorder):
         return filename
     return None
 
+# Updated to call direct from transcription to chat inference.
 filename = save_and_play_audio(audio_recorder)
 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']
-
-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}")
+    #if st.button("Transcribe"):
+    transcription = transcribe_audio(openai.api_key, filename, "whisper-1")
+    st.markdown('### Transcription:')
+    st.write(transcription)
 
 def truncate_document(document, length):
     return document[:length]
@@ -98,6 +102,8 @@ def get_table_download_link(file_path):
     ext = os.path.splitext(file_name)[1]  # get the file extension
     if ext == '.txt':
         mime_type = 'text/plain'
+    elif ext == '.wav':
+        mime_type = 'audio/x-wav'
     elif ext == '.htm':
         mime_type = 'text/html'
     elif ext == '.md':
@@ -191,11 +197,14 @@ def main():
     for file in all_files:
         col1, col3 = st.sidebar.columns([5,1])  # adjust the ratio as needed
         with col1:
-            st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+            try:
+                st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+            except Exception as e:
+                st.error(f"Error occurred while processing file {file}: {str(e)}")
         with col3:
             if st.button("🗑", key="delete_"+file):
                 os.remove(file)
                 st.experimental_rerun()
-            
+
 if __name__ == "__main__":
     main()