add waveform loading functionality and update Gradio interface for user interaction

app.py

@@ -601,6 +601,40 @@ def plot_intensity_map(pga_list, target_names):
     return fig
 
 
+def load_and_display_waveform(event_name, start_time, end_time, epicenter_lon, epicenter_lat):
+    """載入並顯示波形，讓使用者確認範圍"""
+    try:
+        # 1. 載入完整的 mseed 檔案
+        logger.info(f"載入地震事件: {event_name}")
+        st = load_waveform(event_name)
+        logger.info(f"載入了 {len(st)} 個 trace")
+
+        # 2. 根據震央距離選擇最近的 25 個測站
+        logger.info(f"選擇距離震央 ({epicenter_lat}, {epicenter_lon}) 最近的測站...")
+        selected_stations = select_nearest_stations(st, epicenter_lat, epicenter_lon, n_stations=25)
+
+        if len(selected_stations) == 0:
+            return None, "錯誤：找不到有效的測站資料", gr.update(interactive=False)
+
+        # 3. 繪製波形
+        waveform_plot = plot_waveform(st, selected_stations, start_time, end_time)
+
+        info_text = f"✅ 已載入波形資料\n"
+        info_text += f"選取時間範圍: {start_time:.1f} - {end_time:.1f} 秒\n"
+        info_text += f"震央位置: ({epicenter_lon:.4f}, {epicenter_lat:.4f})\n"
+        info_text += f"選擇了 {len(selected_stations)} 個最近的測站\n"
+        info_text += f"請確認波形範圍後，點擊「執行預測」按鈕"
+
+        logger.info("波形載入完成")
+        return waveform_plot, info_text, gr.update(interactive=True)
+
+    except Exception as e:
+        logger.error(f"波形載入發生錯誤: {e}")
+        import traceback
+        traceback.print_exc()
+        return None, f"錯誤: {str(e)}", gr.update(interactive=False)
+
+
 def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter_lat, vs30_input):
     """執行震度預測"""
     try:
@@ -614,7 +648,7 @@ def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter
         selected_stations = select_nearest_stations(st, epicenter_lat, epicenter_lon, n_stations=25)
 
         if len(selected_stations) == 0:
-            return None, None, "錯誤：找不到有效的測站資料"
+            return None, "錯誤：找不到有效的測站資料"
 
         # 3. 從選定的測站提取波形
         logger.info(f"提取波形資料（時間範圍: {start_time}-{end_time} 秒）...")
@@ -623,7 +657,7 @@ def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter
         )
 
         if len(waveforms) == 0:
-            return None, None, "錯誤：無法提取波形資料"
+            return None, "錯誤：無法提取波形資料"
 
         # 4. Padding 到 25 個測站（模型要求）
         max_stations = 25
@@ -660,24 +694,24 @@ def predict_intensity(event_name, start_time, end_time, epicenter_lon, epicenter
             pga_list = torch.sum(weight * mu, dim=2).cpu().detach().numpy().flatten().tolist()
 
         # 8. 繪製結果
-        waveform_plot = plot_waveform(st, selected_stations, start_time, end_time)
         intensity_plot = plot_intensity_map(pga_list, target_names)
 
         # 9. 統計資訊
         max_intensity = max([calculate_intensity(pga, label=True) for pga in pga_list])
-        stats = f"選取時間範圍: {start_time:.1f} - {end_time:.1f} 秒\n"
+        stats = f"✅ 預測完成！\n"
+        stats += f"選取時間範圍: {start_time:.1f} - {end_time:.1f} 秒\n"
         stats += f"震央位置: ({epicenter_lon:.4f}, {epicenter_lat:.4f})\n"
         stats += f"使用測站數: {len(waveforms)} / 25\n"
         stats += f"預測最大震度: {max_intensity}"
 
         logger.info("預測完成！")
-        return waveform_plot, intensity_plot, stats
+        return intensity_plot, stats
 
     except Exception as e:
         logger.error(f"預測過程發生錯誤: {e}")
         import traceback
         traceback.print_exc()
-        return None, None, f"錯誤: {str(e)}"
+        return None, f"錯誤: {str(e)}"
 
 
 # ============ Gradio 介面 ============
@@ -712,32 +746,46 @@ with gr.Blocks(title="TTSAM 震度預測系統") as demo:
                     info="剪力波速度（預設 600 m/s，若有 Vs30 資料會自動覆蓋）"
                 )
 
-            predict_btn = gr.Button("🔮 執行預測", variant="primary")
+            load_waveform_btn = gr.Button("📊 載入波形", variant="secondary")
+            predict_btn = gr.Button("🔮 執行預測", variant="primary", interactive=False)
 
             gr.Markdown("""
-            ### 說明
-            - 系統會根據震央位置自動選擇最近的 25 個測站
-            - 從選定的時間範圍提取波形資料（30 秒）
-            - 預測全台灣目標測站的震度分布
+            ### 使用步驟
+            1. 選擇地震事件和時間範圍
+            2. 輸入震央位置
+            3. 點擊「載入波形」確認波形範圍
+            4. 確認無誤後，點擊「執行預測」
+            
+            ℹ️ 系統會自動選擇距離震央最近的 25 個測站
             """)
 
+            info_output = gr.Textbox(label="狀態資訊", lines=5, interactive=False)
+
         # 右側：震度分布圖
         with gr.Column(scale=1):
             gr.Markdown("## 預測震度分布")
             intensity_plot = gr.Plot(label="震度分布圖")
-            stats_output = gr.Textbox(label="預測統計", lines=3)
+            stats_output = gr.Textbox(label="預測統計", lines=4)
 
     # 下方：波形圖
     with gr.Row():
         gr.Markdown("## 輸入波形")
     with gr.Row():
-        waveform_plot = gr.Plot(label="地震波形")
+        waveform_plot = gr.Plot(label="地震波形（前 10 個最近測站）")
 
     # 綁定事件
+    # 第一步：載入波形
+    load_waveform_btn.click(
+        fn=load_and_display_waveform,
+        inputs=[event_dropdown, start_slider, end_slider, epicenter_lon_input, epicenter_lat_input],
+        outputs=[waveform_plot, info_output, predict_btn]
+    )
+
+    # 第二步：執行預測
     predict_btn.click(
         fn=predict_intensity,
         inputs=[event_dropdown, start_slider, end_slider, epicenter_lon_input, epicenter_lat_input, vs30_input],
-        outputs=[waveform_plot, intensity_plot, stats_output]
+        outputs=[intensity_plot, stats_output]
     )
 
 demo.launch()