在前面輸出的時候,用的檔名是同樣的,在覆蓋檔案的同時,有時候會讓人找不到修改的部分,所以我希望今天更新的程式,後面會根據輸入的城市與天數,自動生成檔名,
if name == "main":
print("=== AI 旅遊行程規劃器 ===")
city = input("請輸入旅遊城市:")
days = input("請輸入旅遊天數:")
style = input("請輸入旅遊風格(美食/文化/購物/自然/混合):")
user_request = f"設計一個{days}日的{city}旅遊行程,偏好{style}"
full_prompt = f"""
根據以下需求設計詳細行程:{user_request}
請包含以下資訊:
時間安排(上午、下午、晚上)
推薦景點與餐廳
建議交通方式(步行、大眾運輸、計程車)
提供 Google Maps 連結(如:https://maps.google.com/?q=地點名稱)
"""
try:
response = model.generate_content(full_prompt)
itinerary = response.text.strip()
except Exception as e:
print("❌ AI 產生行程失敗:", e)
sys.exit(1)
if not itinerary:
print("⚠️ 沒有產生到行程,請再試一次。")
sys.exit(1)
print("\n=== AI 行程建議 ===\n")
print(itinerary)
主要更新在這,檔名依據城市與天數命名
base_filename = f"{city}_{days}日遊"
with open(f"{base_filename}.txt", "w", encoding="utf-8") as f:
f.write(itinerary)
print(f"\n✅ 行程已存成 {base_filename}.txt")
save_as_pdf(f"{base_filename}.pdf", itinerary)
print(f"✅ 行程已存成 {base_filename}.pdf")
更新的如上
明天來確認結果吧