mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-09 02:00:44 +08:00
88e9c19c27
* remove preprocess_example * removing methods * added path support for images * fixes * video * formatting * fixing preprocess * fixes * removed from audio * fixed file * formatting * serialization * foramtting * formatting * removed save flag / restore flag * formatting * removed flagging * removed * load value * fixing typing * fixes, typing * fixes * file * handling images * formatting * fixed serializing for flagging * formatting * json * temp file * removed processing * changed processing * fixed temp FINALLY * flagging works * fix examples test * formatting * async examples * working on mix * comment out failing test * fixed interface problem * fix kitchen sink deprecation warning * gallery examples * fixes * fixes to serialization * fixing label serializing * fixed file serialization * kitchen sink restored * outbreak forecast updated * formatting * formatting and api mode * fix 1 test :/ * fixing tests * fixed components tests * remvoed test files * formatting * fixed examples * fixes * formatting * restored certain files * added encryption * fixed syntax mistake * formatting * fixed 1 test * clean up interface * formatting * fixed route tests * more fixes * formatting * formatting * fixing pipeline * format frontend * format backend * tweaks * fix * fix final test? * merged * Sanitize for CSV (#2017) * sanitize for csv * added sanitization logic * fixed examples * turn cache off * fixed example caching with optional inputs * fixed review problems * fixed Interface.load * updating the tests * updating the tests * fix * fixed seriailizing * testing * rewrite run prediction * formatting * await * fixes * formatting * finally fixed mix * fixed tests * formatting * formatting * deserialize fix * formatting * fixes * fixes * fix * fix tests * fixes Co-authored-by: Freddy Boulton <alfonsoboulton@gmail.com>
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
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
|
|
|
|
import gradio as gr
|
|
|
|
|
|
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
|
|
else:
|
|
raise ValueError("A plot type must be selected")
|
|
|
|
inputs = [
|
|
gr.Dropdown(["Matplotlib", "Plotly"], 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],
|
|
["Plotly", 3.6, "February", ["Canada", "Mexico", "UK"], False],
|
|
], cache_examples=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|