Update app8.py

app8.py

@@ -388,22 +388,22 @@ if "task_choice" in st.session_state:
                 else:
                     st.warning("Please enter at least one example to classify.")
 
-##if st.session_state.messages:
-    ##st.markdown("### Output:")
-    ##for message in st.session_state.messages[-1:]:
-        ##st.markdown(message["content"])
-
 if st.session_state.messages:
     st.markdown("### Output:")
-    last_message = st.session_state.messages[-1]["content"]
+    for message in st.session_state.messages[-1:]:
+        st.markdown(message["content"])
+
+##if st.session_state.messages:
+    ##st.markdown("### Output:")
+    ##last_message = st.session_state.messages[-1]["content"]
 
     # Find the position of "Output:" if it exists
-    output_start = last_message.find("Output:")
+    ##output_start = last_message.find("Output:")
 
-    if output_start != -1:
+    ##if output_start != -1:
         # Display only the content after "Output:"
-        cleaned_output = last_message[output_start + 7:].strip()
-        st.markdown(cleaned_output)
-    else:
+        ##cleaned_output = last_message[output_start + 7:].strip()
+        ##st.markdown(cleaned_output)
+    ##else:
         # If "Output:" is not found, display the content as is
-        st.markdown(last_message)
+        ##st.markdown(last_message)