removing after done?

app.py

@@ -19,13 +19,6 @@ import os
 #Load environment variables 
 load_dotenv()
 
-#Create Custom styling
-custom_css = """
-#component-2 {
-    height: 90vh !important;
-    min-height: 90vh !important;
-}
-"""
 
 #Create an agent that will create a prompt that can be used as input for the query agent from the Griptape Cloud. 
 
@@ -168,6 +161,11 @@ def send_message(message:str, history, request:gr.Request) -> Any:
         session_hash = request.session_hash
         agent = build_agent(session_hash,message)
     response = agent.run(message)
+    if "done" in message.lower():
+        try:
+            os.remove(f'{session_hash}.json')
+        except OSError as e:
+            pass
     return response.output.value
 
 demo = gr.ChatInterface(
@@ -175,10 +173,3 @@ demo = gr.ChatInterface(
     )
 demo.launch()
 
-json_files = glob.glob("*.json")
-for f in json_files:
-    try:
-        os.remove(f)
-    except OSError as e:
-        continue
-