Update app.py

app.py

@@ -778,24 +778,35 @@ def main():
             
         if next_action=='search':
             file_content_area = st.text_area("File Contents:", file_contents, height=500)
-            st.write('Reasoning with your inputs...')
+            st.write('🔍Running with Llama and GPT.')
 
             user_prompt = file_contents
             
             # Llama versus GPT Battle!
-            # llama
-            response = StreamLLMChatResponse(file_contents)
-            filename = generate_filename(user_prompt, ".md")
-            create_file(filename, file_contents, response, should_save)
-            SpeechSynthesis(response)
+            all=""
+            try:
+                st.write('🔍Running with Llama.')
+                response = StreamLLMChatResponse(file_contents)
+                filename = generate_filename(user_prompt, ".md")
+                create_file(filename, file_contents, response, should_save)
+                all=response
+                SpeechSynthesis(response)
+            except:
+                st.markdown('Llama is sleeping.  Restart ETA 30 seconds.')
             
             # gpt
-            response2 = chat_with_model(user_prompt, file_contents, model_choice)
-            filename2 = generate_filename(file_contents, choice)
-            create_file(filename2, user_prompt, response, should_save)
-            SpeechSynthesis(response)
+            try:
+                st.write('🔍Running with GPT.')
+                response2 = chat_with_model(user_prompt, file_contents, model_choice)
+                filename2 = generate_filename(file_contents, choice)
+                create_file(filename2, user_prompt, response, should_save)
+                all=all+response2
+                SpeechSynthesis(response2)
+            except:
+                st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
+
+            SpeechSynthesis(all)
             
-            #st.experimental_rerun()
 
     # Function to encode file to base64
     def get_base64_encoded_file(file_path):