Update app.py

app.py

@@ -147,10 +147,15 @@ def main():
             # Show SIR
             SIR_param = SIR(country_code,R0,recovery,pop)
             
+            if SIR_param[0] <= 1:
+                a = 'No epidemic.'
+            else:
+                a = 'Epidemic has began.'
+            
             st.pyplot(SIR_param[-2])
             st.pyplot(SIR_param[-1])
             st.success("SIR model parameters of Thailand "+" is")
-            st.success("R0 (Basic Reproduction Number) = "+str(SIR_param[0]))
+            st.success("R0 (Basic Reproduction Number) = "+str(SIR_param[0])+' '+a)
             st.success("Beta (Rate of transmission) = "+str(round(SIR_param[2],3)))
             st.success("Gamma (Rate of Recovery) = "+str(round(SIR_param[3],3)))
             st.success("MAPE = "+str(round(SIR_param[4],3))+"%")