ISSが飛行中の地域を画像化・地図上にマッピングする

import os
import requests
from datetime import datetime
from skyfield.api import load
from geopy.geocoders import Nominatim
from geopy.exc import GeopyError
import openai
import folium # 地図描画用
from PIL import Image, ImageDraw, ImageFont # 画像編集用


# OpenAI APIキー設定
OPENAI_API_KEY = "APIキーをここに記述"
openai.api_key = OPENAI_API_KEY


# 画像保存用ディレクトリ
IMG_DIR = "img"
os.makedirs(IMG_DIR, exist_ok=True) # ディレクトリが存在しない場合は作成


# Linux 専用の全世界対応フォント（文字化け防止）
if os.path.exists("/usr/share/fonts/opentype/noto/NotoSans-Regular.ttf"):
FONT_PATH = "/usr/share/fonts/opentype/noto/NotoSans-Regular.ttf" # 全言語対応
elif os.path.exists("/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc"):
FONT_PATH = "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc" # 日本語・中国語・韓国語
else:
FONT_PATH = "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf" # 代替フォント


def get_location(latitude, longitude):
"""緯度・経度から住所を取得"""
geolocator = Nominatim(user_agent="myGeocoder")
try:
location = geolocator.reverse((latitude, longitude), exactly_one=True)
if location and location.address:
return location.address
else:
return "The ISS flying over the ocean"
except GeopyError as e:
print(f"An error occurred: {e}")
return "ジオコーディングエラー"
except AttributeError:
return "The ISS flying over the ocean"


def wrap_text(draw, text, font, max_width):
"""テキストを指定の幅内で自動改行"""
lines = []
words = text.split()
current_line = ""


for word in words:
test_line = current_line + " " + word if current_line else word
bbox = draw.textbbox((0, 0), test_line, font=font)
text_width = bbox[2] - bbox[0]


if text_width <= max_width:
current_line = test_line
else:
lines.append(current_line)
current_line = word


if current_line:
lines.append(current_line)


return lines


def add_text_overlay(image_path, output_path, text):
"""全世界対応フォントで地名をオーバーレイ表示し、改行対応"""
try:
img = Image.open(image_path)
draw = ImageDraw.Draw(img)


# フォント設定（適切なサイズ）
font_size = max(20, img.width // 30)
try:
font = ImageFont.truetype(FONT_PATH, font_size)
except IOError:
print("❌ フォントが見つかりません！デフォルトフォントを使用します")
font = ImageFont.load_default()


# 最大横幅を画像の90%に制限
max_text_width = int(img.width * 0.9)
lines = wrap_text(draw, text, font, max_text_width)


# ✅ 修正: `getsize()` → `textbbox()` で高さを取得
bbox = draw.textbbox((0, 0), "A", font=font)
line_height = bbox[3] - bbox[1]
text_height = line_height * len(lines)


# テキストの配置 (中央下部)
x = (img.width - max_text_width) // 2
y = img.height - text_height - 20 # 下部から少し上


# テキストの背景を半透明に（黒）
padding = 10
background_box = [x - padding, y - padding, x + max_text_width + padding, y + text_height + padding]
draw.rectangle(background_box, fill=(0, 0, 0, 180))


# 改行対応でテキスト描画
for i, line in enumerate(lines):
draw.text((x, y + i * line_height), line, font=font, fill="white")


img.save(output_path)
print(f"✅ 地名をオーバーレイした画像を保存しました: {output_path}")
except Exception as e:
print(f"❌ 画像へのオーバーレイ処理エラー: {e}")



def generate_image_with_dalle(prompt, file_name_png, file_name_jpg, address):
"""DALL·E API を使用して画像を生成し、ローカルに保存"""
try:
response = openai.images.generate(
model="dall-e-3",
prompt = f"Create an realistic image of {address}. The image should represent the culture and nature.",
n=1,
size="1024x1024"
)


# 画像URLを取得
image_url = response.data[0].url


# 画像をダウンロード
image_response = requests.get(image_url, stream=True)
if image_response.status_code == 200:
with open(file_name_png, "wb") as img_file:
for chunk in image_response.iter_content(1024):
img_file.write(chunk)
print(f"✅ PNG画像を保存しました: {file_name_png}")


# PNGをJPEGに変換
with Image.open(file_name_png) as img:
img = img.convert("RGB") # JPEGはRGB形式のみ対応
img.save(file_name_jpg, "JPEG")
print(f"✅ JPEG画像を保存しました: {file_name_jpg}")


# 地名をオーバーレイ
add_text_overlay(file_name_jpg, file_name_jpg, address)


return file_name_jpg
else:
print("❌ 画像のダウンロードに失敗しました")
return None
except openai.OpenAIError as e:
print(f"❌ 画像の生成エラー: {e}")
return None # エラー時は None を返す


def generate_map(latitude, longitude, address):
"""ISSの現在地を地図に表示し、HTMLファイルとしてカレントディレクトリに保存"""
file_name_map = "iss_map.html"
try:
print("🗺️ 地図を生成中...")
map_ = folium.Map(location=[latitude, longitude], zoom_start=5)
folium.Marker(
location=[latitude, longitude],
popup=address,
tooltip="ISSの現在地",
icon=folium.Icon(color="red", icon="cloud"),
).add_to(map_)


map_.save(file_name_map)
print(f"✅ 地図を保存しました: {file_name_map}")
except Exception as e:
print(f"❌ 地図の生成エラー: {e}")


# ISS の現在位置を取得
stations_url = 'http://celestrak.com/NORAD/elements/stations.txt'
try:
satellites = load.tle_file(stations_url)
by_name = {sat.name: sat for sat in satellites}
iss = by_name['ISS (ZARYA)']


# タイムスケールオブジェクトをロード
ts = load.timescale()
t = ts.now()


# ISS の現在位置を取得
geocentric = iss.at(t)


# 緯度経度を取得
subpoint = geocentric.subpoint()
latitude = subpoint.latitude.degrees
longitude = subpoint.longitude.degrees


print(f'Latitude: {latitude}, Longitude: {longitude}')


address = get_location(latitude, longitude)
print(f"Address: {address}")


# 地図を生成（カレントディレクトリに保存）
generate_map(latitude, longitude, address)


# 画像の保存用ファイル名を生成
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
file_name_png = os.path.join(IMG_DIR, f"iss_image_{timestamp}.png")
file_name_jpg = os.path.join(IMG_DIR, f"iss_image_{timestamp}.jpg")


# 画像生成
saved_file = generate_image_with_dalle(f"Create an image of {address}", file_name_png, file_name_jpg, address)


if saved_file:
print(f"✅ 画像が正常に保存されました: {saved_file}")


except Exception as e:
print(f"❌ ISSのデータ取得中にエラーが発生しました: {e}")