fix unload bug

app.py

@@ -528,8 +528,8 @@ def update_log_display():
 ###############################################################################
 
 
-def cleanup_session(request: gr.Request):
-    session_id = request.session_hash
+def cleanup_session(req: gr.Request):
+    session_id = req.session_hash
     task_id = SESSION_TASKS.pop(session_id, None)
     
     if task_id:
@@ -544,7 +544,9 @@ def cleanup_session(request: gr.Request):
                     print(f"终止任务失败 {task_id}: {res.get('status', 'unknown issue')}")
         except Exception as e:
             print(f"终止任务失败 {task_id}: {e}")
-
+    
+    user_dir = os.path.join(TMP_ROOT, str(req.session_hash))
+    shutil.rmtree(user_dir)
 
 
 ###############################################################################
@@ -639,9 +641,7 @@ def start_session(req: gr.Request):
     os.makedirs(user_dir, exist_ok=True)
     
     
-def end_session(req: gr.Request):
-    user_dir = os.path.join(TMP_ROOT, str(req.session_hash))
-    shutil.rmtree(user_dir)
+    
 
 
 # 创建Gradio界面
@@ -794,7 +794,7 @@ with gr.Blocks(title="InternManip Model Inference Demo", css=custom_css) as demo
     
     demo.queue(default_concurrency_limit=8)
 
-    demo.unload(fn=cleanup_session).then(end_session)
+    demo.unload(fn=cleanup_session)
 
 
 # 启动应用