support for single-line input

app.py

@@ -3,50 +3,45 @@ import gradio as gr
 
 
 def insert_points(text):
-    # Split into lines first (handles the "after each line" insertion)
+    # Split the input by newline to handle multi-line text
     lines = text.split('\n')
 
     counter = 1
-    output_lines = []
+    output = []
 
     for line in lines:
-        # Split the line by sequences of #, but keep the # parts as separate "segments"
-        # e.g. "Hello ## World # Another" -> ["Hello ", "##", " World ", "#", " Another"]
-        segments = re.split(r'(#+)', line)
+        # This pattern captures sequences of '#' (1 or more) plus optional trailing spaces
+        # We'll split the line on these sequences, but keep them so we can re-insert with tags.
+        segments = re.split(r'(\#+\s*)', line)
 
-        # We'll rebuild the line piece by piece, inserting tags whenever we hit a # segment
-        rebuilt_line = ""
+        # We'll rebuild this line piece by piece
+        rebuilt_line = []
 
         for seg in segments:
-            if re.match(r'^#+$', seg):
-                # seg is a sequence of '#' (e.g., "##", "###")
-
-                # 1) If there's any "normal text" waiting in rebuilt_line,
-                #    we push that as a separate line to the output first.
-                if rebuilt_line:
-                    output_lines.append(rebuilt_line)
-                    rebuilt_line = ""
-
-                # 2) Insert a tag line right BEFORE the # sequence
-                output_lines.append(f"\n[INSERT_POINT_{counter:03d}]\n")
+            if re.match(r'^\#+\s*$', seg):
+                # This segment is a run of '#' (with optional trailing space)
+                # Insert a tag BEFORE it
+                tag = f"[INSERT_POINT_{counter:03d}]"
                 counter += 1
+                rebuilt_line.append(tag)
 
-                # 3) Now add the actual # sequence on its own line
-                output_lines.append(seg)
+                # Then append the actual '#' segment
+                rebuilt_line.append(seg)
             else:
-                # Normal text (no #)
-                rebuilt_line += seg
+                # Normal text
+                rebuilt_line.append(seg)
 
-        # If there's leftover normal text after processing all # segments
-        if rebuilt_line:
-            output_lines.append(rebuilt_line)
-
-        # Finally, insert a tag after the entire line
-        output_lines.append(f"[INSERT_POINT_{counter:03d}]")
+        # After processing all segments in this line,
+        # append a final tag to match your "insert a tag after every line" requirement
+        line_tag = f"[INSERT_POINT_{counter:03d}]"
         counter += 1
+        rebuilt_line.append(line_tag)
+
+        # Join the processed segments for this line
+        output.append("".join(rebuilt_line))
 
-    # Join everything with newlines
-    return "\n".join(output_lines)
+    # Finally, join all lines with newlines
+    return "\n".join(output)
 
 
 demo = gr.Interface(
@@ -60,7 +55,7 @@ demo = gr.Interface(
     title="Insert Point Tagger",
     description=(
         "Paste a block of text and get '[INSERT_POINT_###]' tags added "
-        "after every newline and before every sequence of '#'."
+        "1) after every newline, and 2) before every '#' sequence."
     ),
 )