added user data remove

app.py

@@ -14,6 +14,21 @@ import shutil
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(f"😎 {__name__}") # Do not wonder why I uses emoj in logger - It is visually easier to track
 
+def clear_data(sessionId):
+    folder_path = f"tmp/{sessionId}"
+    message = ""
+    try:
+        session_keys[sessionId] = {}
+        message += "API keys and model selection cleared! If you chat without adding new, there will be error."
+    except:
+        message += "Could not del API keys and model selection"
+    try:
+        if os.path.exists(folder_path):
+            shutil.rmtree(folder_path)
+        message += "Image Data and Color Analysis Data are cleared."
+    except:
+        message += "Failed to remove Image Data and Color Analysis Data"
+    return message
 
 def show_images(*image_urls):
     updates = []
@@ -168,6 +183,8 @@ def gradio_interface():
             bind_message_submission(send_btn.click)
 
         with gr.Tab("Settings"):
+            clear_data_btn = gr.Button("Clear Data", variant="primary", size="sm", interactive=True) 
+            data_clear_status = gr.Markdown("")
             gr.Markdown("## 🔧 Model Settings")
 
             provider_selector = gr.Dropdown(
@@ -216,6 +233,9 @@ def gradio_interface():
                 set_keys_btn = gr.Button("🔐 Save Keys")
                 key_feedback = gr.Markdown("")
 
+            clear_data_btn.click(fn=clear_data, inputs=session_id_state, outputs=data_clear_status)
+
+
             def set_user_keys(session_id, provider, openai_key, nebius_key, tool_model, response_model, vllm_model):
                 openai_key = openai_key.strip()
                 nebius_key = nebius_key.strip()
@@ -372,10 +392,10 @@ def gradio_interface():
     return demo
 
 if __name__ == "__main__":
-    folder_path = f"tmp"
-    if os.path.exists(folder_path):
-        shutil.rmtree(folder_path)
-        logger.info(f"Cleaned folder: {folder_path}")
+    # folder_path = f"tmp"
+    # if os.path.exists(folder_path):
+    #     shutil.rmtree(folder_path)
+    #     logger.info(f"Cleaned folder: {folder_path}")
     
     asyncio.get_event_loop().create_task(cleanup_old_sessions(threshold_seconds=600)) # 10 min
     interface = gradio_interface()