Update app.py

app.py

@@ -1,8 +1,7 @@
 import pandas as pd
 import numpy as np
-import matplotlib.pyplot as plt
-import plotly.graph_objs as go
 import gradio as gr
+import plotly.express as px
 
 def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, system3, users3, timeline3):
     # Create empty dataframe to hold usage data
@@ -23,22 +22,10 @@ def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, sy
         data = pd.concat([data, system_data], axis=0)
 
     # Create a larger plot with multiple subplots
-    fig = go.Figure()
-
-    # Plot the usage volume for each system
-    for i, system in enumerate([system1, system2, system3]):
-        system_data = data[data["System"] == system]
-        fig.add_trace(go.Scatter(x=system_data["Month"], y=system_data["Usage"], mode='lines', name=system))
-
-    # Add a line plot that aggregates the usage of all systems
-    aggregated_data = data.groupby("Month").sum().reset_index()
-    fig.add_trace(go.Scatter(x=aggregated_data["Month"], y=aggregated_data["Usage"], mode='lines', name="All Systems", line=dict(dash="dash", width=4)))
-
-    # Set the layout of the plot
-    fig.update_layout(xaxis_title="Month", yaxis_title="Usage Volume", title="System Rollout Usage Volume Plot")
+    fig = px.line(data, x="Month", y="Usage", color="System", title="System Rollout Usage Volume Plot")
 
     # Return the plot as a Gradio output
-    return fig
+    return gr.outputs.PlotlyOutput(fig)
 
 # Define the input components
 inputs = [
@@ -50,13 +37,17 @@ inputs = [
     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)]
-
-#Define the output component
-output = gr.outputs.Plotly(fig)
-
-#Create the interface
-iface = gr.Interface(fn=plot_usage_volume, title="System Rollout Usage Volume Plot", inputs=inputs, outputs=output, layout="horizontal")
-
-#Launch the interface
+    gr.inputs.Number(label="System 3 Timeline (months)", default=24)
+]
+
+# Define the interface
+iface = gr.Interface(
+    fn=plot_usage_volume,
+    inputs=inputs,
+    outputs="plot",
+    title="System Rollout Usage Volume Plot",
+    layout="horizontal"
+)
+
+# Launch the interface
 iface.launch()