Update utils.py

utils.py

@@ -198,11 +198,11 @@ def generate_response(text_input):
     return stream, warnings
 
 
-def retry_generate_response(retry=3):
+def retry_generate_response(prompt, retry=3):
     response, warnings = "", []
     
     try:
-        stream, warnings = generate_response(st.session_state.prompt)
+        stream, warnings = generate_response(prompt)
         for warning_msg in warnings:
             st.warning(warning_msg)
         response = st.write_stream(stream)
@@ -210,7 +210,7 @@ def retry_generate_response(retry=3):
         if retry == 0:
             raise e
             
-        st.error(f"Internet connection is down. Trying to re-establish connection ({retry}).")
+        st.warning(f"Internet connection is down. Trying to re-establish connection ({retry}).")
 
         if st.session_state.server.is_down():
             st.session_state.server.restart()