Spaces:
Sleeping
Sleeping
File size: 1,211 Bytes
686a009 |
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 |
import os
import datetime
def save_document(response_text: str, directory: str = "./output"):
"""Export travel plan to Markdown file with proper formatting"""
os.makedirs(directory, exist_ok=True)
# Create markdown content with metadata header
markdown_content = f"""# 🌍 AI Travel Plan
# **Generated:** {datetime.datetime.now().strftime('%Y-%m-%d at %H:%M')}
# **Created by:** kube9t's Travel Agent
---
{response_text}
---
*This travel plan was generated by AI. Please verify all information, especially prices, operating hours, and travel requirements before your trip.*
"""
try:
# Write to markdown file with UTF-8 encoding
# Generate timestamp-based filename
timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
filename = f"{directory}/AI_Trip_Planner_{timestamp}.md"
print(filename)
with open(filename, 'w', encoding='utf-8') as f:
f.write(markdown_content)
print(f"Markdown file saved as: {filename}")
return filename
except Exception as e:
print(f"Error saving markdown file: {e}")
return None |