Update app.py

app.py

@@ -9,13 +9,25 @@ warnings.filterwarnings("ignore")
 #dowload file
 
 #read files
-data = pd.read_csv('myfile.csv')
-data = data[['location','date','new_cases','total_cases','new_deaths','total_deaths']]
+data = pd.read_csv('owid-monkeypox-data.csv')
+data = data[['location','iso_code','date','new_cases','total_cases','new_deaths','total_deaths']]
 
+pop = pd.read_csv('API_SP.POP.TOTL_DS2_en_csv_v2_4578059.csv')
 #preprocessiong data
 all_location = {}
-for i in data['location'].unique():
-  all_location[i] = data[data['location'] == i].reset_index(drop=True)
+for i in data['iso_code'].unique():
+  all_location[i] = data[data['iso_code'] == i].reset_index(drop=True)
+
+popu = pop[['Country Code','2021']].to_dict('index')
+pop_dict = {}
+for i in popu.values():
+  pop_dict[i['Country Code']] = i['2021']
+
+pop_dict['GLP'] = 400000
+pop_dict['MTQ'] = 376480
+pop_dict['OWID_WRL'] = 7836630792
+
+code = dict(data.groupby('location')['iso_code'].unique())
 
 # SIR model differential equations.
 def deriv(x, t, beta, gamma):
@@ -70,17 +82,16 @@ def plotdata(t, s, i,r,R0, e=None):
 #final model
 def SIR(country,t_infective):
   # parameter values
-  R0 = (all_location[country]['new_cases'].sum()/len(all_location[country]['date'].unique()))/t_infective
   t_infective = t_infective
+  gamma = 1/t_infective
+  beta = (all_location[country]['new_cases'].sum()/pop_dict[country])/len(all_location[country]['date'].unique())
+  R0 = beta/gamma
 
   # initial number of infected and recovered individuals
-  i_initial = 1/20000
+  i_initial = all_location[country]['new_cases'].sum()/pop_dict[country]
   r_initial = 0.00
   s_initial = 1 - i_initial - r_initial
-
-  gamma = 1/t_infective
-  beta = R0*gamma
-
+  
   t = np.linspace(0, 100, 1000)
   x_initial = s_initial, i_initial, r_initial
   soln = odeint(deriv, x_initial, t, args=(beta, gamma))
@@ -96,14 +107,11 @@ def main():
     with st.form("questionaire"):
         country = st.selectbox("Country",data['location'].unique())# user's input
         recovery = st.slider("How long Monkeypox recover?", 21, 31, 21)# user's input
-
+        country = code[country][0]
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Show Graph")
         if clicked:
-
-            #show total cases graph
-            all_location[country]['total_cases'].plot()
-
+        
             # Show SIR
             SIR_param = SIR(country,recovery)