Update app8.py

app8.py

@@ -388,7 +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:")
-    for message in st.session_state.messages[-1:]:
-        st.markdown(message["content"])
+    last_message = st.session_state.messages[-1]["content"]
+
+    # Find the position of "Output:" if it exists
+    output_start = last_message.find("Output:")
+
+    if output_start != -1:
+        # Display only the content after "Output:"
+        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)