Update app.py

app.py

@@ -137,6 +137,25 @@ def main():
         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]
+        
+        fig = plt.figure(figsize=(12,6))
+        ax = [fig.add_subplot(121, axisbelow=True),fig.add_subplot(122)]
+        ax[0].set_title('Monkeypox confirmed cases')
+        ax[0].plot(all_location[country]['total_cases'],lw=3,label='Infective')
+        ax[0].set_xlabel('Days')
+        ax[0].set_ylabel('Number of cases')
+        ax[0].legend()
+        
+        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')
+        ax[1].set_ylim(0,0.00005)
+        ax[1].set_xlim(0,200)
+        ax[1].set_xlabel('Days')
+        ax[1].set_ylabel('Fraction Number of cases')
+        ax[1].legend()
+        plt.tight_layout()
  
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Show Graph")
@@ -146,12 +165,12 @@ def main():
             SIR_param = SIR(country_code,R0,recovery,pop)
             
             st.pyplot(SIR_param[-1])
-            #st.pyplot(compare_plt(country_code,pop))
+            st.pyplot(fig)
             st.success("SIR model parameters for "+str(country)+" is")
             st.success("R0 = "+str(SIR_param[0]))
             st.success("Beta = "+str(SIR_param[2]))
             st.success("Gamma = "+str(SIR_param[3]))
-            st.success("RMPE = "+str(SIR_param[4]+"%"))
+            st.success("RMPE = "+str(SIR_param[4])+"%")
 
 # Run main()
 if __name__ == "__main__":