Update src/streamlit_app.py

src/streamlit_app.py

@@ -461,13 +461,24 @@ if submitted:
             surv_fn, landmarks = predict_patient_survival(cph, patient_design, years=[3, 5, 10])
         
             st.subheader("Individual adjusted survival curve")
+            # Convert days → years for plotting
+            surv_years = surv_fn.copy()
+            surv_years.index = surv_years.index / 365.25
+            
             fig, ax = plt.subplots(figsize=(8, 5))
-            surv_fn.plot(ax=ax, legend=False)
+            surv_years.plot(ax=ax, legend=False)
+            
             ax.set_title("Predicted adjusted survival curve")
-            ax.set_xlabel("Days")
+            ax.set_xlabel("Years")
             ax.set_ylabel("Survival probability")
+            
+            # Optional: make ticks clean integers
+            ax.set_xlim(left=0)
+            ax.xaxis.set_major_locator(plt.MaxNLocator(integer=True))
+            
             st.pyplot(fig, bbox_inches="tight")
             plt.clf()
+            
         
             st.write("Landmark survival probabilities:")
             for y in [3, 5, 10]: