Upload 2 files

src/package/app.py

@@ -98,7 +98,7 @@ def calculate_reference_value_k(h: str, arl_0: str) -> float:
     k = get_ref_value_k(h=h, ARL_0=arl_0)
     k = "{:.2f}".format(k)
 
-    return k
+    return k, k, k
 
 
 def calculate_arl1_h_k_mu1(h: str, k: str, mu1: str) -> float:
@@ -220,7 +220,7 @@ with gr.Blocks(
             h_phase1 = gr.Textbox(
                 label="h value =",
                 placeholder="h = normalized threshold, default = 4. Range: between 4 and 5 ([4, 5])",
-                value="4",
+                value="3",
                 autofocus=True,
             )
 
@@ -301,7 +301,7 @@ with gr.Blocks(
                 h_phase2 = gr.Textbox(
                     label="h value =",
                     placeholder="normalized threshold, default = 4. Range: between 4 and 5 ([4, 5])",
-                    value="4",
+                    value="3",
                 )
 
                 k_phase2 = gr.Textbox(
@@ -343,7 +343,7 @@ with gr.Blocks(
 
     # Calculate specific k for ARL_0
     button_calculate_k.click(
-        fn=calculate_reference_value_k, inputs=[h_phase1, arl_0], outputs=[output_k]
+        fn=calculate_reference_value_k, inputs=[h_phase1, arl_0], outputs=[output_k, k_phase1, k_phase2]
     )
     button_calculate_k.click(
         fn=lambda: gr.update(visible=True), inputs=[], outputs=output_k

src/package/cusum.py

@@ -588,7 +588,7 @@ class CUSUM:
 
         fig.add_shape(
             type="rect",
-            x0=0, x1=60,
+            x0=0, x1=self.pre_change_days,
             y0=0, y1=1,  # use relative values (0 to 1) for full y-range
             xref="x", yref="paper",  # "paper" for full plot height
             fillcolor=self.config["color"]["blue_005"],
@@ -599,7 +599,7 @@ class CUSUM:
 
         fig.add_shape(
             type="rect",
-            x0=60, x1=len(self.S_lo),  # x1=1 means extend to right edge of plot (paper coordinates)
+            x0=self.pre_change_days, x1=len(self.S_lo),  # x1=1 means extend to right edge of plot (paper coordinates)
             y0=0, y1=1,
             xref="x", yref="paper",
             fillcolor="rgb(253, 243, 235)",