Upload app.py with huggingface_hub

app.py

@@ -193,7 +193,6 @@ def generate_spec(
 
         generated = outputs[0][input_len:]
         spec = tokenizer.decode(generated, skip_special_tokens=True)
-        spec = add_function_separators(spec)
         return render_mermaid_images(spec)
 
     except Exception as e:
@@ -206,29 +205,6 @@ def generate_spec(
 # --- Gradio UI ---
 
 
-def add_function_separators(spec: str) -> str:
-    """Insert horizontal rules between function/method spec sections.
-
-    The model doesn't generate separators between ### headings, so we
-    inject --- before each ### that follows content (not the first one
-    after a ## heading).
-    """
-    lines = spec.split("\n")
-    result = []
-    prev_was_section_header = True
-    for line in lines:
-        if line.startswith("### "):
-            if not prev_was_section_header:
-                result.append("\n---\n")
-            prev_was_section_header = False
-        elif line.startswith("## "):
-            prev_was_section_header = True
-        elif line.strip():
-            prev_was_section_header = False
-        result.append(line)
-    return "\n".join(result)
-
-
 def render_mermaid_images(spec: str) -> str:
     """Replace mermaid code blocks with rendered SVG images via mermaid.ink.