Update app.py

app.py

@@ -3,26 +3,18 @@ import pypistats
 from datetime import date
 from dateutil.relativedelta import relativedelta
 import pandas as pd
-from fbprophet import Prophet
+from prophet import Prophet
 
 
 pd.options.plotting.backend = "plotly"
 
-def get_data(lib,time):
-    data = pypistats.overall(lib, total=True, format="pandas")
-    data = data.groupby("category").get_group("with_mirrors").sort_values("date")
-    start_date = date.today() - relativedelta(months=int(time.split(" ")[0]))
-    data = data[(data['date'] > str(start_date))] 
-    return data
-
-def get_plot2(lib, time):
+def get_forecast(lib, time):
 
     data = pypistats.overall(lib, total=True, format="pandas")
     data = data.groupby("category").get_group("with_mirrors").sort_values("date")
     start_date = date.today() - relativedelta(months=int(time.split(" ")[0]))
     df = data[(data['date'] > str(start_date))] 
 
-    #df = get_data(lib,time)
     df1 = df[['date','downloads']]
     df1.columns = ['ds','y']
 
@@ -33,11 +25,6 @@ def get_plot2(lib, time):
     fig1 = m.plot(forecast)
     return fig1 
 
-def get_plot(lib, time):
-    data = get_data(lib,time)
-    chart = data.plot(x="date", y="downloads")
-    return chart
-
 with gr.Blocks() as demo:
 
     gr.Markdown(
@@ -51,8 +38,8 @@ with gr.Blocks() as demo:
 
     plt = gr.Plot()
 
-    lib.change(get_plot2, [lib, time], plt)
-    time.change(get_plot2, [lib, time], plt)    
-    demo.load(get_plot2, [lib, time], plt)    
+    lib.change(get_forecast, [lib, time], plt)
+    time.change(get_forecast, [lib, time], plt)    
+    demo.load(get_forecast, [lib, time], plt)    
 
 demo.launch()