Update app.py

app.py

@@ -1,9 +1,21 @@
 import pandas as pd
 import numpy as np
+
 import matplotlib.pyplot as plt
 import gradio as gr
 
-def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, system3, users3, timeline3):
+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"]
+
     # Create empty dataframe to hold usage data
     data = pd.DataFrame(columns=["System", "Month", "Usage"])
 
@@ -17,11 +29,6 @@ def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, sy
         ])
 
         # 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
@@ -29,27 +36,28 @@ def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, sy
         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()
 
     # Return the plot as a Gradio output
     return gr.outputs.Image(fig)
 
+
+
+# Create input components for the Gradio app
+system1 = gr.inputs.Textbox(label="System 1 Name", default="System1")
+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()