Update displayed string and add comment for PySparkCurrentOrigin

sql/api/src/main/scala/org/apache/spark/sql/catalyst/trees/QueryContexts.scala

@@ -160,32 +160,28 @@ case class DataFrameQueryContext(stackTrace: Seq[StackTraceElement]) extends Que
   val pysparkOriginInfo: mutable.Map[String, String] = PySparkCurrentOrigin.get()
 
   val pysparkFragment: String = pysparkOriginInfo.getOrElse("fragment", "")
+
   val pysparkCallSite: String = pysparkOriginInfo.getOrElse("callSite", "")
 
+  val (displayedFragment, displayedCallsite) = if (pysparkOriginInfo.nonEmpty) {
+    (pysparkFragment, pysparkCallSite)
+  } else {
+    (fragment, callSite)
+  }
+
+  PySparkCurrentOrigin.clear()
+
   override lazy val summary: String = {
     val builder = new StringBuilder
     builder ++= "== DataFrame ==\n"
     builder ++= "\""
 
-    builder ++= fragment
+    builder ++= displayedFragment
     builder ++= "\""
     builder ++= " was called from\n"
-    builder ++= callSite
+    builder ++= displayedCallsite
     builder += '\n'
 
-    if (pysparkOriginInfo.nonEmpty) {
-      builder ++= "\n== PySpark call site ==\n"
-      builder ++= "\""
-
-      builder ++= pysparkFragment
-      builder ++= "\""
-      builder ++= " was called from\n"
-      builder ++= pysparkCallSite
-      builder += '\n'
-    }
-
-    PySparkCurrentOrigin.clear()
-
     builder.result()
   }
 }

sql/api/src/main/scala/org/apache/spark/sql/catalyst/trees/origin.scala

@@ -90,6 +90,11 @@ object CurrentOrigin {
 /**
  * Provides detailed call site information on PySpark.
  * This information is generated in PySpark and stored in the form of a Map.
+ * The 'fragment' represents the method name or the action within the PySpark code
+ * that initiated the call, serving as a brief descriptor of the operation being performed.
+ * The 'callSite' provides a more detailed context, including the file name and line number in the
+ * PySpark user code where the operation originated, offering precise location information for
+ * debugging purposes.
  */
 object PySparkCurrentOrigin {
   private val pysparkCallSite = new ThreadLocal[mutable.Map[String, String]]() {