Update app.py

app.py

@@ -58,7 +58,6 @@ def plot_seismic_exploration(v1, v2, h, x_max, num_receivers, gain):
     receiver_x = np.linspace(0, x_max, int(num_receivers))
     
     # 計算每個測站的抵達時間
-    t_direct_rx = receiver_x / v1
     t_reflected_rx = np.sqrt(t0**2 + (receiver_x / v1)**2)
     t_first_arrival_rx = np.minimum(receiver_x / v1, (receiver_x / v2) + ti)
 
@@ -71,14 +70,14 @@ def plot_seismic_exploration(v1, v2, h, x_max, num_receivers, gain):
         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=1)
+        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.5)
+        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)
 
     # 繪製地表、震源與測站
@@ -86,15 +85,19 @@ def plot_seismic_exploration(v1, v2, h, x_max, num_receivers, gain):
     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=16, loc='left')
+    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)")
     ax2.set_xlim(-x_max * 0.05, x_max * 1.05)
     ax2.set_ylim(y_max * 1.1, -y_max*0.05)
-    ax2.legend(loc='lower left')
     
-    plt.tight_layout()
-    fig1.tight_layout()
+    # 【*** FIX 1: Move legend to a better position ***】
+    ax2.legend(loc='upper right', fontsize='small')
+    
+    # 【*** 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"""
@@ -134,7 +137,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             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)
-            gain_slider = gr.Slider(label="剖面增益 (Display Gain)", minimum=1, maximum=20, value=5, step=1)
+            # 【*** FIX 3: Lower default gain for clarity ***】
+            gain_slider = gr.Slider(label="剖面增益 (Display Gain)", minimum=1, maximum=20, value=4, step=1)
             
             submit_btn = gr.Button("🚀 開始探勘！", variant="primary")
             
@@ -147,8 +151,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         results_output = gr.Markdown("### 🔬 分析結果\n請設計你的地層模型並點擊「開始探勘！」以顯示計算結果。")
 
     # --- 事件監聽 ---
-    # 【*** THIS IS THE FIX ***】
-    # The incorrect 'demo.outputs[-1]' has been replaced with the correct variable name 'results_output'.
     submit_btn.click(
         fn=plot_seismic_exploration,
         inputs=[v1_slider, v2_slider, h_slider, xmax_slider, receivers_slider, gain_slider],