Reset Genie Streamlit conversation

Author: antontruly

streamlit/views/genie_api.py

@@ -28,8 +28,9 @@ def reset_conversation():
     genie_space_id = st.text_input(
         "Genie Space ID", placeholder="01efe16a65e21836acefb797ae6a8fe4", help="Room ID in the Genie Space URL"
     )
-    reset_conversation()
-    st.session_state.genie_space_id = genie_space_id
+    if genie_space_id != st.session_state.get("genie_space_id", ""):
+        reset_conversation()
+        st.session_state.genie_space_id = genie_space_id
 
 
     def display_message(message: Dict):
@@ -91,23 +92,24 @@ def process_genie_response(response: GenieMessage):
                         conversation = w.genie.create_message_and_wait(
                             genie_space_id, st.session_state.conversation_id, prompt
                         )
-                    except OperationFailed as e:
-                         st.error("Conversation failed. Check the required permissions.")
-                         raise e
+                    except Exception as e:
+                        status.update(label="Conversation failed. Check the required permissions.", state="error")
+                        st.button("New Chat", on_click=reset_conversation)
+                        raise e
                     if conversation.error:
                         st.error(conversation.error.type, conversation.error.error)
                     process_genie_response(conversation)
                 else:
                     try:
                         conversation = w.genie.start_conversation_and_wait(genie_space_id, prompt)
-                    except OperationFailed as e:
-                        st.error("Failed to initialize Genie. Check the required permissions.")
+                    except Exception as e:
+                        status.update(label="Failed to initialize Genie. Check the required permissions.", state="error")
+                        st.button("New Chat", on_click=reset_conversation)
                         raise e
                     if conversation.error:
                         st.error(conversation.error.type, conversation.error.error)
                     process_genie_response(conversation)
                 status.update(label="", state="complete")
-
                 st.button("New Chat", on_click=reset_conversation)
                 st.link_button("Open Genie", f"{w.config.host}/genie/rooms/{genie_space_id}/chats/{st.session_state.conversation_id}")
             else: