Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import matplotlib.pyplot as plt
 import pandas as np
+import tempfile
 from datetime import datetime
 from langchain_core.messages import HumanMessage
 from tools import tools
@@ -84,11 +85,13 @@ def visualize_bmi_and_calories(bmi, calories):
     ax2.legend(loc="upper right")
 
     plt.tight_layout()
-    plt_path = "bmi_calorie_chart.png"
-    plt.savefig(plt_path)
+
+    # Save to a temporary file
+    temp_file = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
+    plt.savefig(temp_file.name)
     plt.close()
 
-    return plt_path
+    return temp_file.name
 
 
 def calculate_calories(age, weight, height, activity_level, gender):
@@ -160,10 +163,10 @@ with gr.Blocks() as demo:
         response = run_graph(f"{message}\n\n{personalized_metrics}", history, user_details)
         history.append((f"{user_details['name']}", message))
         history.append(("FIT.AI", response))
-        return history, ""
+        return history, chart_path
 
-    submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
-    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, text_input])
+    submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, bmi_chart])
+    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, bmi_chart])
 
     # Actions
     calculate_button = gr.Button("Calculate")
@@ -180,4 +183,4 @@ with gr.Blocks() as demo:
         outputs=[bmi_output, calorie_output, bmi_chart]
     )
 
-demo.launch(share=True)
+demo.launch(share=True)