Update app.py

app.py

@@ -1,21 +1,9 @@
 import pandas as pd
 import numpy as np
-
 import matplotlib.pyplot as plt
 import gradio as gr
 
-def plot_usage_volume(inputs):
-    # Unpack the inputs
-    system1 = inputs["system1"]
-    users1 = inputs["users1"]
-    timeline1 = inputs["timeline1"]
-    system2 = inputs["system2"]
-    users2 = inputs["users2"]
-    timeline2 = inputs["timeline2"]
-    system3 = inputs["system3"]
-    users3 = inputs["users3"]
-    timeline3 = inputs["timeline3"]
-
+def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, system3, users3, timeline3):
     # Create empty dataframe to hold usage data
     data = pd.DataFrame(columns=["System", "Month", "Usage"])
 
@@ -29,6 +17,11 @@ def plot_usage_volume(inputs):
         ])
 
         # Add the usage data to the dataframe
+        months = ["Month {}".format(j) for j in range(1, len(usage) + 1)]
+        system_data = pd.DataFrame({"System": [system] * len(usage), "Month": months, "Usage": usage})
+        data = pd.concat([data, system_data], axis=0)
+
+    # Create a larger plot with multiple subplots
     fig, axs = plt.subplots(2, 1, figsize=(12, 8), sharex=True, gridspec_kw={'hspace': 0.5})
 
     # Plot the usage volume for each system
@@ -36,17 +29,24 @@ def plot_usage_volume(inputs):
         system_data = data[data["System"] == system]
         axs[0].plot(system_data["Month"], system_data["Usage"], label=system)
         axs[0].set_ylabel("Usage Volume")
+        axs[0].set_title("Usage Volume for Various Systems")
+
+    # Add a line plot that aggregates the usage of all systems
+    aggregated_data = data.groupby("Month").sum().reset_index()
+    axs[1].plot(aggregated_data["Month"], aggregated_data["Usage"], label="All Systems", linewidth=3, linestyle="--")
+    axs[1].set_xlabel("Month")
     axs[1].set_ylabel("Usage Volume")
     axs[1].set_title("Aggregated Usage Volume for All Systems")
 
- #Set xticks to only show every 3 months
+    # Set xticks to only show every 3 months
     axs[0].set_xticks(aggregated_data.index[::3])
     axs[1].set_xticks(aggregated_data.index[::3])
 
+    # Set xticklabels to show only the months with a step of 3
     axs[0].set_xticklabels(aggregated_data["Month"][::3], rotation=45, fontsize=10)
     axs[1].set_xticklabels(aggregated_data["Month"][::3], rotation=45, fontsize=10)
 
-        # Add a legend to the plot
+    # Add a legend to the plot
     axs[0].legend()
     axs[1].legend()
 
@@ -54,24 +54,25 @@ def plot_usage_volume(inputs):
     return gr.outputs.Image(fig)
 
 
-
-# Define the input components for the Gradio app
-inputs = {
-    "system1": gr.inputs.Textbox(label="System 1 Name", default="System1"),
-    "users1": gr.inputs.Slider(minimum=0, maximum=1000, default=500, label="System 1 Users"),
-    "timeline1": gr.inputs.Slider(minimum=1, maximum=12, default=6, label="System 1 Rollout Timeline (Months)"),
-    "system2": gr.inputs.Textbox(label="System 2 Name", default="System2"),
-    "users2": gr.inputs.Slider(minimum=0, maximum=1000, default=500, label="System 2 Users"),
-    "timeline2": gr.inputs.Slider(minimum=1, maximum=12, default=6, label="System 2 Rollout Timeline (Months)"),
-    "system3": gr.inputs.Textbox(label="System 3 Name", default="System3"),
-    "users3": gr.inputs.Slider(minimum=0, maximum=1000, default=500, label="System 3 Users"),
-    "timeline3": gr.inputs.Slider(minimum=1, maximum=12, default=6, label="System 3 Rollout Timeline (Months)")
-}
+# Create input components for the Gradio app
+inputs = [
+    gr.inputs.Textbox(label="System 1 Name", default="System1"),
+    gr.inputs.Number(label="Number of Users for System 1", default=1000),
+    gr.inputs.Number(label="Rollout Timeframe for System 1 (in Months)", default=12),
+    gr.inputs.Textbox(label="System 2 Name", default="System2"),
+    gr.inputs.Number(label="Number of Users for System 2", default=2000),
+    gr.inputs.Number(label="Rollout Timeframe for System 2 (in Months)", default=18),
+    gr.inputs.Textbox(label="System 3 Name", default="System3"),
+    gr.inputs.Number(label="Number of Users for System 3", default=3000),
+    gr.inputs.Number(label="Rollout Timeframe for System 3 (in Months)", default=24)
+]
 
 # Define the output component for the Gradio app
 output = gr.outputs.Image()
 
-# Define the Gradio interface and launch it
+# Define the title and description for the Gradio app
 title = "System Rollout Usage Volume Plot"
 description = "Enter the rollout parameters for three different systems and see the resulting usage volume plot."
-gr.Interface(plot_usage_volume, inputs, output, title=title, description=description).launch()
+
+# Create the Gradio interface and launch the app
+gr.Interface(plot_usage_volume, inputs, output, title=title, description=description).launch(share=True)