Update app.py

app.py

@@ -16,42 +16,39 @@ def outbreak(plot_type, r, month, countries, social_distancing):
     pop_count = {"USA": 350, "Canada": 40, "Mexico": 300, "UK": 120}
     if social_distancing:
         r = sqrt(r)
-df = pd.DataFrame({"day": x})
-for country in countries:
-    df[country] = x ** (r) * (pop_count[country] + 1)
-
-    if plot_type == "Matplotlib":
-        fig = plt.figure()
-        plt.plot(df["day"], df[countries].to_numpy())
-        plt.title("Outbreak in " + month)
-        plt.ylabel("Cases")
-        plt.xlabel("Days since Day 0")
-        plt.legend(countries)
+    df = pd.DataFrame({"day": x})
+    for country in countries:
+        df[country] = x ** (r) * (pop_count[country] + 1)
+        if plot_type == "Matplotlib":
+            fig = plt.figure()
+            plt.plot(df["day"], df[countries].to_numpy())
+            plt.title("Outbreak in " + month)
+            plt.ylabel("Cases")
+            plt.xlabel("Days since Day 0")
+            plt.legend(countries)
             return fig
-    else:
-        raise ValueError("A plot type must be selected")
-
-
+        else:
+            raise ValueError("A plot type must be selected")
 inputs = [
-gr.Dropdown(["Matplotlib", "Plotly", "Altair"], label="Plot Type"),
-gr.Slider(1, 4, 3.2, label="R"),
-gr.Dropdown(["January", "February", "March", "April", "May"], label="Month"),
-gr.CheckboxGroup(
-["USA", "Canada", "Mexico", "UK"], label="Countries", value=["USA", "Canada"]
-),
-gr.Checkbox(label="Social Distancing?"),
-]
+    gr.Dropdown(["Matplotlib", "Plotly", "Altair"], label="Plot Type"),
+    gr.Slider(1, 4, 3.2, label="R"),
+    gr.Dropdown(["January", "February", "March", "April", "May"], label="Month"),
+    gr.CheckboxGroup(
+        ["USA", "Canada", "Mexico", "UK"], label="Countries", value=["USA", "Canada"]
+    ),
+    gr.Checkbox(label="Social Distancing?"),
+    ]
 outputs = gr.Plot()
 
 demo = gr.Interface(
-fn=outbreak,
-inputs=inputs,
-outputs=outputs,
-examples=[
-["Matplotlib", 2, "March", ["Mexico", "UK"], True]
-],
-cache_examples=True,
-)
+    fn=outbreak,
+    inputs=inputs,
+    outputs=outputs,
+    examples=[
+        ["Matplotlib", 2, "March", ["Mexico", "UK"], True]
+    ],
+    cache_examples=True,
+    )
 
 if __name__ == "__main__":
     demo.launch(share=False)