support for ###

app.py

@@ -1,24 +1,67 @@
+import re
 import gradio as gr
 
+
 def insert_points(text):
+    # Split into lines first (handles the "after each line" insertion)
     lines = text.split('\n')
-    output = []
-    for i, line in enumerate(lines, start=1):
-        # Append the original line
-        output.append(line)
-        # Append the tag with leading zeroes (001, 002, etc.)
-        insert_tag = f"[INSERT_POINT_{i:03d}]"
-        # Place it on its own line
-        output.append(insert_tag)
-    # Join all lines with newline characters
-    return "\n".join(output)
+
+    counter = 1
+    output_lines = []
+
+    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)
+
+        # We'll rebuild the line piece by piece, inserting tags whenever we hit a # segment
+        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")
+                counter += 1
+
+                # 3) Now add the actual # sequence on its own line
+                output_lines.append(seg)
+            else:
+                # Normal text (no #)
+                rebuilt_line += 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}]")
+        counter += 1
+
+    # Join everything with newlines
+    return "\n".join(output_lines)
+
 
 demo = gr.Interface(
     fn=insert_points,
-    inputs=gr.Textbox(lines=10, placeholder="Paste your text here..."),
+    inputs=gr.Textbox(
+        lines=10,
+        placeholder="Paste your text here...",
+        label="Your Input Text"
+    ),
     outputs="text",
     title="Insert Point Tagger",
-    description="Paste a block of text and get '[INSERT_POINT_###]' added after each line."
+    description=(
+        "Paste a block of text and get '[INSERT_POINT_###]' tags added "
+        "after every newline and before every sequence of '#'."
+    ),
 )
 
 if __name__ == "__main__":