Update app.py

app.py

@@ -19,7 +19,6 @@ def reset_conversation():
     Resets Conversation
     '''
     st.session_state.messages = []
-    return None
 
 # Set the temperature value directly in the code
 temperature = 0.5
@@ -31,42 +30,44 @@ st.button('Reset Chat', on_click=reset_conversation)
 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"]):
+        st.markdown(message["content"])
+
 # Accept user input
 if prompt := st.chat_input("Type your message here..."):
 
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
+
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    # Display assistant response in chat message container
-    with st.chat_message("assistant"):
-        try:
-            # Prepare the messages for the API call
-            messages_for_api = [
-                {"role": "system", "content": "As a helpful and friendly assistant, provide concise and accurate responses to the user's queries."}
-            ] + st.session_state.messages  # Add system message and user messages to the list
-
-            response = client.chat.completions.create(
-                model=model_link,
-                messages=messages_for_api,
-                temperature=temperature,
-                max_tokens=150  # Adjust the max tokens according to your needs
-            )
-            
-            # Get the response content
-            assistant_response = response["choices"][0]["message"]["content"]
-            st.markdown(assistant_response)
+    # Interact with the model
+    try:
+        # Send the user and system messages to the API
+        messages_for_api = [{"role": "system", "content": "You are a helpful assistant."}] + st.session_state.messages
 
-        except Exception as e:
-            st.write("An error occurred. Please try again later.")
-            st.write(e)
+        response = client.chat.completions.create(
+            model=model_link,
+            messages=messages_for_api,
+            temperature=temperature,
+            max_tokens=150  # Adjust the max tokens according to your needs
+        )
 
-    # Append the assistant's response to the chat history
-    st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+        assistant_response = response["choices"][0]["message"]["content"]
 
-# Display chat messages from history on app rerun
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+        # Display assistant response in chat message container
+        with st.chat_message("assistant"):
+            st.markdown(assistant_response)
+
+        # Append the assistant's response to the chat history
+        st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+
+    except Exception as e:
+        # Display error message to user
+        with st.chat_message("assistant"):
+            st.markdown("Sorry, I couldn't process your request. Please try again later.")
+        st.write(e)