Update app.py

app.py

@@ -4,19 +4,23 @@ import matplotlib.pyplot as plt
 
 # --- 輔助函數：產生 Ricker 震波 ---
 def ricker_wavelet(t, f=25.0):
-    t = t - 2.0 / f
+    """ 產生一個 Ricker 震波 (墨西哥帽函數) """
+    t = t - 2.0 / f # 將震波峰值對齊時間點
     p = (np.pi * f * t) ** 2
     return (1 - 2 * p) * np.exp(-p)
 
 # --- 核心計算與繪圖函數 ---
-def plot_seismic_exploration(v1, v2, v3, h1, h2, x_max, num_receivers, gain):
+# 【*** FIX 1: Add scenario_name_val as a function parameter ***】
+def plot_seismic_exploration(scenario_name_val, 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)。"
+        error_msg = "### ⚠️ 模型警告\n速度必須隨深度增加 (V3 > V2 > V1)，折射波分析可能無效。"
 
     # 計算關鍵物理量 (第一層介面)
     t0_1 = (2 * h1) / v1
@@ -27,55 +31,57 @@ def plot_seismic_exploration(v1, v2, v3, h1, h2, x_max, num_receivers, gain):
     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(0, color='saddlebrown', 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.text(x_max/2, -h1/2, f'Layer 1\nV1 = {v1:.0f} m/s\nh1 = {h1:.0f} m', ha='center', va='center', fontsize=9, 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=9, 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=9, color='white')
     ax0.set_title("Geological Model Cross-section")
     ax0.set_ylabel("Depth (m)")
     ax0.set_yticks([0, -h1, -(h1+h2)])
     ax0.set_xticks([])
+    fig0.tight_layout(pad=1.1)
 
     # === PART 3: 繪製 T-X 走時曲線圖 ===
     x_continuous = np.linspace(0, x_max, 500)
-    # 反射波
+    # 反射波 (使用RMS速度近似)
+    v_rms_2 = np.sqrt((v1**2 * 2*h1/v1 + v2**2 * 2*h2/v2) / (2*h1/v1 + 2*h2/v2))
     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
+    t_refl_2 = np.sqrt(t0_2**2 + (x_continuous / v_rms_2)**2)
     
     fig1, ax1 = plt.subplots(figsize=(10, 6))
     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
+        ti_23 = 2 * h1 * np.sqrt(v3**2 - v1**2)/(v1*v3) + 2 * h2 * np.sqrt(v3**2 - v2**2)/(v2*v3)
         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.legend(fontsize='small')
     ax1.grid(True)
     ax1.set_xlim(0, x_max)
     y_max = np.max(t_refl_2) * 1.1
     ax1.set_ylim(0, y_max)
+    fig1.tight_layout(pad=1.1)
     
     # === PART 4: 繪製視覺化震測剖面圖 ===
     fig2, ax2 = plt.subplots(figsize=(10, 5))
     receiver_x = np.linspace(0, x_max, int(num_receivers))
     # 反射波到時
     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)
+    t_refl_2_rx = np.sqrt(t0_2**2 + (receiver_x / v_rms_2)**2)
 
     wavelet_duration = y_max / 10
     wavelet_t = np.linspace(0, wavelet_duration, 100)
@@ -103,14 +109,14 @@ def plot_seismic_exploration(v1, v2, v3, h1, h2, x_max, num_receivers, gain):
     # === PART 5: 準備探勘日誌 ===
     log_md = f"""
     ### 📝 現場探勘日誌
-    **任務目標**: {scenario_name.get() if 'scenario_name' in globals() else '自訂模式'}
+    **任務目標**: {scenario_name_val}
     **儀器設定**: {int(num_receivers)} 個測站, 測線長度 {x_max} 公尺。
 
     **初步分析**:
     - **第一介面反射 (黑色震波)**: 雙程走時 (TWT) 約 **{t0_1*1000:.1f} ms**。
     - **第二介面反射 (藍色震波)**: 雙程走時 (TWT) 約 **{t0_2*1000:.1f} ms**。
     
-    {'' if valid_model else '**警告**: 模型速度設定有誤 (V 未隨深度增加)，折射波分析可能無效。'}
+    {error_msg}
     """
     return fig0, fig1, fig2, log_md
 
@@ -167,9 +173,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     scenario_dropdown.change(lambda x: x, inputs=scenario_dropdown, outputs=scenario_name)
     
+    # 【*** FIX 2: Add scenario_name to the inputs list ***】
     submit_btn.click(
         fn=plot_seismic_exploration,
-        inputs=[v1_slider, v2_slider, v3_slider, h1_slider, h2_slider, xmax_slider, receivers_slider, gain_slider],
+        inputs=[scenario_name, 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] 
     )