mirror of
https://github.com/gradio-app/gradio.git
synced 2025-02-11 11:19:58 +08:00
bed288a509
* Try clean install * Resolve peer dependencies? * frozen lockfile install * Explicitly use version 16 * Add env * Add altair demo * Add unit test * CHANGELOG * Add image to changelog * Add outbreak_forcast notebook * Try again * generate again * Update ui/packages/plot/src/Plot.svelte Co-authored-by: aliabid94 <aabid94@gmail.com> * Fix title Co-authored-by: aliabid94 <aabid94@gmail.com>
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
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()
|