save plot example

app.py

@@ -12,15 +12,17 @@ df = pd.read_csv("USSTHPI.csv")
 python_repl_tool = PythonAstREPLTool(locals={"df": df})
 
 import matplotlib.pyplot as nplt
-nplt.figure(figsize=(10, 5))
-nplt.plot(df['DATE'], df['USSTHPI'], label='USSTHPI')
-nplt.title('Time Series of USSTHPI')
-nplt.xlabel('Date')
-nplt.ylabel('USSTHPI')
-nplt.legend()
-nplt.xticks(rotation=45)
-nplt.tight_layout()
-nplt.savefig('plots/plot.png')
+def gen_plot():
+    nplt.figure(figsize=(10, 5))
+    nplt.plot(df['DATE'], df['USSTHPI'], label='USSTHPI')
+    nplt.title('Time Series of USSTHPI')
+    nplt.xlabel('Date')
+    nplt.ylabel('USSTHPI')
+    nplt.legend()
+    nplt.xticks(rotation=45)
+    nplt.tight_layout()
+    nplt.savefig('plots/plot.png')
+    return "plots/plot.png"
 
 # cell 2
 from langchain.agents import AgentExecutor, create_openai_tools_agent
@@ -106,5 +108,5 @@ def greet(name):
     resp = graph.invoke({"messages": [HumanMessage(content="Run the analysis")]}, debug=True)
     return resp
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+demo = gr.Interface(fn=gen_plot, inputs="text", outputs="text")
 demo.launch()