Update app.py

app.py

@@ -38,21 +38,24 @@ def app():
       
       fig, ax = plt.subplots()
       
-      if "Top Selling" in plot["title"]:
-        sns.countplot(data=df, x=plot["x"], ax=plot["top"])
-      if "Category and Time of the Day" in plot["title"]:
-        sns.countplot(data=df, x=plot["x"], hue=plot["hue"])
-      if "Size and Time of the Day" in plot["title"]:
-        sns.countplot(data=df, x=plot["x"], hue=plot["hue"])
-      if "Revenue" in plot["title"]:
-        sns.countplot(data=df, x=plot["x"], ax=plot["top"], hue=plot["hue"], estimator=plot["estimator"], marker=plot["marker"])
+      if "Top Selling Pizzas" in plot["title"]:
+        data_aux = df.groupby(plot["x"])[plot["y"]].sum().reset_index().sort_values(by=plot["y"], ascending=False).head(plot["top"])
+        ax.bar(data_aux[plot["x"]].values.tolist(), data_aux[plot["y"]].values.tolist())
+      
+      if "Quantity of Pizzas" in plot["title"]:
+        sns.countplot(data=df, x=plot["x"], hue=plot["hue"], ax=ax)
+      
+      if "Monthly Revenue" in plot["title"]:
+        sns.lineplot(data=df, x=plot["x"], y=plot["y"], hue=plot["hue"], estimator=plot["estimator"], errorbar=None, marker=plot["marker"], ax=ax)
 
+      ax.set_xlabel(" ".join(plot["x"].split("_")).capitalize())
+      if "y" in plot.keys():
+        ax.set_ylabel(" ".join(plot["y"].split("_")).capitalize())
       else:
-        ax.set_ylabel("quantity")
-      ax.legend(bbox_to_anchor=(1,1)) 
-    
+        ax.set_ylabel("Quantity")
+      ax.legend(bbox_to_anchor=(1,1))
+
       st.pyplot(fig)
-      plt.show()
     
 if __name__ == "__main__":
     app()