Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import plotly
 import plotly.express as px
 import gradio as gr
+import json
 
 def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, system3, users3, timeline3):
     # Create empty dataframe to hold usage data
@@ -25,7 +26,7 @@ def plot_usage_volume(system1, users1, timeline1, system2, users2, timeline2, sy
     # Create a line plot of the usage volume for each system
     fig = px.line(data, x="Month", y="Usage", color="System", title="System Rollout Usage Volume Plot")
 
-    # Return the plot as a Plotly figure object
+    # Return the plot as a Gradio output
     return fig
 
 # Define the input components
@@ -42,11 +43,10 @@ inputs = [
 ]
 
 # Define the output component
-output = gr.outputs.Plot(label="Usage Volume Plot")
+output = gr.outputs.Graph(label="Usage Volume Plot")
 
 # Create the interface
 iface = gr.Interface(fn=plot_usage_volume, inputs=inputs, outputs=output, title="System Rollout Usage Volume Plot")
 
 # Launch the interface
 iface.launch()
-