add function to load Ground Truth images and update prediction display

app.py

@@ -729,6 +729,29 @@ def create_intensity_map(pga_list, target_names, epicenter_lat=None, epicenter_l
     return m
 
 
+def load_ground_truth_image(event_name):
+    """從 ground_truth 資料夾載入對應的 Ground Truth 圖片"""
+    import os
+
+    # 根據事件名稱找對應的圖片
+    # 假設圖片命名格式為：20240403.png 或類似
+    event_file = EARTHQUAKE_EVENTS[event_name]
+    event_date = os.path.basename(event_file).replace('.mseed', '')
+
+    # 嘗試不同的圖片格式
+    ground_truth_dir = "ground_truth"
+    possible_extensions = ['.png', '.jpg', '.jpeg', '.gif']
+
+    for ext in possible_extensions:
+        image_path = os.path.join(ground_truth_dir, f"{event_date}{ext}")
+        if os.path.exists(image_path):
+            logger.info(f"載入 Ground Truth 圖片: {image_path}")
+            return image_path
+
+    logger.warning(f"找不到 Ground Truth 圖片: {event_date}")
+    return None
+
+
 def load_and_display_waveform(event_name, start_time, end_time, epicenter_lon, epicenter_lat):
     """載入並顯示波形，讓使用者確認範圍"""
     try:
@@ -855,7 +878,10 @@ def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter
         intensity_map = create_intensity_map(pga_list, target_names, epicenter_lat, epicenter_lon)
         map_html = intensity_map._repr_html_()
 
-        # 9. 統計資訊
+        # 9. 載入 Ground Truth 圖片
+        ground_truth_path = load_ground_truth_image(event_name)
+
+        # 10. 統計資訊
         max_intensity = max([calculate_intensity(pga, label=True) for pga in pga_list])
         stats = f"✅ 預測完成！\n"
         stats += f"選取時間範圍: {start_time:.1f} - {end_time:.1f} 秒\n"
@@ -864,13 +890,13 @@ def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter
         stats += f"預測最大震度: {max_intensity}"
 
         logger.info("預測完成！")
-        return map_html, stats
+        return ground_truth_path, map_html, stats
 
     except Exception as e:
         logger.error(f"預測過程發生錯誤: {e}")
         import traceback
         traceback.print_exc()
-        return None, f"錯誤: {str(e)}"
+        return None, None, f"錯誤: {str(e)}"
 
 
 # ============ Gradio 介面 ============
@@ -919,18 +945,24 @@ with gr.Blocks(title="TTSAM 震度預測系統") as demo:
             """)
 
             info_output = gr.Textbox(label="狀態資訊", lines=5, interactive=False)
+            stats_output = gr.Textbox(label="預測統計", lines=4)
 
-        # 右側：震度分布圖
+        # 右側：波形圖
         with gr.Column(scale=1):
-            gr.Markdown("## 預測震度分布")
-            intensity_map = gr.HTML(label="互動式震度地圖")
-            stats_output = gr.Textbox(label="預測統計", lines=4)
+            gr.Markdown("## 輸入波形")
+            waveform_plot = gr.Plot(label="地震波形（選定的最近測站）")
 
-    # 下方：波形圖
+    # 下方：Ground Truth vs 預測結果
     with gr.Row():
-        gr.Markdown("## 輸入波形")
-    with gr.Row():
-        waveform_plot = gr.Plot(label="地震波形（前 10 個最近測站）")
+        # 左側：Ground Truth
+        with gr.Column(scale=1):
+            gr.Markdown("## Ground Truth 震度分布")
+            ground_truth_image = gr.Image(label="實際觀測震度", type="filepath")
+
+        # 右側：預測震度地圖
+        with gr.Column(scale=1):
+            gr.Markdown("## 預測震度分布")
+            intensity_map = gr.HTML(label="互動式震度地圖")
 
     # 綁定事件
     # 第一步：載入波形
@@ -944,7 +976,7 @@ with gr.Blocks(title="TTSAM 震度預測系統") as demo:
     predict_btn.click(
         fn=predict_intensity,
         inputs=[event_dropdown, start_slider, end_slider, epicenter_lon_input, epicenter_lat_input, vs30_input],
-        outputs=[intensity_map, stats_output]
+        outputs=[ground_truth_image, intensity_map, stats_output]
     )
 
 demo.launch()