Update app.py

app.py

@@ -40,11 +40,9 @@ def ask_assistant_stream(st_model, st_messages, st_temp_value, st_max_tokens):
             stream=True,
             max_tokens=st_max_tokens,
         )
-        st.write("success")
         response["stream"] = stream
     
     except Exception as e:
-        st.write("exception")
         pass
 
     return response
@@ -64,30 +62,24 @@ max_token_value = st.sidebar.slider('Select a max_token value', 1000, 9000, (500
 #Add reset button to clear conversation
 st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
 
-
 # Create model description
 st.sidebar.write(f"You're now chatting with **{selected_model}**")
 st.sidebar.markdown("*Generated content may be inaccurate or false.*")
-# st.sidebar.markdown("\n[TypeGPT](https://typegpt.net).")
-
-
-
 
 if "prev_option" not in st.session_state:
     st.session_state.prev_option = selected_model
 
+'''
 if st.session_state.prev_option != selected_model:
     st.session_state.messages = []
     # st.write(f"Changed to {selected_model}")
     st.session_state.prev_option = selected_model
     reset_conversation()
-
-
+'''
 
 #Pull in the model we want to use
 repo_id = model_links[selected_model]
 
-
 st.subheader(f'{selected_model}')
 # # st.title(f'ChatBot Using {selected_model}')
 
@@ -99,7 +91,6 @@ if selected_model not in st.session_state:
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
@@ -108,9 +99,8 @@ for message in st.session_state.messages:
 if "retry" not in st.session_state:
     st.session_state.retry= False
 
-def retry_click():
-    st.session_state.retry= True
-
+#def retry_click():
+#    st.session_state.retry= True
 
 if st.session_state.retry:
     lastmessage = st.session_state.messages.pop()
@@ -123,8 +113,8 @@ if st.session_state.retry:
 if "remove" not in st.session_state:
     st.session_state.remove= False
 
-def remove_click():
-    st.session_state.remove= True
+#def remove_click():
+#    st.session_state.remove= True
 
 if st.session_state.remove:
     lastmessage = st.session_state.messages.pop()
@@ -155,5 +145,9 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
 
 if len(st.session_state.messages)>0:
     col1, col2 = st.columns(2)
+    col1.button("retry", key="retry")
+    col2.button("remove", key="remove")
+'''
     col1.button("retry", key="retryButton", on_click=retry_click)
     col2.button("remove", key="removeButton", on_click=remove_click)
+'''