Update app.py

app.py

@@ -4,17 +4,17 @@ import pandas as pd
 import numpy as np
 
 
-def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, system3, users3, timeline3):
+def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2):
     # Create empty dataframe to hold usage data
     data = pd.DataFrame(columns=["System", "Month", "Usage"])
 
     # Generate usage data for each system based on typical rollout behaviors
-    for i, system in enumerate([system1, system2, system3]):
+    for i, system in enumerate([system1, system2]):
         # Calculate the monthly usage volume based on the estimated number of users and timeline
         usage = np.concatenate([
-            np.linspace(0, [users1, users2, users3][i], int([timeline1, timeline2, timeline3][i] * 0.25)),
-            np.linspace([users1, users2, users3][i] * 0.5, [users1, users2, users3][i], int([timeline1, timeline2, timeline3][i] * 0.5)),
-            np.linspace([users1, users2, users3][i], [users1, users2, users3][i] * 0.75, int([timeline1, timeline2, timeline3][i] * 0.25))
+            np.linspace(0, [users1, users2][i], int([timeline1, timeline2][i] * 0.25)),
+            np.linspace([users1, users2][i] * 0.5, [users1, users2][i], int([timeline1, timeline2][i] * 0.5)),
+            np.linspace([users1, users2][i], [users1, users2][i] * 0.75, int([timeline1, timeline2][i] * 0.25))
         ])
 
         # Add the usage data to the dataframe
@@ -39,16 +39,12 @@ inputs = [
     gr.inputs.Number(label="System 1 Timeline (months)", default=12),
     gr.inputs.Textbox(label="System 2 Name", default="System2"),
     gr.inputs.Number(label="System 2 Users", default=500),
-    gr.inputs.Number(label="System 2 Timeline (months)", default=18),
-    gr.inputs.Textbox(label="System 3 Name", default="System3"),
-    gr.inputs.Number(label="System 3 Users", default=2000),
-    gr.inputs.Number(label="System 3 Timeline (months)", default=24)
+    gr.inputs.Number(label="System 2 Timeline (months)", default=18)
 ]
 
 # Define the output component
 output = gr.Plot(label="Usage Volume Plot")
 
-
 # Create the interface
 iface = gr.Interface(fn=plot_usage_volume, inputs=inputs, outputs=output, title="System Rollout Usage Volume Plot")