initial commit

app.py

@@ -0,0 +1,76 @@
+import altair
+
+import gradio as gr
+from math import sqrt
+import matplotlib
+
+matplotlib.use("Agg")
+
+import matplotlib.pyplot as plt
+import numpy as np
+import plotly.express as px
+import pandas as pd
+
+
+def outbreak(plot_type, r, month, countries, social_distancing):
+    months = ["January", "February", "March", "April", "May"]
+    m = months.index(month)
+    start_day = 30 * m
+    final_day = 30 * (m + 1)
+    x = np.arange(start_day, final_day + 1)
+    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)
+        return fig
+    elif plot_type == "Plotly":
+        fig = px.line(df, x="day", y=countries)
+        fig.update_layout(
+            title="Outbreak in " + month,
+            xaxis_title="Cases",
+            yaxis_title="Days Since Day 0",
+        )
+        return fig
+    elif plot_type == "Altair":
+        df = df.melt(id_vars="day").rename(columns={"variable": "country"})
+        fig = altair.Chart(df).mark_line().encode(x="day", y='value', color='country')
+        return fig
+    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?"),
+]
+outputs = gr.Plot()
+
+demo = gr.Interface(
+    fn=outbreak,
+    inputs=inputs,
+    outputs=outputs,
+    examples=[
+        ["Matplotlib", 2, "March", ["Mexico", "UK"], True],
+        ["Altair", 2, "March", ["Mexico", "Canada"], True],
+        ["Plotly", 3.6, "February", ["Canada", "Mexico", "UK"], False],
+    ],
+    cache_examples=True,
+)
+
+if __name__ == "__main__":
+    demo.launch()