Update app.py

app.py

@@ -65,28 +65,11 @@ def cleanup_session_files(request: gr.Request):
         # Get the session ID from the request
         session_id = request.session_hash
         
-        # Check common gradio temp locations
-        gradio_temp_dirs = [
-            "/tmp/gradio",
-            os.path.join(tempfile.gettempdir(), "gradio"),
-            os.path.join(os.getcwd(), "gradio")
-        ]
-        
-        files_cleaned = 0
-        
-        # Clean session directories in gradio temp folders
-        for temp_dir in gradio_temp_dirs:
-            session_dir = os.path.join(temp_dir, session_id)
-            if os.path.exists(session_dir):
-                try:
-                    shutil.rmtree(session_dir)
-                    files_cleaned += 1
-                    print(f"Cleaned up session directory: {session_dir}")
-                except (OSError, FileNotFoundError):
-                    pass
-        
-        if files_cleaned > 0:
-            print(f"Session cleanup completed: {files_cleaned} directories removed for session {session_id}")
+        # Clean up the session directory
+        session_dir = os.path.join("/tmp/gradio", session_id)
+        if os.path.exists(session_dir):
+            shutil.rmtree(session_dir)
+            print(f"Cleaned up session directory: {session_dir}")
             
     except Exception as e:
         print(f"Error during session cleanup: {e}")