Create app.py

app.py

@@ -0,0 +1,63 @@
+import gradio as gr
+from docx import Document
+from docx.shared import Pt
+import tempfile
+import os
+
+
+def build_docx(num_episodes, num_segments, episode_label, segment_label):
+    num_episodes = int(num_episodes)
+    num_segments = int(num_segments)
+    episode_label = episode_label.strip() or "Episode"
+    segment_label = segment_label.strip() or "Segment"
+
+    doc = Document()
+
+    for e in range(1, num_episodes + 1):
+        doc.add_heading(f"{episode_label} {e}", level=1)
+
+        for s in range(1, num_segments + 1):
+            doc.add_heading(f"{segment_label} {s}", level=2)
+            # 2 blank lines inside each segment
+            doc.add_paragraph("")
+            doc.add_paragraph("")
+
+    tmp_dir = tempfile.mkdtemp()
+    out_path = os.path.join(tmp_dir, "document_structure.docx")
+    doc.save(out_path)
+    return out_path
+
+
+def generate(num_episodes, num_segments, episode_label, segment_label):
+    if num_episodes < 1 or num_segments < 1:
+        raise gr.Error("Episodes and Segments must be at least 1.")
+    path = build_docx(num_episodes, num_segments, episode_label, segment_label)
+    return path
+
+
+with gr.Blocks(title="Docx Structure Generator") as demo:
+    gr.Markdown(
+        "## Docx Structure Generator\n"
+        "Creates a Word document with **Episodes (H1)** each containing **Segments (H2)**, "
+        "with 2 blank lines under every segment heading."
+    )
+
+    with gr.Row():
+        num_episodes = gr.Number(label="Number of Episodes", value=3, precision=0, minimum=1)
+        num_segments = gr.Number(label="Segments per Episode", value=3, precision=0, minimum=1)
+
+    with gr.Row():
+        episode_label = gr.Textbox(label="Episode Heading Label", value="Episode")
+        segment_label = gr.Textbox(label="Segment Heading Label", value="Segment")
+
+    generate_btn = gr.Button("Generate .docx", variant="primary")
+    output_file = gr.File(label="Download generated document")
+
+    generate_btn.click(
+        fn=generate,
+        inputs=[num_episodes, num_segments, episode_label, segment_label],
+        outputs=output_file,
+    )
+
+if __name__ == "__main__":
+    demo.launch()