Update app.py

app.py

@@ -64,10 +64,19 @@ simulation = Executor(exec_context=exec_context, configs=experiment.configs)
 
 
   
-def greet(name):
+def plot_seir(name):
     raw_result, tensor_fields, sessions = simulation.execute()
     result = pd.DataFrame(raw_result)
-    return "Hello " + name + "!!"
+    pd.options.plotting.backend = "plotly"
+    fig = result.plot(
+    kind = "line",
+    x = "timestep",
+    y= ["S","E","I", "R"])
+    fig.update_layout(title = name,
+            xaxis_title="Time (Days)",
+            yaxis_title="People")
+    return fig
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+graph = gr.Plot()
+iface = gr.Interface(fn=plot_seir, inputs="text", outputs=graph)
 iface.launch()