RAGWeatherStaytimeTripPlanner / sample_data_seed.py
Corin1998's picture
Upload sample_data_seed.py
2bc3698 verified
import os, json ,csv
BLOG_MD = """# Tokyo local food pickes
## Tsukiji Outer Market
Fresh sushi, street food, great for Food, outdoor.
## TeamLab Planets
Immersive art museum, perfect for Culture, indoor.
"""
REVIEWS = [
{
"title": "Tukiji Outer Market",
"text": "Vibrant market with many stalls. Best in the morning. Food,outdoor",
"rating": 4.6,
"lat": 35.6655,
"lon": 139.7708,
"url": "https://example.com/tsukiji",
"tag":["Food", "outdoor"]
},
{
"title": "teamLab Planets",
"text": "Immersive digital art museum. Reserve tickets. Culture, indoor",
"rating": 4.7,
"lat": 35.6457,
"lon": 139.7823,
"url": "https://example.com/teamlab",
"tag":["Culture", "indoor"]
},
]
EVENTS =[
["Sumida River Fireworks", "River fireworks festival(summer)", "2025-07-26", "2025-07-26", 35.7100, 139.8107, "https://example.com//fireworks", "outdoor,festival","Tokyo"]
]
def seed():
os.makedirs("data/blogs",exist_ok=True)
os.makedirs("data/reviews", exist_ok=True)
os.makedirs("data/events", exist_ok=True)
with open("data/blogs/local.md", "w",encoding="utf-8") as f:
f.write(BLOG_MD)
with open("data/reviews/reviews.jsonl", "w", encoding="utf-8") as f:
for r in REVIEWS:
f.write(json.dumps(r, ensure_ascii=False) + "\n")
with open("data/events/events.csv", "w", encoding="utf-8", newline="") as f:
w = csv.writer(f)
w.writerow(["title", "description", "start", "end", "lat", "lon", "url", "tags", "city"])
for row in EVENTS:
w.writerow(row)
return 1 + len(REVIEWS) +len(EVENTS)