Upload app.py with huggingface_hub

app.py

@@ -193,6 +193,7 @@ 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:
@@ -205,6 +206,29 @@ 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.