reintroduce error catching, but not with global semaphore yet

app.py

@@ -101,11 +101,11 @@ def text_stream(stream):
 def get_stream_warning_error(stream):
     error = None
     warning = None
-    # for chunk in stream:
-    #     if chunk["error"] is not None:
-    #         error = chunk["error"]
-    #     if chunk["warning"] is not None:
-    #         warning = chunk["warning"]
+    for chunk in stream:
+        if chunk["error"] is not None:
+            error = chunk["error"]
+        if chunk["warning"] is not None:
+            warning = chunk["warning"]
     return warning, error
 
 # @retry(wait=wait_random_exponential(min=0.5, max=2), stop=stop_after_attempt(3))
@@ -239,10 +239,10 @@ with main:
             with st.chat_message(message["role"],avatar=avatar):
                 if message["content"] is not None:
                     st.markdown(message["content"])
-                # if message["error"] is not None:
-                #     st.error(message["error"],icon="🚨")
-                # if message["warning"] is not None:
-                #     st.warning(message["warning"],icon="⚠️")
+                if message["error"] is not None:
+                    st.error(message["error"],icon="🚨")
+                if message["warning"] is not None:
+                    st.warning(message["warning"],icon="⚠️")
 
     if prompt := st.chat_input("Type a message!", max_chars=1000):
         handle_user_input(prompt)