Update app.py

app.py

@@ -563,7 +563,7 @@ def read_file_content(file,max_length):
 
 # 11. Chat with GPT - Caution on quota - now favoring fastest AI pipeline STT Whisper->LLM Llama->TTS
 @st.cache_resource
-def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):
+def chat_with_model(prompt, document_section='', model_choice='gpt-3.5-turbo'):
     model = model_choice
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
     conversation.append({'role': 'user', 'content': prompt})
@@ -760,34 +760,34 @@ def whisper_main():
     filename = save_and_play_audio(audio_recorder)
     if filename is not None:
         transcription = transcribe_audio(filename)
-        try:
-            transcript = transcription['text']
-            st.write(transcript)
-
-            
-            # Whisper to GPT: New!! ---------------------------------------------------------------------
-            st.write('Reasoning with your inputs with GPT..')
-            response = chat_with_model(prompt_section, ''.join(list(document_sections)), model_choice)
-            st.write('Response:')
-            st.write(response)
-            filename = generate_filename(user_prompt, choice)
-            create_file(filename, user_prompt, response, should_save)
-            # Whisper to GPT: New!! ---------------------------------------------------------------------
-           
+        #try:
+        transcript = transcription['text']
+        st.write(transcript)
 
-            
-            # Whisper to Llama:
-            response = StreamLLMChatResponse(transcript)
-            filename_txt = generate_filename(transcript, ".txt")
-            create_file(filename_txt, transcript, response, should_save)
-            filename_wav = filename_txt.replace('.txt', '.wav')
-            import shutil
-            shutil.copyfile(filename, filename_wav)
-            if os.path.exists(filename):
-                os.remove(filename)
+        
+        # Whisper to GPT: New!! ---------------------------------------------------------------------
+        st.write('Reasoning with your inputs with GPT..')
+        response = chat_with_model(transcript)
+        st.write('Response:')
+        st.write(response)
+        filename = generate_filename(transcript, "txt")
+        create_file(filename, transcript, response, should_save)
+        # Whisper to GPT: New!! ---------------------------------------------------------------------
+        
 
-        except:
-            st.write('Starting Whisper Model on GPU.  Please retry in 30 seconds.')
+        
+        # Whisper to Llama:
+        response = StreamLLMChatResponse(transcript)
+        filename_txt = generate_filename(transcript, ".txt")
+        create_file(filename_txt, transcript, response, should_save)
+        filename_wav = filename_txt.replace('.txt', '.wav')
+        import shutil
+        shutil.copyfile(filename, filename_wav)
+        if os.path.exists(filename):
+            os.remove(filename)
+
+        #except:
+        #    st.write('Starting Whisper Model on GPU.  Please retry in 30 seconds.')
 
 
 import streamlit as st