File size: 2,689 Bytes
6369972 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
import json
def create_markdown_document(plan_prompt: str, team_member_list_json_file_path: str, output_file_path: str):
"""
Reads text content and JSON data, then writes a Markdown document.
:param text_content: str, the main topic text to include in the Markdown
:param json_file_path: str, path to the JSON file
:param output_file_path: str, path to output the generated Markdown file
"""
# Load JSON data
with open(team_member_list_json_file_path, 'r', encoding='utf-8') as f:
roles_data = json.load(f)
# Begin constructing the Markdown content
rows = []
# 1. Add the main text content as a top-level section
rows.append("# The plan")
rows.append("")
rows.append(plan_prompt.strip())
rows.append("")
rows.append("---")
rows.append("")
# 2. Loop through each entry in the JSON and format as Markdown
for entry in roles_data:
# Each entry will be displayed under a subsection
rows.append(f"## Role {entry['id']} - {entry['category']}")
if 'explanation' in entry:
rows.append("")
rows.append(f"**Explanation**:\n{entry['explanation']}")
if 'count' in entry:
rows.append("")
rows.append(f"**People Count**:\n{entry['count']}")
if 'typical_job_activities' in entry:
rows.append("")
rows.append(f"**Typical Activities**:\n{entry['typical_job_activities']}")
if 'background_story' in entry:
rows.append("")
rows.append(f"**Background Story**:\n{entry['background_story']}")
if 'equipment_needs' in entry:
rows.append("")
rows.append(f"**Equipment Needs**:\n{entry['equipment_needs']}")
if 'facility_needs' in entry:
rows.append("")
rows.append(f"**Facility Needs**:\n{entry['facility_needs']}")
rows.append("")
rows.append("---")
rows.append("")
# Write everything to the output markdown file
with open(output_file_path, 'w', encoding='utf-8') as out_f:
out_f.write("\n".join(rows))
print(f"Markdown document has been created at: {output_file_path}")
if __name__ == "__main__":
# Your text snippet
plan_prompt = "Deep cave exploration to find new lifeforms in extreme conditions."
# Path to your JSON file
# TODO: Eliminate hardcoded paths
json_path = "/Users/neoneye/Desktop/planexe_data/005-enriched_team_members_list.json"
# Output Markdown file path
output_path = "output.md"
# Create the markdown document
create_markdown_document(plan_prompt, json_path, output_path) |