Update app.py

app.py

@@ -4,181 +4,189 @@ import matplotlib.pyplot as plt
 
 # --- 輔助函數：產生 Ricker 震波 ---
 def ricker_wavelet(t, f=25.0):
-    """ 產生一個 Ricker 震波 (墨西哥帽函數) """
-    t = t - 2.0 / f # 將震波峰值對齊時間點
+    t = t - 2.0 / f
     p = (np.pi * f * t) ** 2
     return (1 - 2 * p) * np.exp(-p)
 
 # --- 核心計算與繪圖函數 ---
-def plot_seismic_exploration(v1, v2, h, x_max, num_receivers, gain):
-    """
-    根據輸入的地層參數，計算並繪製震測的走時曲線與視覺化震測剖面圖。
-    """
-    # === PART 1: 物理計算 ===
-    if v2 <= v1:
-        fig1, ax1 = plt.subplots(figsize=(10, 6))
-        ax1.text(0.5, 0.5, 'Error: V2 must be greater than V1', ha='center', va='center', color='red')
-        ax1.set_title("Travel-Time Curve")
-        fig2, ax2 = plt.subplots(figsize=(10, 5))
-        ax2.text(0.5, 0.5, 'Please ensure V2 > V1 to generate a valid model.', ha='center', va='center', color='red')
-        ax2.set_title("Visualized Seismic Profile")
-        return fig1, fig2, "### 參數錯誤\n請確保第二層速度 (V2) 大於第一層速度 (V1)。"
-
-    theta_c_rad = np.arcsin(v1 / v2)
-    theta_c_deg = np.rad2deg(theta_c_rad)
-    ti = (2 * h * np.cos(theta_c_rad)) / v1
-    xc = 2 * h * np.sqrt((v2 + v1) / (v2 - v1))
-    t0 = (2 * h) / v1
-
-    # === PART 2: 繪製 T-X 走時曲線圖 (Plot 1) ===
+def plot_seismic_exploration(v1, v2, v3, h1, h2, x_max, num_receivers, gain):
+    # === PART 1: 物理計算 (升級至三層模型) ===
+    # 物理條件檢查
+    valid_model = True
+    error_msg = ""
+    if v2 <= v1 or v3 <= v2:
+        valid_model = False
+        error_msg = "### 模型錯誤\n速度必須隨深度增加 (V3 > V2 > V1)。"
+
+    # 計算關鍵物理量 (第一層介面)
+    t0_1 = (2 * h1) / v1
+    # 計算關鍵物理量 (第二層介面)
+    t0_2 = (2 * h1 / v1) + (2 * h2 / v2)
+    
+    # === PART 2: 繪製地質模型圖 (新圖表) ===
+    fig0, ax0 = plt.subplots(figsize=(10, 2))
+    ax0.set_xlim(0, x_max)
+    ax0.set_ylim(-(h1 + h2) * 1.5, 5)
+    ax0.axhline(0, color='brown', linewidth=3)
+    ax0.axhline(-h1, color='gray', linestyle='--')
+    ax0.axhline(-(h1+h2), color='darkgray', linestyle='--')
+    ax0.fill_between([0, x_max], 0, -h1, color='sandybrown', alpha=0.6)
+    ax0.fill_between([0, x_max], -h1, -(h1+h2), color='darkkhaki', alpha=0.6)
+    ax0.fill_between([0, x_max], -(h1+h2), -(h1 + h2) * 1.5, color='dimgray', alpha=0.6)
+    ax0.text(x_max/2, -h1/2, f'Layer 1\nV1 = {v1:.0f} m/s\nh1 = {h1:.0f} m', ha='center', va='center', fontsize=10, color='black')
+    ax0.text(x_max/2, -h1-h2/2, f'Layer 2\nV2 = {v2:.0f} m/s\nh2 = {h2:.0f} m', ha='center', va='center', fontsize=10, color='black')
+    ax0.text(x_max/2, -(h1+h2)*1.25, f'Layer 3 (Basement)\nV3 = {v3:.0f} m/s', ha='center', va='center', fontsize=10, color='white')
+    ax0.set_title("Geological Model Cross-section")
+    ax0.set_ylabel("Depth (m)")
+    ax0.set_yticks([0, -h1, -(h1+h2)])
+    ax0.set_xticks([])
+
+    # === PART 3: 繪製 T-X 走時曲線圖 ===
     x_continuous = np.linspace(0, x_max, 500)
-    t_direct = x_continuous / v1
-    t_refracted = (x_continuous / v2) + ti
-    t_reflected = np.sqrt(t0**2 + (x_continuous / v1)**2)
-    t_first_arrival_continuous = np.minimum(t_direct, t_refracted)
+    # 反射波
+    t_refl_1 = np.sqrt(t0_1**2 + (x_continuous / v1)**2)
+    t_refl_2 = np.sqrt(t0_2**2 + (x_continuous / ((v1*h1 + v2*h2)/(h1+h2)) )**2) # RMS velocity approximation
     
     fig1, ax1 = plt.subplots(figsize=(10, 6))
-    ax1.plot(x_continuous, t_direct, 'b--', label='Direct Wave')
-    ax1.plot(x_continuous, t_refracted, 'g--', label='Refracted Wave')
-    ax1.plot(x_continuous, t_reflected, 'm:', linewidth=2, label='Reflected Wave')
-    ax1.plot(x_continuous, t_first_arrival_continuous, 'r-', linewidth=3, label='First Arrival')
-    if xc < x_max:
-        ax1.axvline(x=xc, color='k', linestyle=':', label=f'Crossover = {xc:.1f} m')
-    ax1.set_title("1. Travel-Time (T-X) Curve", fontsize=16, loc='left')
-    ax1.set_xlabel("Distance (m)")
-    ax1.set_ylabel("Travel Time (s)")
+    ax1.plot(x_continuous, t_refl_1, 'm:', linewidth=2, label='Reflection 1 (from Layer 2)')
+    ax1.plot(x_continuous, t_refl_2, 'c:', linewidth=2, label='Reflection 2 (from Layer 3)')
+    
+    # 折射波 (僅在速度增加時繪製)
+    if valid_model:
+        theta_c12 = np.arcsin(v1 / v2)
+        ti_12 = (2 * h1 * np.cos(theta_c12)) / v1
+        t_refr_12 = (x_continuous / v2) + ti_12
+        ax1.plot(x_continuous, t_refr_12, 'g--', label='Refraction (from Layer 2)')
+        
+        theta_c23 = np.arcsin(v2 / v3)
+        ti_23 = 2 * h1 * np.cos(np.arcsin(v1/v3))/v1 + 2 * h2 * np.cos(theta_c23)/v2
+        t_refr_23 = (x_continuous / v3) + ti_23
+        ax1.plot(x_continuous, t_refr_23, 'y--', label='Refraction (from Layer 3)')
+
+    ax1.set_title("1. Travel-Time (T-X) Curve")
     ax1.legend()
     ax1.grid(True)
     ax1.set_xlim(0, x_max)
-    y_max = max(np.max(t_direct), np.max(t_reflected))
-    ax1.set_ylim(0, y_max * 1.1)
+    y_max = np.max(t_refl_2) * 1.1
+    ax1.set_ylim(0, y_max)
     
-    # === PART 3: 繪製視覺化震測剖面圖 (Plot 2) ===
+    # === PART 4: 繪製視覺化震測剖面圖 ===
     fig2, ax2 = plt.subplots(figsize=(10, 5))
     receiver_x = np.linspace(0, x_max, int(num_receivers))
-    
-    # 計算每個測站的抵達時間
-    t_reflected_rx = np.sqrt(t0**2 + (receiver_x / v1)**2)
-    t_first_arrival_rx = np.minimum(receiver_x / v1, (receiver_x / v2) + ti)
+    # 反射波到時
+    t_refl_1_rx = np.sqrt(t0_1**2 + (receiver_x / v1)**2)
+    t_refl_2_rx = np.sqrt(t0_2**2 + (receiver_x / ((v1*h1 + v2*h2)/(h1+h2)) )**2)
 
-    # 繪製每一條帶有震波的震波線
-    wavelet_duration = 0.08
+    wavelet_duration = y_max / 10
     wavelet_t = np.linspace(0, wavelet_duration, 100)
     
     for i in range(int(num_receivers)):
-        # 繪製反射波震波
-        wavelet_amp_refl = ricker_wavelet(wavelet_t, f=40) * gain
-        x_trace_refl = receiver_x[i] + wavelet_amp_refl
-        y_trace_refl = t_reflected_rx[i] - wavelet_duration/2 + wavelet_t
-        ax2.plot(x_trace_refl, y_trace_refl, 'k-', linewidth=0.8)
-        ax2.fill_betweenx(y_trace_refl, receiver_x[i], x_trace_refl, where=(x_trace_refl > receiver_x[i]), color='black')
-
-        # 繪製初達波震波
-        wavelet_amp_first = ricker_wavelet(wavelet_t, f=30) * gain * 1.2
-        x_trace_first = receiver_x[i] + wavelet_amp_first
-        y_trace_first = t_first_arrival_rx[i] - wavelet_duration/2 + wavelet_t
-        ax2.plot(x_trace_first, y_trace_first, 'r-', linewidth=1)
-        ax2.fill_betweenx(y_trace_first, receiver_x[i], x_trace_first, where=(x_trace_first > receiver_x[i]), color='red', alpha=0.8)
-
-    # 繪製地表、震源與測站
-    ax2.axhline(0, color='brown', linewidth=2)
-    ax2.plot(0, 0, 'r*', markersize=20, label='Source')
-    ax2.plot(receiver_x, np.zeros_like(receiver_x), 'kv', markersize=8, label='Receivers')
-
-    ax2.set_title(f"2. Visualized Seismic Profile ({int(num_receivers)} Traces)", fontsize=14, loc='left')
-    ax2.set_xlabel("Distance (m)")
-    ax2.set_ylabel("Two-Way Time (s)")
+        # 繪製第一層反射
+        wavelet_amp_1 = ricker_wavelet(wavelet_t, f=40) * gain
+        x_trace_1 = receiver_x[i] + wavelet_amp_1
+        y_trace_1 = t_refl_1_rx[i] - wavelet_duration/2 + wavelet_t
+        ax2.plot(x_trace_1, y_trace_1, 'k-', linewidth=0.8)
+        ax2.fill_betweenx(y_trace_1, receiver_x[i], x_trace_1, where=(x_trace_1 > receiver_x[i]), color='black')
+
+        # 繪製第二層反射
+        wavelet_amp_2 = ricker_wavelet(wavelet_t, f=30) * gain * 0.8 # Deeper reflections are weaker
+        x_trace_2 = receiver_x[i] + wavelet_amp_2
+        y_trace_2 = t_refl_2_rx[i] - wavelet_duration/2 + wavelet_t
+        ax2.plot(x_trace_2, y_trace_2, 'b-', linewidth=0.8)
+        ax2.fill_betweenx(y_trace_2, receiver_x[i], x_trace_2, where=(x_trace_2 > receiver_x[i]), color='blue')
+    
+    ax2.set_title(f"2. Visualized Seismic Profile ({int(num_receivers)} Traces)")
+    ax2.set_ylim(y_max, -y_max*0.05)
     ax2.set_xlim(-x_max * 0.05, x_max * 1.05)
-    ax2.set_ylim(y_max * 1.1, -y_max*0.05)
+    fig2.subplots_adjust(left=0.1, right=0.98, top=0.9, bottom=0.15)
     
-    # 【*** FIX 1: Move legend to a better position ***】
-    ax2.legend(loc='upper right', fontsize='small')
+    # === PART 5: 準備探勘日誌 ===
+    log_md = f"""
+    ### 📝 現場探勘日誌
+    **任務目標**: {scenario_name.get() if 'scenario_name' in globals() else '自訂模式'}
+    **儀器設定**: {int(num_receivers)} 個測站, 測線長度 {x_max} 公尺。
+
+    **初步分析**:
+    - **第一介面反射 (黑色震波)**: 雙程走時 (TWT) 約 **{t0_1*1000:.1f} ms**。
+    - **第二介面反射 (藍色震波)**: 雙程走時 (TWT) 約 **{t0_2*1000:.1f} ms**。
     
-    # 【*** FIX 2: Manually adjust layout to prevent overlaps ***】
-    fig1.tight_layout(pad=1.1)
-    fig2.subplots_adjust(left=0.1, right=0.98, top=0.9, bottom=0.15)
-
-
-    # === PART 4: 準備輸出的說明文字 ===
-    results_md = f"""
-    ### 🔬 分析結果
-
-    根據您設計的地層模型，我們計算出以下關鍵物理量：
+    {'' if valid_model else '**警告**: 模型速度設定有誤 (V 未隨深度增加)，折射波分析可能無效。'}
+    """
+    return fig0, fig1, fig2, log_md
 
-    #### 折射波 (Refracted Wave)
-    - **臨界角 (Critical Angle, θc)**: `arcsin({v1:.0f} / {v2:.0f})` = **{theta_c_deg:.2f}°**
-    - **截時 (Intercept Time, tᵢ)**: `(2 * {h:.0f} * cos({theta_c_deg:.2f}°)) / {v1:.0f}` = **{ti*1000:.1f} ms**
-    - **交越距離 (Crossover Distance, Xc)**: `2 * {h:.0f} * sqrt(...)` = **{xc:.1f} m**
+# --- Gradio 介面與任務設定 ---
+scenarios = {
+    "自訂模式 (Custom Mode)": {"v1": 800, "v2": 2500, "v3": 4500, "h1": 20, "h2": 50},
+    "尋找淺層地下水 (Find Groundwater)": {"v1": 500, "v2": 2200, "v3": 3500, "h1": 15, "h2": 40},
+    "桃園台地工程鑽探 (Taoyuan Engineering)": {"v1": 600, "v2": 1800, "v3": 3000, "h1": 10, "h2": 30},
+    "油氣田探勘 (Oil & Gas Prospecting)": {"v1": 1500, "v2": 2800, "v3": 4200, "h1": 100, "h2": 250},
+}
 
-    ---
-    #### 反射波 (Reflected Wave)
-    - **雙程走時 (Two-Way Time, t₀)**: `2 * {h:.0f} / {v1:.0f}` = **{t0*1000:.1f} ms**
-    """
-    return fig1, fig2, results_md
+def update_sliders(scenario_key):
+    params = scenarios[scenario_key]
+    return params['v1'], params['v2'], params['v3'], params['h1'], params['h2']
 
-# --- Gradio 介面設定 ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        """
-        # 地心震波奇幻之旅：地球物理遊樂場 🌍
-        > 創意的發揮是一種學習，過程中，每個人同時是學生也是老師。
-        
-        這個實驗室就是你的遊樂場。透過親手設計地層模型、佈放虛擬測站，你將不只是學習，更是在**創造和發現**地底下的物理法則。
-        """
-    )
+    scenario_name = gr.State("自訂模式 (Custom Mode)")
+    gr.Markdown("# 地球物理探勘總部 🛰️")
     
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### ⚙️ 1. 設計你的地層模型")
-            v1_slider = gr.Slider(label="V1: 第一層速度 (m/s)", minimum=300, maximum=3000, value=800, step=50)
-            v2_slider = gr.Slider(label="V2: 第二層速度 (m/s)", minimum=500, maximum=6000, value=2500, step=50)
-            h_slider = gr.Slider(label="h: 第一層厚度 (m)", minimum=5, maximum=100, value=20, step=1)
+            gr.Markdown("### 🎯 1. 選擇探勘任務")
+            scenario_dropdown = gr.Dropdown(list(scenarios.keys()), label="Select Mission", value="自訂模式 (Custom Mode)")
+            
+            gr.Markdown("### ⚙️ 2. 微調地層參數")
+            v1_slider = gr.Slider(label="V1 (m/s)", minimum=300, maximum=5000, value=800, step=50)
+            h1_slider = gr.Slider(label="h1 (m)", minimum=5, maximum=500, value=20, step=5)
+            v2_slider = gr.Slider(label="V2 (m/s)", minimum=500, maximum=6000, value=2500, step=50)
+            h2_slider = gr.Slider(label="h2 (m)", minimum=10, maximum=1000, value=50, step=10)
+            v3_slider = gr.Slider(label="V3 (m/s)", minimum=1000, maximum=8000, value=4500, step=50)
             
-            gr.Markdown("### ⚙️ 2. 佈放儀器與顯示設定")
-            xmax_slider = gr.Slider(label="最大觀測距離 (m)", minimum=100, maximum=500, value=250, step=10)
-            receivers_slider = gr.Slider(label="測站數量", minimum=5, maximum=100, value=40, step=1)
-            # 【*** FIX 3: Lower default gain for clarity ***】
-            gain_slider = gr.Slider(label="剖面增益 (Display Gain)", minimum=1, maximum=20, value=4, step=1)
+            gr.Markdown("### 📡 3. 設定儀器")
+            xmax_slider = gr.Slider(label="最大觀測距離 (m)", minimum=100, maximum=2000, value=500, step=50)
+            receivers_slider = gr.Slider(label="測站數量", minimum=10, maximum=200, value=50, step=5)
+            gain_slider = gr.Slider(label="剖面增益", minimum=1, maximum=20, value=4, step=1)
             
-            submit_btn = gr.Button("🚀 開始探勘！", variant="primary")
+            submit_btn = gr.Button("🚀 發射震波！", variant="primary")
             
         with gr.Column(scale=2):
-            gr.Markdown("### 📊 觀測結果")
-            plot_output1 = gr.Plot(label="走時-距離圖 (T-X Plot)")
-            plot_output2 = gr.Plot(label="視覺化震測剖面圖 (Visualized Seismic Profile)")
+            gr.Markdown("### 🗺️ 地質模型")
+            plot_output0 = gr.Plot(label="Geological Model")
+            gr.Markdown("### 📊 探勘數據")
+            plot_output1 = gr.Plot(label="走時-距離圖")
+            plot_output2 = gr.Plot(label="視覺化震測剖面圖")
             
     with gr.Row():
-        results_output = gr.Markdown("### 🔬 分析結果\n請設計你的地層模型並點擊「開始探勘！」以顯示計算結果。")
+        log_output = gr.Markdown("### 📝 現場探勘日誌\n請選擇任務或調整參數，然後點擊「發射震波！」")
 
     # --- 事件監聽 ---
+    scenario_dropdown.change(
+        fn=update_sliders,
+        inputs=scenario_dropdown,
+        outputs=[v1_slider, v2_slider, v3_slider, h1_slider, h2_slider]
+    )
+    scenario_dropdown.change(lambda x: x, inputs=scenario_dropdown, outputs=scenario_name)
+    
     submit_btn.click(
         fn=plot_seismic_exploration,
-        inputs=[v1_slider, v2_slider, h_slider, xmax_slider, receivers_slider, gain_slider],
-        outputs=[plot_output1, plot_output2, results_output] 
+        inputs=[v1_slider, v2_slider, v3_slider, h1_slider, h2_slider, xmax_slider, receivers_slider, gain_slider],
+        outputs=[plot_output0, plot_output1, plot_output2, log_output] 
     )
     
     gr.Markdown(
         """
         ---
-        ### 📖 剖面圖是如何誕生的？
-        下方的 **視覺化震測剖面圖** 完美模擬了真實的探勘情���。每一條黑色的垂直線代表一個**測站 (Receiver)**，它記錄到的訊號就是一條帶有**震波 (Wiggle)** 的**震波線 (Trace)**。
-        
-        - **紅色震波** 代表 **初達波 (First Arrival)**，是能量最早抵達的波。
-        - **黑色震波** 代表 **反射波 (Reflected Wave)**，它們來自地下介面的反射。
-        
-        地球物理學家最重要的工作，就是在成千上萬條震波線中，**尋找並追蹤這些連續排列的震波（稱為「同相軸」）**。例如，圖中那條優美的黑色雙曲線同相軸，就清楚地標示出了地下第一層介面的位置！
-        
-        ### 🚀 探索與發現
-        1.  **增益的效果**: 試著調整「剖面增益」，看看震波的振幅如何變化。在真實資料中，深層的反射信號很微弱，就需要提高增益才能看清楚。
-        2.  **看見雙曲線**: 專注觀察剖面圖中的黑色震波。當你增加「測站數量」時，是不是能更清楚地「描繪」出那條對應到上方 T-X 圖的紫色雙曲線？
-        3.  **初達波的威力**: 紅色的初達波在剖面圖中形成了一條明顯的分界線。觀察它的轉折點，思考一下這個轉折點（交越距離）告訴了我們關於地下速度結構的什麼資訊？
+        ### 🧠 總工程師的挑戰
+        1.  **看見儲油構造**: 在「油氣田探勘」任務中，來自第二介面（藍色震波）的反射同相軸呈現一個向上彎曲的「背斜」形狀，這正是油氣最喜歡聚集的地方！你能透過微調 `h1` 和 `h2` 讓這個構造更明顯嗎？
+        2.  **折射的極限**: 試著在自訂模式中，將 `V2` 調得比 `V1` 慢，看看走時圖和日誌會出現什麼警告？這在真實地質中稱為「低速帶」，是折射法的一大挑戰。
+        3.  **解析度問題**: 將「測站數量」調到最低，再慢慢增加。你需要多少個測站，才能清楚地分辨出剖面圖中來自兩個不同介面的反射波？這就是探勘的「解析度」概念。
         """
     )
     
     gr.HTML("""
         <footer style="text-align:center; margin-top: 30px; color:grey;">
             <p>「創意的發揮是一種學習，過程中，每個人同時是學生也是老師。」</p>
-            <p>地球物理遊樂場 &copy; 2025 - 由 Gemini 根據課程文件與靈感生成</p>
+            <p>地球物理探勘總部 &copy; 2025 - 由 Gemini 根據課程文件與靈感生成</p>
         </footer>
     """)