Update app.py

app.py

@@ -111,7 +111,7 @@ def SIR(country,R0,t_infective):
 
   return R0,t_infective,beta,gamma,rmpe,plotdata(t, s, i,r,R0)
   
-def compare_plt(country):
+def compare_plt(country,pop):
     fig = plt.figure(figsize=(12,6))
     ax = [fig.add_subplot(121, axisbelow=True),fig.add_subplot(122)]
     ax[0].set_title('Monkeypox confirmed cases')
@@ -120,7 +120,7 @@ def compare_plt(country):
     ax[0].set_ylabel('Number of cases')
     ax[0].legend()
     
-    scaler = all_location[country]['total_cases'].apply(lambda x : x/pop_dict[country])
+    scaler = all_location[country]['total_cases'].apply(lambda x : x/pop)
     ax[1].set_title('Monkeypox confirmed cases compare with model')
     ax[1].plot(scaler,lw=3,label='Real Infective')
     ax[1].plot(i,lw=3,label='SIR model Infective')
@@ -141,6 +141,7 @@ def main():
         recovery = st.slider("How long Monkeypox recover?", 21, 31, 21)
         R0 = st.slider("Basic Reproduction Number (R0)", 0.57, 3.00, 0.57)# user's input
         country_code = code[country][0]
+        pop = pop_dict[country_code]
  
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Show Graph")
@@ -150,7 +151,7 @@ def main():
             SIR_param = SIR(country_code,R0,recovery)
             
             st.pyplot(SIR_param[-1])
-            st.pyplot(compare_plt(country_code))
+            st.pyplot(compare_plt(country_code,pop))
             st.success("SIR model parameters for "+str(country)+" is")
             st.success("R0 = "+str(SIR_param[0]))
             st.success("Beta = "+str(SIR_param[2]))